import { AdminService, useAdminService } from "@services/useAdminService.ts";

export class ArtifactTypes {
    public static AVRO = "AVRO";
    public static PROTOBUF = "PROTOBUF";
    public static JSON = "JSON";
    public static OPENAPI = "OPENAPI";
    public static ASYNCAPI = "ASYNCAPI";
    public static GRAPHQL = "GRAPHQL";
    public static KCONNECT = "KCONNECT";
    public static WSDL = "WSDL";
    public static XSD = "XSD";
    public static XML = "XML";
    public static AGENT_CARD = "AGENT_CARD";
    public static MODEL_SCHEMA = "MODEL_SCHEMA";
    public static PROMPT_TEMPLATE = "PROMPT_TEMPLATE";

    public static getTitle(type: string): string {
        let title: string = type;
        switch (type) {
            case "AVRO":
                title = "Avro Schema";
                break;
            case "PROTOBUF":
                title = "Protobuf Schema";
                break;
            case "JSON":
                title = "JSON Schema";
                break;
            case "OPENAPI":
                title = "OpenAPI Definition";
                break;
            case "ASYNCAPI":
                title = "AsyncAPI Definition";
                break;
            case "GRAPHQL":
                title = "GraphQL Definition";
                break;
            case "KCONNECT":
                title = "Kafka Connect Schema";
                break;
            case "WSDL":
                title = "WSDL";
                break;
            case "XSD":
                title = "XML Schema";
                break;
            case "XML":
                title = "XML";
                break;
            case "AGENT_CARD":
                title = "A2A Agent Card";
                break;
            case "MODEL_SCHEMA":
                title = "AI Model Schema";
                break;
            case "PROMPT_TEMPLATE":
                title = "Prompt Template";
                break;
        }
        return title;
    }

    public static getLabel(type: string): string {
        let title: string = type;
        switch (type) {
            case "AVRO":
                title = "Avro Schema";
                break;
            case "PROTOBUF":
                title = "Protocol Buffer Schema";
                break;
            case "JSON":
                title = "JSON Schema";
                break;
            case "OPENAPI":
                title = "OpenAPI";
                break;
            case "ASYNCAPI":
                title = "AsyncAPI";
                break;
            case "GRAPHQL":
                title = "GraphQL";
                break;
            case "KCONNECT":
                title = "Kafka Connect Schema";
                break;
            case "WSDL":
                title = "WSDL";
                break;
            case "XSD":
                title = "XML Schema";
                break;
            case "XML":
                title = "XML";
                break;
            case "AGENT_CARD":
                title = "Agent Card";
                break;
            case "MODEL_SCHEMA":
                title = "Model Schema";
                break;
            case "PROMPT_TEMPLATE":
                title = "Prompt Template";
                break;
        }
        return title;
    }

    public static getClassNames(type: string): string {
        let classes: string = "artifact-type-icon";
        switch (type) {
            case "AVRO":
                classes += " avro-icon24";
                break;
            case "PROTOBUF":
                classes += " protobuf-icon24";
                break;
            case "JSON":
                classes += " json-icon24";
                break;
            case "OPENAPI":
                classes += " oai-icon24";
                break;
            case "ASYNCAPI":
                classes += " aai-icon24";
                break;
            case "GRAPHQL":
                classes += " graphql-icon24";
                break;
            case "KCONNECT":
                classes += " kconnect-icon24";
                break;
            case "WSDL":
                classes += " xml-icon24";
                break;
            case "XSD":
                classes += " xml-icon24";
                break;
            case "XML":
                classes += " xml-icon24";
                break;
            case "AGENT_CARD":
                classes += " agentcard-icon24";
                break;
            case "MODEL_SCHEMA":
                classes += " modelschema-icon24";
                break;
            case "PROMPT_TEMPLATE":
                classes += " prompttemplate-icon24";
                break;
            default:
                classes += " questionmark-icon24";
                break;
        }
        return classes;
    }

}

let ALL_TYPES: string[] | undefined = undefined;

const allTypes = async (admin: AdminService): Promise<string[]> => {
    if (ALL_TYPES === undefined) {
        try {
            ALL_TYPES = (await admin.getArtifactTypes()).map(t => t.name!);
        } catch {
            ALL_TYPES = ["AVRO", "PROTOBUF", "JSON", "OPENAPI", "ASYNCAPI", "GRAPHQL", "KCONNECT", "WSDL", "XSD", "XML"];
        }
    }
    return Promise.resolve(ALL_TYPES);
};

const allTypesWithLabels = async (admin: AdminService): Promise<any[]> => {
    return allTypes(admin).then(types => {
        return types.map(t => { return { id: t, label: ArtifactTypes.getLabel(t) }; });
    });
};


export interface ArtifactTypesService {
    allTypes(): Promise<string[]>;
    allTypesWithLabels(): Promise<any[]>;
}

export const useArtifactTypesService: () => ArtifactTypesService = (): ArtifactTypesService => {
    const admin: AdminService = useAdminService();

    return {
        allTypes(): Promise<string[]> {
            return allTypes(admin);
        },
        allTypesWithLabels(): Promise<any[]> {
            return allTypesWithLabels(admin);
        }
    };
};
