import type { z } from "zod";

export type Scope = "global" | "project";
export type JsonObject = Record<string, unknown>;

export interface ChiConfigChange {
	scope: Scope;
	key: string;
	previous: unknown;
	current: unknown;
}

export interface ChiConfigContract<TConfig extends JsonObject = JsonObject> {
	schemaVersion: number;
	schema: z.ZodObject<any>;
	migrate?: (
		data: JsonObject,
		fromSchemaVersion: number,
		scope: Scope,
	) => JsonObject;
}

export interface ChiModuleDefinition<
	TApi = unknown,
	TConfig extends JsonObject = JsonObject,
> {
	id: string;
	version: string;
	dependencies?: readonly string[];
	api?: TApi;
	config: ChiConfigContract<TConfig>;
	initialize(chi: ChiBase, config: TConfig): void | Promise<void>;
	onConfigChange?(
		chi: ChiBase,
		current: TConfig,
		changes: readonly ChiConfigChange[],
	): void | Promise<void>;
}

export interface CohortIdentity extends JsonObject {
	github: string;
	displayName: string;
	team: string;
	role: string;
	relayPublicKey: string;
}

export interface ChiBaseApi {
	getIdentity(): CohortIdentity;
	onIdentityChange(
		listener: (identity: CohortIdentity) => void,
	): () => void;
}

export interface PromptContributionInput {
	priority: number;
	maxChars: number;
	content: string;
}

export interface PromptContribution extends PromptContributionInput {
	moduleId: string;
	updatedAt: string;
}

export type ChiModuleState =
	| "registered"
	| "initializing"
	| "ready"
	| "blocked"
	| "failed";

export interface ChiModuleStatus {
	id: string;
	version: string;
	dependencies: readonly string[];
	state: ChiModuleState;
	diagnostic?: string;
}

export interface ChiBase {
	register<TApi, TConfig extends JsonObject>(
		definition: ChiModuleDefinition<TApi, TConfig>,
	): void;
	get<TApi>(id: string): TApi | undefined;
	require<TApi>(id: string): TApi;
	list(): readonly ChiModuleStatus[];
	getConfig<TConfig extends JsonObject>(id: string): TConfig | undefined;
	setConfigValue<TConfig extends JsonObject>(
		id: string,
		scope: Scope,
		key: string,
		value: unknown | undefined,
	): Promise<TConfig>;
	setPromptContribution(
		moduleId: string,
		contribution: PromptContributionInput | undefined,
	): void;
	getPromptContributions(): readonly PromptContribution[];
}