model

type CallEdge struct {
	From *FunctionDefinition `json:"from"`
	To   *FunctionDefinition `json:"to"`
}

type CallGraph struct {
	Roots     []FunctionDefinition `json:"roots"`
	Functions []FunctionDefinition `json:"functions"`
	Edges     []CallEdge           `json:"edges"`
	// contains filtered or unexported fields
}

type ChunkType string

type CodeChunk struct {
	// Unique identifier for this chunk
	ID string `json:"id"`

	// FileID from MySQL file_versions table (shared with CodeGraph)
	FileID int32 `json:"file_id"`

	// Hierarchical metadata
	ChunkType ChunkType `json:"chunk_type"`
	Level     int       `json:"level"` // 1=file, 2=class, 3=function, 4=block
	ParentID  string    `json:"parent_id,omitempty"`

	// Code content
	Content   string     `json:"content"`
	Language  string     `json:"language"`
	FilePath  string     `json:"file_path"`
	StartLine int        `json:"start_line"`
	EndLine   int        `json:"end_line"`
	Range     base.Range `json:"range"`

	// Semantic metadata
	Name      string `json:"name,omitempty"`      // Function/class/file name
	Signature string `json:"signature,omitempty"` // Function signature
	Docstring string `json:"docstring,omitempty"` // Documentation

	// Context for better understanding
	ModuleName string `json:"module_name,omitempty"` // Package/module name
	ClassName  string `json:"class_name,omitempty"`  // Parent class if method

	// Vector embedding (generated by embedding model)
	Embedding []float32 `json:"embedding,omitempty"`

	// Additional metadata
	Metadata map[string]interface{} `json:"metadata,omitempty"`
}

type FileInfo struct {
	Path     string `json:"path"`
	Language string `json:"language"`
}

type FunctionDefinition struct {
	Name       string        `json:"name"`
	Location   base.Location `json:"location"`
	IsExternal bool          `json:"is_external"`
	Module     string        `json:"module,omitempty"`
	Params     string        `json:"params"`
	Returns    string        `json:"returns"`
}

type FunctionDependency struct {
	Name          string             `json:"name"`
	CallLocations []base.Location    `json:"call_locations"`
	Definition    FunctionDefinition `json:"definition"`
}

type FunctionDetails struct {
	Name          string        `json:"name"`
	Signature     string        `json:"signature"`
	Parameters    []Parameter   `json:"parameters"`
	ReturnType    string        `json:"return_type"`
	IsAsync       bool          `json:"is_async"`
	Documentation string        `json:"documentation"`
	Location      base.Location `json:"location"`
}

type GetFunctionDependenciesRequest struct {
	RepoName     string `json:"repo_name" binding:"required"`
	RelativePath string `json:"relative_path" binding:"required"`
	FunctionName string `json:"function_name" binding:"required"`
	Depth        int    `json:"depth"`
}

type GetFunctionDependenciesResponse struct {
	RepoName     string               `json:"repo_name"`
	FilePath     string               `json:"file_path"`
	FunctionName string               `json:"function_name"`
	Dependencies []FunctionDependency `json:"dependencies"`
}

type GetFunctionDetailsRequest struct {
	RepoName     string `json:"repo_name" binding:"required"`
	RelativePath string `json:"relative_path" binding:"required"`
	FunctionName string `json:"function_name" binding:"required"`
}

type GetFunctionDetailsResponse struct {
	RepoName     string          `json:"repo_name"`
	FilePath     string          `json:"file_path"`
	FunctionName string          `json:"function_name"`
	Details      FunctionDetails `json:"details"`
}

type GetFunctionsInFileRequest struct {
	RepoName     string `json:"repo_name" binding:"required"`
	RelativePath string `json:"relative_path" binding:"required"`
}

type GetFunctionsInFileResponse struct {
	RepoName  string               `json:"repo_name"`
	FilePath  string               `json:"file_path"`
	Functions []FunctionDefinition `json:"functions"`
}

type Parameter struct {
	Name          string `json:"name"`
	Type          string `json:"type"`
	Optional      bool   `json:"optional"`
	Documentation string `json:"documentation"`
}

type ProcessDirectoryRequest struct {
	RepoName       string `json:"repo_name" binding:"required"`
	CollectionName string `json:"collection_name"`
}

type ProcessDirectoryResponse struct {
	RepoName       string `json:"repo_name"`
	CollectionName string `json:"collection_name"`
	TotalChunks    int    `json:"total_chunks"`
	Success        bool   `json:"success"`
	Message        string `json:"message,omitempty"`
}

type ProcessRepoResponse struct {
	RepoName  string               `json:"repo_name"`
	Files     []FileInfo           `json:"files"`
	Functions []FunctionDefinition `json:"functions"`
}

type QueryInfo struct {
	CodeSnippet string       `json:"code_snippet"`
	Language    string       `json:"language"`
	ChunksFound int          `json:"chunks_found"`
	Chunks      []*CodeChunk `json:"chunks"` // The parsed chunks from the input snippet
}

type SearchSimilarCodeRequest struct {
	RepoName       string `json:"repo_name" binding:"required"`
	CollectionName string `json:"collection_name"`
	CodeSnippet    string `json:"code_snippet" binding:"required"`
	Language       string `json:"language" binding:"required"`
	Limit          int    `json:"limit"`
	IncludeCode    bool   `json:"include_code"`
}

type SearchSimilarCodeResponse struct {
	RepoName       string              `json:"repo_name"`
	CollectionName string              `json:"collection_name"`
	Query          QueryInfo           `json:"query"`
	Results        []SimilarCodeResult `json:"results"`
	Success        bool                `json:"success"`
	Message        string              `json:"message,omitempty"`
}

type SimilarCodeResult struct {
	Chunk           *CodeChunk `json:"chunk"`
	Score           float32    `json:"score"`
	QueryChunkIndex int        `json:"query_chunk_index"` // Index of the input chunk that matched this result (0-based)
	Code            string     `json:"code,omitempty"`    // Actual code content from file (if include_code is true)
}