persistence

type AssociationMapper struct{}

type AssociationStore struct {
	database.Repository[enrichment.Association, EnrichmentAssociationModel]
}

type BranchMapper struct{}

type BranchModel struct {
	RepoID        int64           `gorm:"column:repo_id;primaryKey;index"`
	Repo          RepositoryModel `gorm:"foreignKey:RepoID;references:ID;constraint:OnDelete:CASCADE"`
	Name          string          `gorm:"column:name;primaryKey;index;size:255"`
	HeadCommitSHA string          `gorm:"column:head_commit_sha;index;size:64"`
	IsDefault     bool            `gorm:"column:is_default;default:false"`
	CreatedAt     time.Time       `gorm:"column:created_at"`
	UpdatedAt     time.Time       `gorm:"column:updated_at"`
}

type BranchStore struct {
	database.Repository[repository.Branch, BranchModel]
}

type CommitMapper struct{}

type CommitModel struct {
	CommitSHA       string          `gorm:"column:commit_sha;primaryKey;size:64"`
	RepoID          int64           `gorm:"column:repo_id;index"`
	Repo            RepositoryModel `gorm:"foreignKey:RepoID;references:ID;constraint:OnDelete:CASCADE"`
	Date            time.Time       `gorm:"column:date"`
	Message         string          `gorm:"column:message;type:text"`
	ParentCommitSHA *string         `gorm:"column:parent_commit_sha;index;size:64"`
	Author          string          `gorm:"column:author;index;size:255"`
	CreatedAt       time.Time       `gorm:"column:created_at"`
	UpdatedAt       time.Time       `gorm:"column:updated_at"`
}

type CommitStore struct {
	database.Repository[repository.Commit, CommitModel]
}

type EmbeddingModel struct {
	ID        int64     `gorm:"column:id;primaryKey;autoIncrement"`
	SnippetID string    `gorm:"column:snippet_id;index"`
	Type      string    `gorm:"column:type;index"`
	Embedding []float64 `gorm:"column:embedding;type:json"`
	CreatedAt time.Time `gorm:"column:created_at;not null"`
	UpdatedAt time.Time `gorm:"column:updated_at;not null"`
}

type EnrichmentAssociationModel struct {
	ID           int64           `gorm:"column:id;primaryKey;autoIncrement"`
	EnrichmentID int64           `gorm:"column:enrichment_id;not null;index;uniqueIndex:idx_enrichment_assoc"`
	Enrichment   EnrichmentModel `gorm:"foreignKey:EnrichmentID;references:ID;constraint:OnDelete:CASCADE"`
	EntityType   string          `gorm:"column:entity_type;size:50;not null;index:idx_enrichment_entity;uniqueIndex:idx_enrichment_assoc"`
	EntityID     string          `gorm:"column:entity_id;size:255;not null;index:idx_enrichment_entity;uniqueIndex:idx_enrichment_assoc"`
	CreatedAt    time.Time       `gorm:"column:created_at;not null"`
	UpdatedAt    time.Time       `gorm:"column:updated_at;not null"`
}

type EnrichmentMapper struct{}

type EnrichmentModel struct {
	ID        int64     `gorm:"column:id;primaryKey;autoIncrement"`
	Type      string    `gorm:"column:type;not null;index"`
	Subtype   string    `gorm:"column:subtype;not null;index"`
	Content   string    `gorm:"column:content;type:text;not null"`
	Language  string    `gorm:"column:language;size:50"`
	CreatedAt time.Time `gorm:"column:created_at;not null"`
	UpdatedAt time.Time `gorm:"column:updated_at;not null"`
}

type EnrichmentStore struct {
	database.Repository[enrichment.Enrichment, EnrichmentModel]
}

type FileMapper struct{}

type FileModel struct {
	ID        int64       `gorm:"column:id;autoIncrement"`
	CommitSHA string      `gorm:"column:commit_sha;primaryKey;uniqueIndex:idx_commit_path;size:64"`
	Commit    CommitModel `gorm:"foreignKey:CommitSHA;references:CommitSHA;constraint:-"`
	Path      string      `gorm:"column:path;primaryKey;uniqueIndex:idx_commit_path;size:1024"`
	BlobSHA   string      `gorm:"column:blob_sha;index;size:64"`
	MimeType  string      `gorm:"column:mime_type;index;size:255"`
	Extension string      `gorm:"column:extension;index;size:255"`
	Size      int64       `gorm:"column:size"`
	CreatedAt time.Time   `gorm:"column:created_at"`
}

type FileStore struct {
	database.Repository[repository.File, FileModel]
}

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

type FilterOperator int

type Float64Slice []float64

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

type PgEmbeddingModel struct {
	ID        int64             `gorm:"column:id;primaryKey;autoIncrement"`
	SnippetID string            `gorm:"column:snippet_id;uniqueIndex"`
	Embedding database.PgVector `gorm:"column:embedding;type:vector"`
}

type PipelineMapper struct{}

type PipelineStepMapper struct{}

type PipelineStepStore struct {
	database.Repository[repository.PipelineStep, models.PipelineStep]
}

type PipelineStore struct {
	database.Repository[repository.Pipeline, models.Pipeline]
}

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

type RepositoryMapper struct{}

