mcp

type DataProductService interface {
	Get(ctx context.Context, id string) (*dataproduct.DataProduct, error)
	List(ctx context.Context, offset, limit int) (*dataproduct.ListResult, error)
	Search(ctx context.Context, filter dataproduct.SearchFilter) (*dataproduct.ListResult, error)
	GetResolvedAssets(ctx context.Context, dataProductID string, limit, offset int) (*dataproduct.ResolvedAssets, error)
	GetDataProductsForAsset(ctx context.Context, assetID string) ([]*dataproduct.DataProduct, error)
}

type DiscoverDataInput struct {
	Query           string                `json:"query,omitempty"`
	ID              string                `json:"id,omitempty"`
	MRN             string                `json:"mrn,omitempty"`
	Types           []string              `json:"types,omitempty"`
	Providers       []string              `json:"providers,omitempty"`
	Tags            []string              `json:"tags,omitempty"`
	MetadataFilters []MetadataFilterInput `json:"metadata_filters,omitempty"`
	Limit           int                   `json:"limit,omitempty"`
	Offset          int                   `json:"offset,omitempty"`
}

type ExploreDataProductsInput struct {
	Query  string   `json:"query,omitempty"`
	ID     string   `json:"id,omitempty"`
	Name   string   `json:"name,omitempty"`
	Tags   []string `json:"tags,omitempty"`
	Limit  int      `json:"limit,omitempty"`
	Offset int      `json:"offset,omitempty"`
}

type ExploreTeamsInput struct {
	TeamID   string `json:"team_id,omitempty"`
	TeamName string `json:"team_name,omitempty"`
	UserID   string `json:"user_id,omitempty"`
	Username string `json:"username,omitempty"`
	Limit    int    `json:"limit,omitempty"`
	Offset   int    `json:"offset,omitempty"`
}

type FindOwnershipInput struct {
	AssetID              string `json:"asset_id,omitempty"`
	UserID               string `json:"user_id,omitempty"`
	Username             string `json:"username,omitempty"`
	TeamID               string `json:"team_id,omitempty"`
	TeamName             string `json:"team_name,omitempty"`
	IncludeAssets        bool   `json:"include_assets,omitempty"`
	IncludeGlossaryTerms bool   `json:"include_glossary_terms,omitempty"`
	Limit                int    `json:"limit,omitempty"`
	Offset               int    `json:"offset,omitempty"`
}

type GlossaryService interface {
	Get(ctx context.Context, id string) (*glossary.GlossaryTerm, error)
	Search(ctx context.Context, filter glossary.SearchFilter) (*glossary.ListResult, error)
	GetChildren(ctx context.Context, parentID string) ([]*glossary.GlossaryTerm, error)
	GetAncestors(ctx context.Context, termID string) ([]*glossary.GlossaryTerm, error)
}

type LookupTermInput struct {
	Query        string `json:"query,omitempty"`
	TermID       string `json:"term_id,omitempty"`
	ParentTermID string `json:"parent_term_id,omitempty"`
	Limit        int    `json:"limit,omitempty"`
	Offset       int    `json:"offset,omitempty"`
}

type MetadataFilterInput struct {
	Key      string `json:"key"`
	Operator string `json:"operator"`
	Value    any    `json:"value"`
}

type Owner struct {
	Type  string  `json:"type"`
	ID    string  `json:"id"`
	Name  string  `json:"name"`
	Email *string `json:"email,omitempty"`
}

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

type Team struct {
	ID          string   `json:"id"`
	Name        string   `json:"name"`
	Description string   `json:"description,omitempty"`
	Tags        []string `json:"tags,omitempty"`
}

type TeamMember struct {
	UserID   string  `json:"user_id"`
	Username string  `json:"username"`
	Name     string  `json:"name"`
	Email    *string `json:"email,omitempty"`
	Role     string  `json:"role"`
}

type TeamService interface {
	GetTeam(ctx context.Context, id string) (*Team, error)
	GetTeamByName(ctx context.Context, name string) (*Team, error)
	FindSimilarTeamNames(ctx context.Context, searchTerm string, limit int) ([]string, error)
	ListAssetOwners(ctx context.Context, assetID string) ([]Owner, error)
	ListTeams(ctx context.Context, limit, offset int) ([]*Team, int, error)
	ListMembers(ctx context.Context, teamID string) ([]*TeamMember, error)
	ListUserTeams(ctx context.Context, userID string) ([]*Team, error)
}

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

type TraceLineageInput struct {
	AssetID   string `json:"asset_id,omitempty"`
	MRN       string `json:"mrn,omitempty"`
	Direction string `json:"direction,omitempty"`
	Depth     int    `json:"depth,omitempty"`
}