cmd

type AssetResult struct {
	Name     string      `json:"name"`
	Type     string      `json:"type"`
	Provider string      `json:"provider"`
	MRN      string      `json:"mrn"`
	Asset    interface{} `json:"asset"`
	Status   string      `json:"status"`
	Error    string      `json:"error,omitempty"`
}

type BatchCreateRequest struct {
	Assets        []CreateAssetRequest     `json:"assets"`
	Lineage       []CreateLineageRequest   `json:"lineage"`
	Documentation []CreateDocRequest       `json:"documentation"`
	Statistics    []CreateStatisticRequest `json:"statistics"`
	Config        plugin.RawPluginConfig   `json:"config"`
	PipelineName  string                   `json:"pipeline_name"`
	SourceName    string                   `json:"source_name"`
	RunID         string                   `json:"run_id"`
}

type BatchCreateResponse struct {
	Assets               []AssetResult         `json:"assets"`
	Lineage              []LineageResult       `json:"lineage"`
	Documentation        []DocumentationResult `json:"documentation"`
	StaleEntitiesRemoved []string              `json:"stale_entities_removed,omitempty"`
}

type CompleteRunRequest struct {
	RunID   string             `json:"run_id"`
	Status  plugin.RunStatus   `json:"status"`
	Summary *plugin.RunSummary `json:"summary"`
	Error   string             `json:"error,omitempty"`
}

type CreateAssetRequest struct {
	Name          string                 `json:"name"`
	Type          string                 `json:"type"`
	Providers     []string               `json:"providers"`
	Description   *string                `json:"description"`
	Metadata      map[string]interface{} `json:"metadata"`
	Schema        map[string]interface{} `json:"schema"`
	Tags          []string               `json:"tags"`
	Sources       []string               `json:"sources"`
	ExternalLinks []map[string]string    `json:"external_links"`
}

type CreateDocRequest struct {
	AssetMRN string `json:"asset_mrn"`
	Content  string `json:"content"`
	Type     string `json:"type"`
}

type CreateLineageRequest struct {
	Source string `json:"source"`
	Target string `json:"target"`
	Type   string `json:"type"`
}

type CreateStatisticRequest struct {
	AssetMRN   string  `json:"asset_mrn"`
	MetricName string  `json:"metric_name"`
	Value      float64 `json:"value"`
}

type DestroyRunResponse struct {
	AssetsDeleted        int      `json:"assets_deleted"`
	LineageDeleted       int      `json:"lineage_deleted"`
	DocumentationDeleted int      `json:"documentation_deleted"`
	DeletedEntityMRNs    []string `json:"deleted_entity_mrns"`
}

type DocumentationResult struct {
	AssetMRN string `json:"asset_mrn"`
	Type     string `json:"type"`
	Status   string `json:"status"`
	Error    string `json:"error,omitempty"`
}

type LineageResult struct {
	Source string `json:"source"`
	Target string `json:"target"`
	Type   string `json:"type"`
	Status string `json:"status"`
	Error  string `json:"error,omitempty"`
}

type StartRunRequest struct {
	PipelineName string                 `json:"pipeline_name"`
	SourceName   string                 `json:"source_name"`
	Config       plugin.RawPluginConfig `json:"config"`
}

type Summary struct {
	AssetsCreated      int
	AssetsUpdated      int
	AssetsDeleted      int
	AssetsUnchanged    int
	LineageCreated     int
	LineageUpdated     int
	DocumentationAdded int
	ErrorsEncountered  int
}