// Define all possible features
export const ADDON_FEATURES = ["evals", "experiments", "prompts"] as const;
export const NON_FREE_FEATURES = [
  "sessions",
  "properties",
  "users",
  "datasets",
  "alerts",
] as const;

// Create a combined type of all features
export type FeatureId =
  | (typeof ADDON_FEATURES)[number]
  | (typeof NON_FREE_FEATURES)[number];

// Define all possible subfeatures for each feature
export const SUBFEATURES = {
  prompts: ["versions", "runs", "playground_runs"],
  experiments: ["test_cases", "variants"],
  evals: ["runs"],
  datasets: ["requests"],
} as const;

// Create a type for all possible subfeature IDs
export type SubfeatureId =
  | (typeof SUBFEATURES.prompts)[number]
  | (typeof SUBFEATURES.experiments)[number]
  | (typeof SUBFEATURES.evals)[number]
  | (typeof SUBFEATURES.datasets)[number];

// Helper functions to type check features and subfeatures
export function isFeature(feature: string): feature is FeatureId {
  return [...ADDON_FEATURES, ...NON_FREE_FEATURES].includes(feature as any);
}

export function isSubfeature(subfeature: string): subfeature is SubfeatureId {
  return Object.values(SUBFEATURES)
    .flat()
    .includes(subfeature as any);
}

// Get subfeatures for a specific feature
export function getSubfeaturesForFeature(
  feature: FeatureId,
): readonly string[] {
  return (SUBFEATURES as any)[feature] || [];
}

// Create lookup table for feature display names (useful for UI)
export const FEATURE_DISPLAY_NAMES: Record<FeatureId, string> = {
  prompts: "Prompts",
  experiments: "Experiments",
  evals: "Evaluators",
  sessions: "Sessions",
  properties: "Properties",
  users: "Users",
  datasets: "Datasets",
  alerts: "Alerts",
};

// Create lookup table for subfeature display names
export const SUBFEATURE_DISPLAY_NAMES: Record<SubfeatureId, string> = {
  versions: "Versions",
  test_cases: "Test Cases",
  variants: "Variants",
  runs: "Prompt Runs",
  playground_runs: "Playground Runs",
  requests: "Requests",
};