cli

type AppConfig struct {
	AQLQuery             string
	ArangoUser           string // For authorship in gRPC updates
	NumProcessingWorkers int
	NumGrpcWorkers       int
	Logger               *logrus.Entry
	Metrics              *ProcessingMetrics // Add this field
}

type ArangoProperty struct {
	Name  string `json:"name"`
	Value string `json:"value"`
}

type ArangoResultDoc struct {
	ID    string           `json:"id"` // This is dbx.accession, likely the feature_id
	Props []ArangoProperty `json:"props"`
}

type BatchGeneProductJob struct {
	GeneProducts []ProcessedGeneProduct
}

type BatchGeneProductResult struct {
	GeneID         string
	Success        bool
	Message        string
	Error          error
	ProcessedCount int
	SkippedCount   int
}

type FeatureAnnotationAppConfig struct {
	Ctx              context.Context
	NumPubmedWorkers int
	NumGrpcWorkers   int
	Logger           *logrus.Entry
	Metrics          *FeatureAnnotationMetrics
}

type FeatureAnnotationMetrics struct {
	TotalProcessed int64
	SuccessCount   int64
	ErrorCount     int64
	StartTime      time.Time

	TotalFetchedFromArango      int64
	AllArangoDocsFetched        bool
	JobsSubmittedToPubmedPool   int64
	JobsCompletedFromPubmedPool int64
	JobsSubmittedToGrpcPool     int64
	JobsCompletedFromGrpcPool   int64
	// contains filtered or unexported fields
}

type FileContext struct {
	Setup  SetupConfig
	File   *os.File
	Reader *csv.Reader
	Error  error // To propagate errors
}

type Gene struct {
	FeatureID int    `json:"feature_id"`
	GeneID    string `json:"gene_id"`
	Name      string `json:"name"`
	CreatedBy string `json:"created_by"`
}

type GeneInfo struct {
	Name      string `json:"name"`
	GeneID    string `json:"gene_id"`
	FeatureID int64  `json:"feature_id"`
	CreatedBy string `json:"created_by"`
}

type GeneProduct struct {
	GeneID  string
	Product string
}

type GeneProductAppConfig struct {
	Ctx              context.Context
	LegacyDatabase   string
	NumLegacyWorkers int
	NumGrpcWorkers   int
	Logger           *logrus.Entry
	Metrics          *GeneProductMetrics
}

type GeneProductMetrics struct {
	TotalProcessed int64
	SuccessCount   int64
	ErrorCount     int64
	SkippedCount   int64
	StartTime      time.Time

	TotalFetchedFromArango      int64
	AllArangoDocsFetched        bool
	JobsSubmittedToLegacyPool   int64
	JobsCompletedFromLegacyPool int64
	JobsSubmittedToGrpcPool     int64
	JobsCompletedFromGrpcPool   int64
	// contains filtered or unexported fields
}

type GeneProductResult struct {
	GeneProduct string     `json:"gene_product"`
	CreatedBy   string     `json:"created_by"`
	CreatedOn   LegacyTime `json:"created_on"`
}

type GeneWithPubmed struct {
	Gene
	Pubmeds    []string
	Skip       bool
	SkipReason string
}

type GrpcAnnotationResult struct {
	GeneID  string
	Success bool
	Skipped bool
	Message string
	Error   error
}

type GrpcSynonymResult struct {
	GeneID  string
	Success bool
	Message string
	Error   error
}

type GrpcUpdateResult struct {
	GeneID  string
	Success bool
	Message string
	Error   error
}

type LegacyTime struct {
	time.Time
}

type PipelineResult struct {
	File        *os.File
	Setup       SetupConfig
	UpdateCount int
	Error       error
}

type ProcessSingleRecordParams struct {
	Record             []string
	FeaturePropIDIndex int
	ValueIndex         int
	RowNumForLogging   int // Actual row number in the CSV file for logging
	Logger             *logrus.Entry
}

type ProcessedGeneData struct {
	GeneID            string
	StrippedPropsText []StrippedProperty
}

type ProcessedGeneProduct struct {
	GeneID      string
	GeneName    string
	GeneProduct string
	CreatedBy   string
	CreatedOn   time.Time
}

type ProcessingContext struct {
	FileContext
	FeaturePropIDIndex int
	ValueIndex         int
}

type ProcessingMetrics struct {
	TotalProcessed int64
	SuccessCount   int64
	ErrorCount     int64
	StartTime      time.Time

	// TotalFetchedFromArango stores the total number of items fetched by queryArango.
	// This field is set by queryArango once the total count is known.
	TotalFetchedFromArango int64
	// AllArangoDocsFetched is a flag set to true by queryArango after all documents
	// have been fetched and sent to the processing pipeline.
	AllArangoDocsFetched bool
	// Intermediate tracking counters for detailed pipeline monitoring
	JobsSubmittedToHTMLPool   int64
	JobsCompletedFromHTMLPool int64
	JobsSubmittedToGrpcPool   int64
	JobsCompletedFromGrpcPool int64
	// contains filtered or unexported fields
}

type SetupConfig struct {
	Logger         *logrus.Entry
	DBH            *arangomanager.Database
	CSVFilePath    string
	CollectionName string
	BatchSize      int
	Delimiter      string
	Workers        int
}

type StrippedProperty struct {
	OriginalName string
	StrippedText string
}

type SubmitBatchAndLogParams struct {
	Setup            *SetupConfig
	Docs             []map[string]string
	Logger           *logrus.Entry
	BatchDescription string
}

type SynonymAppConfig struct {
	Ctx            context.Context
	NumGrpcWorkers int
	Logger         *logrus.Entry
	Metrics        *SynonymMetrics
}

type SynonymData struct {
	Name     string   `json:"name"`
	GeneID   string   `json:"gene_id"`
	Synonyms []string `json:"synonyms"`
}

type SynonymMetrics struct {
	TotalProcessed int64
	SuccessCount   int64
	NotFoundCount  int64
	ErrorCount     int64
	StartTime      time.Time

	TotalFetchedFromArango    int64
	AllArangoDocsFetched      bool
	JobsSubmittedToGrpcPool   int64
	JobsCompletedFromGrpcPool int64
	// contains filtered or unexported fields
}