types

type Confidence string

type DetectionResult struct {
	Files            map[FileType][]string `json:"files"`
	TotalFiles       int                   `json:"total_files"`
	TotalWords       int                   `json:"total_words"`
	NeedsGraph       bool                  `json:"needs_graph"`
	Warning          string                `json:"warning,omitempty"`
	SkippedSensitive []string              `json:"skipped_sensitive"`
	GfyIgnoreCount   int                   `json:"gfyignore_patterns"`
}

type Edge struct {
	Source          string     `json:"source"`
	Target          string     `json:"target"`
	Relation        string     `json:"relation"`
	Confidence      Confidence `json:"confidence"`
	ConfidenceScore float64    `json:"confidence_score,omitempty"`
	SourceFile      string     `json:"source_file"`
	SourceLocation  string     `json:"source_location,omitempty"`
	Weight          float64    `json:"weight"`
}

type ExtractionResult struct {
	Nodes        []Node    `json:"nodes"`
	Edges        []Edge    `json:"edges"`
	RawCalls     []RawCall `json:"raw_calls,omitempty"`
	InputTokens  int       `json:"input_tokens"`
	OutputTokens int       `json:"output_tokens"`
	Error        string    `json:"error,omitempty"`
}

type FileType string

type Node struct {
	ID             string   `json:"id"`
	Label          string   `json:"label"`
	FileType       string   `json:"file_type"`
	SourceFile     string   `json:"source_file"`
	SourceLocation string   `json:"source_location,omitempty"`
	Tags           []string `json:"tags,omitempty"`
	Comment        string   `json:"comment,omitempty"`
	// Content holds the full or truncated text body of a document-type node
	// (e.g., markdown/PDF/DOCX/XLSX). It is used as input to embedding and
	// semantic search but is generally not part of the AST extraction path.
	Content       string   `json:"content,omitempty"`
	LogMessages   []string `json:"log_messages,omitempty"`
	ThrowMessages []string `json:"throw_messages,omitempty"`
}

type RawCall struct {
	CallerNID    string `json:"caller_nid"`
	Callee       string `json:"callee"`
	IsMemberCall bool   `json:"is_member_call"`
	SourceFile   string `json:"source_file"`
	SourceLoc    string `json:"source_location"`
}