db

type DBTX interface {
	ExecContext(context.Context, string, ...interface{}) (sql.Result, error)
	PrepareContext(context.Context, string) (*sql.Stmt, error)
	QueryContext(context.Context, string, ...interface{}) (*sql.Rows, error)
	QueryRowContext(context.Context, string, ...interface{}) *sql.Row
}

type Edge struct {
	ID           string          `json:"id"`
	SourceNodeID string          `json:"source_node_id"`
	TargetNodeID string          `json:"target_node_id"`
	Type         string          `json:"type"`
	Confidence   sql.NullFloat64 `json:"confidence"`
	RunID        sql.NullString  `json:"run_id"`
	CreatedAt    time.Time       `json:"created_at"`
}

type GetEdgesBySourceAndTypeParams struct {
	SourceNodeID string `json:"source_node_id"`
	Type         string `json:"type"`
}

type GetEdgesByTargetAndTypeParams struct {
	TargetNodeID string `json:"target_node_id"`
	Type         string `json:"type"`
}

type InsertEdgeParams struct {
	ID           string          `json:"id"`
	SourceNodeID string          `json:"source_node_id"`
	TargetNodeID string          `json:"target_node_id"`
	Type         string          `json:"type"`
	Confidence   sql.NullFloat64 `json:"confidence"`
	RunID        sql.NullString  `json:"run_id"`
}

type InsertNodeParams struct {
	ID                    string          `json:"id"`
	Level                 int16           `json:"level"`
	ContentClass          string          `json:"content_class"`
	ContentType           string          `json:"content_type"`
	EncodingClass         string          `json:"encoding_class"`
	EncodingFormat        string          `json:"encoding_format"`
	ContentSha256         string          `json:"content_sha256"`
	ContentLen            int64           `json:"content_len"`
	ContentCached         sql.NullString  `json:"content_cached"`
	ArtifactCreatedAt     sql.NullTime    `json:"artifact_created_at"`
	ArtifactCreatedAtBlur sql.NullString  `json:"artifact_created_at_blur"`
	Origin                sql.NullString  `json:"origin"`
	OriginalName          sql.NullString  `json:"original_name"`
	ValidFrom             sql.NullTime    `json:"valid_from"`
	ValidFromBlur         sql.NullString  `json:"valid_from_blur"`
	ValidUntil            sql.NullTime    `json:"valid_until"`
	ValidUntilBlur        sql.NullString  `json:"valid_until_blur"`
	Confidence            sql.NullFloat64 `json:"confidence"`
}

type InsertRunParams struct {
	ID             string `json:"id"`
	WorkerConfigID string `json:"worker_config_id"`
}

type ListEdgesByTypeParams struct {
	Type   string `json:"type"`
	Limit  int32  `json:"limit"`
	Offset int32  `json:"offset"`
}

type ListEdgesParams struct {
	Limit  int32 `json:"limit"`
	Offset int32 `json:"offset"`
}

type ListNodesByContentClassAndTypeParams struct {
	ContentClass string `json:"content_class"`
	ContentType  string `json:"content_type"`
	Limit        int32  `json:"limit"`
	Offset       int32  `json:"offset"`
}

type ListNodesByContentClassParams struct {
	ContentClass string `json:"content_class"`
	Limit        int32  `json:"limit"`
	Offset       int32  `json:"offset"`
}

type ListNodesByLevelParams struct {
	Level  int16 `json:"level"`
	Limit  int32 `json:"limit"`
	Offset int32 `json:"offset"`
}

type ListNodesParams struct {
	Limit  int32 `json:"limit"`
	Offset int32 `json:"offset"`
}

type Node struct {
	ID                    string          `json:"id"`
	Level                 int16           `json:"level"`
	ContentClass          string          `json:"content_class"`
	ContentType           string          `json:"content_type"`
	EncodingClass         string          `json:"encoding_class"`
	EncodingFormat        string          `json:"encoding_format"`
	ContentSha256         string          `json:"content_sha256"`
	ContentLen            int64           `json:"content_len"`
	ContentCached         sql.NullString  `json:"content_cached"`
	CreatedAt             time.Time       `json:"created_at"`
	ArtifactCreatedAt     sql.NullTime    `json:"artifact_created_at"`
	ArtifactCreatedAtBlur sql.NullString  `json:"artifact_created_at_blur"`
	Origin                sql.NullString  `json:"origin"`
	OriginalName          sql.NullString  `json:"original_name"`
	ValidFrom             sql.NullTime    `json:"valid_from"`
	ValidFromBlur         sql.NullString  `json:"valid_from_blur"`
	ValidUntil            sql.NullTime    `json:"valid_until"`
	ValidUntilBlur        sql.NullString  `json:"valid_until_blur"`
	Confidence            sql.NullFloat64 `json:"confidence"`
}

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

type Run struct {
	ID             string    `json:"id"`
	WorkerConfigID string    `json:"worker_config_id"`
	CreatedAt      time.Time `json:"created_at"`
}

type SearchNodesByContentParams struct {
	PlaintoTsquery string `json:"plainto_tsquery"`
	Limit          int32  `json:"limit"`
	Offset         int32  `json:"offset"`
}