type RepositoryModel struct {
	ID                 int64           `gorm:"primaryKey;autoIncrement"`
	PipelineID         *int64          `gorm:"column:pipeline_id;index"`
	Pipeline           models.Pipeline `gorm:"foreignKey:PipelineID;constraint:OnDelete:SET NULL"`
	SanitizedRemoteURI string          `gorm:"column:sanitized_remote_uri;index;uniqueIndex;size:1024"`
	RemoteURI          string          `gorm:"column:remote_uri;size:1024"`
	UpstreamURL        *string         `gorm:"column:upstream_url;index;size:1024"`
	ClonedPath         *string         `gorm:"column:cloned_path;size:1024"`
	LastScannedAt      *time.Time      `gorm:"column:last_scanned_at"`
	NumCommits         int             `gorm:"column:num_commits;default:0"`
	NumBranches        int             `gorm:"column:num_branches;default:0"`
	NumTags            int             `gorm:"column:num_tags;default:0"`
	TrackingType       string          `gorm:"column:tracking_type;index;size:255"`
	TrackingName       string          `gorm:"column:tracking_name;index;size:255"`
	ChunkSize          int             `gorm:"column:chunk_size"`
	ChunkOverlap       int             `gorm:"column:chunk_overlap"`
	MinChunkSize       int             `gorm:"column:min_chunk_size"`
	CreatedAt          time.Time       `gorm:"column:created_at"`
	UpdatedAt          time.Time       `gorm:"column:updated_at"`
}

type RepositoryStore struct {
	database.Repository[repository.Repository, RepositoryModel]
}

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

type SQLiteEmbeddingModel struct {
	ID        int64        `gorm:"column:id;primaryKey;autoIncrement"`
	SnippetID string       `gorm:"column:snippet_id;uniqueIndex"`
	Embedding Float64Slice `gorm:"column:embedding;type:json"`
}

type SQLiteEmbeddingStore struct {
	database.Repository[search.Embedding, SQLiteEmbeddingModel]
	// contains filtered or unexported fields
}

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

type SortDirection int

type SourceLocationMapper struct{}

type SourceLocationModel struct {
	ID           int64           `gorm:"column:id;primaryKey;autoIncrement"`
	EnrichmentID int64           `gorm:"column:enrichment_id;not null;uniqueIndex"`
	Enrichment   EnrichmentModel `gorm:"foreignKey:EnrichmentID;references:ID;constraint:OnDelete:CASCADE"`
	Page         int             `gorm:"column:page;not null;default:0"`
	StartLine    int             `gorm:"column:start_line;not null"`
	EndLine      int             `gorm:"column:end_line;not null"`
}

type SourceLocationStore struct {
	database.Repository[sourcelocation.SourceLocation, SourceLocationModel]
}

type StatusStore struct {
	database.Repository[task.Status, TaskStatusModel]
}

type StepDependencyMapper struct{}

type StepDependencyStore struct {
	database.Repository[repository.StepDependency, models.StepDependency]
}

type StepMapper struct{}

type StepStore struct {
	database.Repository[repository.Step, models.Step]
}

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

type TagMapper struct{}

type TagModel struct {
	RepoID          int64           `gorm:"column:repo_id;primaryKey;index"`
	Repo            RepositoryModel `gorm:"foreignKey:RepoID;references:ID;constraint:OnDelete:CASCADE"`
	Name            string          `gorm:"column:name;primaryKey;index;size:255"`
	TargetCommitSHA string          `gorm:"column:target_commit_sha;index;size:64"`
	Message         *string         `gorm:"column:message;type:text"`
	TaggerName      *string         `gorm:"column:tagger_name;size:255"`
	TaggerEmail     *string         `gorm:"column:tagger_email;size:255"`
	TaggedAt        *time.Time      `gorm:"column:tagged_at"`
	CreatedAt       time.Time       `gorm:"column:created_at"`
	UpdatedAt       time.Time       `gorm:"column:updated_at"`
}

type TagStore struct {
	database.Repository[repository.Tag, TagModel]
}

type TaskMapper struct{}

type TaskModel struct {
	ID        int64           `gorm:"column:id;primaryKey;autoIncrement"`
	DedupKey  string          `gorm:"column:dedup_key;type:varchar(255);uniqueIndex;not null"`
	Type      string          `gorm:"column:type;type:varchar(255);index;not null"`
	Payload   json.RawMessage `gorm:"column:payload;type:jsonb"`
	Priority  int             `gorm:"column:priority;not null"`
	CreatedAt time.Time       `gorm:"column:created_at;autoCreateTime"`
	UpdatedAt time.Time       `gorm:"column:updated_at;autoUpdateTime"`
}

type TaskName string

type TaskStatusMapper struct{}

type TaskStatusModel struct {
	ID            string    `gorm:"column:id;type:varchar(255);primaryKey;index;not null"`
	CreatedAt     time.Time `gorm:"column:created_at;not null"`
	UpdatedAt     time.Time `gorm:"column:updated_at;not null"`
	Operation     string    `gorm:"column:operation;type:varchar(255);index;not null"`
	TrackableID   *int64    `gorm:"column:trackable_id;index:idx_trackable"`
	TrackableType *string   `gorm:"column:trackable_type;type:varchar(255);index:idx_trackable"`
	ParentID      *string   `gorm:"column:parent;type:varchar(255);index"`
	Message       string    `gorm:"column:message;type:text;default:''"`
	State         string    `gorm:"column:state;type:varchar(255);default:''"`
	Error         string    `gorm:"column:error;type:text;default:''"`
	Total         int       `gorm:"column:total;default:0"`
	Current       int       `gorm:"column:current;default:0"`
}

type TaskStore struct {
	database.Repository[task.Task, TaskModel]
}

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

type VectorChordEmbeddingStore struct {
	database.Repository[search.Embedding, PgEmbeddingModel]
	// contains filtered or unexported fields
}