query

type CentralityInfo struct {
	// Betweenness is the betweenness centrality score.
	Betweenness float64 `json:"betweenness"`

	// BetweennessRank is the rank among all nodes (1 = highest).
	BetweennessRank int `json:"betweenness_rank"`

	// IsHub indicates if this is a highly connected node.
	IsHub bool `json:"is_hub"`

	// IsBridge indicates if this is a betweenness bridge.
	IsBridge bool `json:"is_bridge"`
}

type CommunityInfo struct {
	// CommunityID is the cluster ID this node belongs to.
	CommunityID int `json:"community_id"`

	// CommunityLabel is a human-readable label for the community.
	CommunityLabel string `json:"community_label"`

	// CommunitySize is the number of nodes in the community.
	CommunitySize int `json:"community_size"`

	// IsBridge indicates if this node connects multiple communities.
	IsBridge bool `json:"is_bridge"`

	// BridgesCommunities lists communities this node connects (if bridge).
	BridgesCommunities []int `json:"bridges_communities,omitempty"`
}

type DepthLayer struct {
	Depth int      `json:"depth"`
	Count int      `json:"count"`
	Nodes []string `json:"nodes"`
}

type EdgeFilter struct {
	// NodeID filters edges that touch this node (from or to).
	NodeID string

	// From filters edges by source node.
	From string

	// To filters edges by target node.
	To string

	// Type filters edges by edge type.
	Type string

	// Types filters edges by multiple edge types (OR).
	Types []string

	// Limit is the maximum number of edges to return.
	Limit int
}

type EdgeInfo struct {
	From       string `json:"from"`
	To         string `json:"to"`
	Type       string `json:"type"`
	Confidence string `json:"confidence,omitempty"`
}

type EdgeListOutput struct {
	Query     string               `json:"query,omitempty"`
	Matches   int                  `json:"matches"`
	Edges     []EdgeOutputWithMeta `json:"edges"`
	Truncated bool                 `json:"truncated,omitempty"`
	Message   string               `json:"message,omitempty"`
}

type EdgeOutput struct {
	From string `json:"from"`
	To   string `json:"to"`
	Type string `json:"type"`
}

type EdgeOutputWithMeta struct {
	From       string `json:"from"`
	To         string `json:"to"`
	Type       string `json:"type"`
	Confidence string `json:"confidence,omitempty"`
}

type ExplainOptions struct {
	// Depth controls how many levels of neighbors to include.
	Depth int

	// MaxNeighbors limits the number of neighbors shown.
	MaxNeighbors int

	// IncludeEdges includes edge details in output.
	IncludeEdges bool
}

type FormatTraversalOptions struct {
	// Limit is the maximum nodes/edges to include per layer.
	Limit int

	// Algorithm is "BFS" or "DFS".
	Algorithm string

	// Direction is the traversal direction.
	Direction string
}

type GodNode struct {
	ID        string `json:"id"`
	EdgeCount int    `json:"edges"`
}

type NeighborInfo struct {
	// InDegree is the number of incoming edges.
	InDegree int `json:"in_degree"`

	// OutDegree is the number of outgoing edges.
	OutDegree int `json:"out_degree"`

	// TotalDegree is in + out degree.
	TotalDegree int `json:"total_degree"`

	// IncomingEdges lists edges pointing to this node.
	IncomingEdges []EdgeInfo `json:"incoming_edges,omitempty"`

	// OutgoingEdges lists edges from this node.
	OutgoingEdges []EdgeInfo `json:"outgoing_edges,omitempty"`

	// EdgeTypeBreakdown shows count by edge type.
	EdgeTypeBreakdown map[string]int `json:"edge_type_breakdown"`
}

type NodeExplanation struct {
	// Node is the basic node information.
	Node NodeInfo `json:"node"`

	// Neighbors describes the node's immediate connections.
	Neighbors NeighborInfo `json:"neighbors"`

	// Community describes which community the node belongs to.
	Community CommunityInfo `json:"community"`

	// Centrality describes the node's centrality metrics.
	Centrality CentralityInfo `json:"centrality"`

	// SourceFile is the file containing this node (if applicable).
	SourceFile string `json:"source_file,omitempty"`

	// Package is the package this node belongs to (if applicable).
	Package string `json:"package,omitempty"`
}

type NodeInfo struct {
	ID    string            `json:"id"`
	Type  string            `json:"type"`
	Label string            `json:"label"`
	Attrs map[string]string `json:"attrs,omitempty"`
}

type NodeMatch struct {
	Query   string   `json:"query"`
	Matches []string `json:"matches"`
	Message string   `json:"message"`
}

type PathOutput struct {
	From    string       `json:"from"`
	To      string       `json:"to"`
	Found   bool         `json:"found"`
	Length  int          `json:"length,omitempty"`
	Path    []string     `json:"path,omitempty"`
	Edges   []EdgeOutput `json:"edges,omitempty"`
	Message string       `json:"message,omitempty"`
}

type Summary struct {
	TotalNodes int            `json:"total_nodes"`
	TotalEdges int            `json:"total_edges"`
	NodeTypes  map[string]int `json:"node_types"`
	EdgeTypes  map[string]int `json:"edge_types"`
	GodNodes   []GodNode      `json:"god_nodes"`
}

type SummaryOptions struct {
	// MaxGodNodes is the maximum number of god nodes to return.
	MaxGodNodes int

	// ExcludePrefixes are node ID prefixes to exclude from god nodes.
	ExcludePrefixes []string

	// ExcludeContains are substrings that exclude nodes from god nodes.
	ExcludeContains []string
}

type TraversalOutput struct {
	Query      string       `json:"query"`
	Algorithm  string       `json:"algorithm"`
	Direction  string       `json:"direction"`
	MaxDepth   int          `json:"max_depth"`
	NodesFound int          `json:"nodes_found"`
	EdgesFound int          `json:"edges_found"`
	Layers     []DepthLayer `json:"layers"`
	Edges      []EdgeOutput `json:"edges"`
}