database

type Document struct {
	ID string

	Created      time.Time
	Sequence     int64  `sql:"seq"`
	DocType      string `sql:"doc_type"`
	EdgarURL     string `sql:"edgar_url"`
	Description  string `sql:"doc_desc"`
	Data         []byte `sql:"data"`
	SizeEstimate string `sql:"size_est"`
	FilingID     string `sql:"filing_id"`
}

type DocumentDao interface {
	GetByFilingID(filingID string) (exists bool, docs []*model.Document, err error)
	GetIDs(filingID string) (exists bool, ids []string, err error)
	Get(documentID string) (exists bool, doc *model.Document, err error)
}

type DocumentDaoImpl struct {
	// contains filtered or unexported fields
}

type Filer struct {
	ID             string
	Symbol         string
	CIK            string
	Name           string `sql:"name"`
	SIC            string
	SICDescription string `sql:"sic_desc"`
}

type FilerDao interface {
	Search(condition string) (bool, []*model.Filer, error)
	GetSet(term string) (bool, []*model.Filer, error)
	Put(filer *model.Filer) (bool, error)
}

type FilerDaoImpl struct {
	// contains filtered or unexported fields
}

type Filing struct {
	ID string

	Created           time.Time
	Updated           time.Time
	Filer             string
	Accession         string
	CIK               string
	EdgarURL          string    `sql:"edgar_url"`
	EdgarTime         time.Time `sql:"edgar_time"`
	FilerRelation     string    `sql:"relation"`
	FormType          string    `sql:"form_type"`
	DocumentCount     int64     `sql:"doc_count"`
	TotalSizeEstimate string    `sql:"total_size_est"`
	Symbol            string
	AllSymbols        []string `sql:"all_symbols,array"`
	AllCIKs           []string `sql:"all_ciks,array"`
}

type FilingDao interface {
	StrictExists(cik, accession, relation string) (bool, error)
	Add(filing *model.Filing) (created *model.Filing, existed bool, err error)
	Update(filing *model.Filing) (bool, error)
	AddDocuments(filing *model.Filing, docs []*model.Document) error
	Get(accession string) (exists bool, filing *model.Filing, err error)

	GetByID(filingID string) (exists bool, filing *model.Filing, err error)
	GetBySymbol(params *QueryParams) (exists bool, result *QueryResult, err error)
	GetByCIK(params *QueryParams) (exists bool, result *QueryResult, err error)
	GetByFiler(params *QueryParams) (exists bool, result *QueryResult, err error)
	GetAll(params *QueryParams) (exists bool, result *QueryResult, err error)
}

type FilingDaoImpl struct {
	// contains filtered or unexported fields
}

type Form struct {
	ID          string
	FormType    string `sql:"form_type"`
	Description string `sql:"form_desc"`
	Active      bool
}

type FormDao interface {
	GetActiveForms() ([]*model.Form, error)
}

type FormDaoImpl struct {
	// contains filtered or unexported fields
}

type Handle struct {
	*pg.DB
	// contains filtered or unexported fields
}

type Options struct {
	Addr     string
	User     string
	Password string
	Database string
}

type QueryParams struct {
	Term  string
	Page  int
	Size  int
	Sort  int
	Form  string
	Start string
	End   string
}

type QueryResult struct {
	Filings    []*model.Filing
	TotalCount int
}