import { terminal_methods, } from "./terminal.js";
import { UpdateStatus, } from "./types.js";
export * as Types from "./types.js";
export class CancelToken {
    cancelled;
    constructor() {
        this.cancelled = false;
    }
    cancel() {
        this.cancelled = true;
    }
}
/** Initialize a new client for Komodo */
export function KomodoClient(url, options) {
    const state = {
        jwt: options.type === "jwt" ? options.params.jwt : undefined,
        key: options.type === "api-key" ? options.params.key : undefined,
        secret: options.type === "api-key" ? options.params.secret : undefined,
    };
    const request = (path, type, params) => new Promise(async (res, rej) => {
        try {
            let response = await fetch(`${url}${path}/${type}`, {
                method: "POST",
                body: JSON.stringify(params),
                headers: {
                    ...(state.jwt
                        ? {
                            authorization: state.jwt,
                        }
                        : state.key && state.secret
                            ? {
                                "x-api-key": state.key,
                                "x-api-secret": state.secret,
                            }
                            : {}),
                    "content-type": "application/json",
                },
            });
            if (response.status === 200) {
                const body = await response.json();
                res(body);
            }
            else {
                try {
                    const result = await response.json();
                    rej({ status: response.status, result });
                }
                catch (error) {
                    rej({
                        status: response.status,
                        result: {
                            error: "Failed to get response body",
                            trace: [JSON.stringify(error)],
                        },
                        error,
                    });
                }
            }
        }
        catch (error) {
            rej({
                status: 1,
                result: {
                    error: "Request failed with error",
                    trace: [JSON.stringify(error)],
                },
                error,
            });
        }
    });
    const auth = async (type, params) => await request("/auth", type, params);
    const user = async (type, params) => await request("/user", type, params);
    const read = async (type, params) => await request("/read", type, params);
    const write = async (type, params) => await request("/write", type, params);
    const execute = async (type, params) => await request("/execute", type, params);
    const execute_and_poll = async (type, params) => {
        const res = await execute(type, params);
        // Check if its a batch of updates or a single update;
        if (Array.isArray(res)) {
            const batch = res;
            return await Promise.all(batch.map(async (item) => {
                if (item.status === "Err") {
                    return item;
                }
                return await poll_update_until_complete(item.data._id?.$oid);
            }));
        }
        else {
            // it is a single update
            const update = res;
            if (update.status === UpdateStatus.Complete || !update._id?.$oid) {
                return update;
            }
            return await poll_update_until_complete(update._id?.$oid);
        }
    };
    const poll_update_until_complete = async (update_id) => {
        while (true) {
            await new Promise((resolve) => setTimeout(resolve, 1000));
            const update = await read("GetUpdate", { id: update_id });
            if (update.status === UpdateStatus.Complete) {
                return update;
            }
        }
    };
    const core_version = () => read("GetVersion", {}).then((res) => res.version);
    const get_update_websocket = ({ on_update, on_login, on_open, on_close, }) => {
        const ws = new WebSocket(url.replace("http", "ws") + "/ws/update");
        // Handle login on websocket open
        ws.addEventListener("open", () => {
            on_open?.();
            const login_msg = options.type === "jwt"
                ? {
                    type: "Jwt",
                    params: {
                        jwt: options.params.jwt,
                    },
                }
                : {
                    type: "ApiKeys",
                    params: {
                        key: options.params.key,
                        secret: options.params.secret,
                    },
                };
            ws.send(JSON.stringify(login_msg));
        });
        ws.addEventListener("message", ({ data }) => {
            if (data == "LOGGED_IN")
                return on_login?.();
            on_update(JSON.parse(data));
        });
        if (on_close) {
            ws.addEventListener("close", on_close);
        }
        return ws;
    };
    const subscribe_to_update_websocket = async ({ on_update, on_open, on_login, on_close, retry = true, retry_timeout_ms = 5_000, cancel = new CancelToken(), on_cancel, }) => {
        while (true) {
            if (cancel.cancelled) {
                on_cancel?.();
                return;
            }
            try {
                const ws = get_update_websocket({
                    on_open,
                    on_login,
                    on_update,
                    on_close,
                });
                // This while loop will end when the socket is closed
                while (ws.readyState !== WebSocket.CLOSING &&
                    ws.readyState !== WebSocket.CLOSED) {
                    if (cancel.cancelled)
                        ws.close();
                    // Sleep for a bit before checking for websocket closed
                    await new Promise((resolve) => setTimeout(resolve, 500));
                }
                if (retry) {
                    // Sleep for a bit before retrying connection to avoid spam.
                    await new Promise((resolve) => setTimeout(resolve, retry_timeout_ms));
                }
                else {
                    return;
                }
            }
            catch (error) {
                console.error(error);
                if (retry) {
                    // Sleep for a bit before retrying, maybe Komodo Core is down temporarily.
                    await new Promise((resolve) => setTimeout(resolve, retry_timeout_ms));
                }
                else {
                    return;
                }
            }
        }
    };
    const { connect_terminal, execute_terminal, execute_terminal_stream, connect_exec, connect_container_exec, execute_container_exec, execute_container_exec_stream, connect_deployment_exec, execute_deployment_exec, execute_deployment_exec_stream, connect_stack_exec, execute_stack_exec, execute_stack_exec_stream, } = terminal_methods(url, state);
    return {
        /**
         * Call the `/auth` api.
         *
         * ```
         * const login_options = await komodo.auth("GetLoginOptions", {});
         * ```
         *
         * https://docs.rs/komodo_client/latest/komodo_client/api/auth/index.html
         */
        auth,
        /**
         * Call the `/user` api.
         *
         * ```
         * const { key, secret } = await komodo.user("CreateApiKey", {
         *   name: "my-api-key"
         * });
         * ```
         *
         * https://docs.rs/komodo_client/latest/komodo_client/api/user/index.html
         */
        user,
        /**
         * Call the `/read` api.
         *
         * ```
         * const stack = await komodo.read("GetStack", {
         *   stack: "my-stack"
         * });
         * ```
         *
         * https://docs.rs/komodo_client/latest/komodo_client/api/read/index.html
         */
        read,
        /**
         * Call the `/write` api.
         *
         * ```
         * const build = await komodo.write("UpdateBuild", {
         *   id: "my-build",
         *   config: {
         *     version: "1.0.4"
         *   }
         * });
         * ```
         *
         * https://docs.rs/komodo_client/latest/komodo_client/api/write/index.html
         */
        write,
        /**
         * Call the `/execute` api.
         *
         * ```
         * const update = await komodo.execute("DeployStack", {
         *   stack: "my-stack"
         * });
         * ```
         *
         * NOTE. These calls return immediately when the update is created, NOT when the execution task finishes.
         * To have the call only return when the task finishes, use [execute_and_poll_until_complete].
         *
         * https://docs.rs/komodo_client/latest/komodo_client/api/execute/index.html
         */
        execute,
        /**
         * Call the `/execute` api, and poll the update until the task has completed.
         *
         * ```
         * const update = await komodo.execute_and_poll("DeployStack", {
         *   stack: "my-stack"
         * });
         * ```
         *
         * https://docs.rs/komodo_client/latest/komodo_client/api/execute/index.html
         */
        execute_and_poll,
        /**
         * Poll an Update (returned by the `execute` calls) until the `status` is `Complete`.
         * https://docs.rs/komodo_client/latest/komodo_client/entities/update/struct.Update.html#structfield.status.
         */
        poll_update_until_complete,
        /** Returns the version of Komodo Core the client is calling to. */
        core_version,
        /**
         * Connects to update websocket, performs login and attaches handlers,
         * and returns the WebSocket handle.
         */
        get_update_websocket,
        /**
         * Subscribes to the update websocket with automatic reconnect loop.
         *
         * Note. Awaiting this method will never finish.
         */
        subscribe_to_update_websocket,
        /**
         * Subscribes to terminal io over websocket message,
         * for use with xtermjs.
         */
        connect_terminal,
        /**
         * Executes a command on a given Server / terminal,
         * and gives a callback to handle the output as it comes in.
         *
         * ```ts
         * await komodo.execute_terminal(
         *   {
         *     server: "my-server",
         *     terminal: "name",
         *     command: 'for i in {1..3}; do echo "$i"; sleep 1; done',
         *   },
         *   {
         *     onLine: (line) => console.log(line),
         *     onFinish: (code) => console.log("Finished:", code),
         *   }
         * );
         * ```
         */
        execute_terminal,
        /**
         * Executes a command on a given Server / terminal,
         * and returns a stream to process the output as it comes in.
         *
         * Note. The final line of the stream will usually be
         * `__KOMODO_EXIT_CODE__:0`. The number
         * is the exit code of the command.
         *
         * If this line is NOT present, it means the stream
         * was terminated early, ie like running `exit`.
         *
         * ```ts
         * const stream = await komodo.execute_terminal_stream({
         *   server: "my-server",
         *   terminal: "name",
         *   command: 'for i in {1..3}; do echo "$i"; sleep 1; done',
         * });
         *
         * for await (const line of stream) {
         *   console.log(line);
         * }
         * ```
         */
        execute_terminal_stream,
        /**
         * Subscribes to container exec io over websocket message,
         * for use with xtermjs. Can connect to container on a Server,
         * or associated with a Deployment or Stack.
         * Terminal permission on connecting resource required.
         */
        connect_exec,
        /**
         * Subscribes to container exec io over websocket message,
         * for use with xtermjs. Can connect to Container on a Server.
         * Server Terminal permission required.
         */
        connect_container_exec,
        /**
         * Executes a command on a given container,
         * and gives a callback to handle the output as it comes in.
         *
         * ```ts
         * await komodo.execute_container_exec(
         *   {
         *     server: "my-server",
         *     container: "name",
         *     shell: "bash",
         *     command: 'for i in {1..3}; do echo "$i"; sleep 1; done',
         *   },
         *   {
         *     onLine: (line) => console.log(line),
         *     onFinish: (code) => console.log("Finished:", code),
         *   }
         * );
         * ```
         */
        execute_container_exec,
        /**
         * Executes a command on a given container,
         * and returns a stream to process the output as it comes in.
         *
         * Note. The final line of the stream will usually be
         * `__KOMODO_EXIT_CODE__:0`. The number
         * is the exit code of the command.
         *
         * If this line is NOT present, it means the stream
         * was terminated early, ie like running `exit`.
         *
         * ```ts
         * const stream = await komodo.execute_container_exec_stream({
         *   server: "my-server",
         *   container: "name",
         *   shell: "bash",
         *   command: 'for i in {1..3}; do echo "$i"; sleep 1; done',
         * });
         *
         * for await (const line of stream) {
         *   console.log(line);
         * }
         * ```
         */
        execute_container_exec_stream,
        /**
         * Subscribes to deployment container exec io over websocket message,
         * for use with xtermjs. Can connect to Deployment container.
         * Deployment Terminal permission required.
         */
        connect_deployment_exec,
        /**
         * Executes a command on a given deployment container,
         * and gives a callback to handle the output as it comes in.
         *
         * ```ts
         * await komodo.execute_deployment_exec(
         *   {
         *     deployment: "my-deployment",
         *     shell: "bash",
         *     command: 'for i in {1..3}; do echo "$i"; sleep 1; done',
         *   },
         *   {
         *     onLine: (line) => console.log(line),
         *     onFinish: (code) => console.log("Finished:", code),
         *   }
         * );
         * ```
         */
        execute_deployment_exec,
        /**
         * Executes a command on a given deployment container,
         * and returns a stream to process the output as it comes in.
         *
         * Note. The final line of the stream will usually be
         * `__KOMODO_EXIT_CODE__:0`. The number
         * is the exit code of the command.
         *
         * If this line is NOT present, it means the stream
         * was terminated early, ie like running `exit`.
         *
         * ```ts
         * const stream = await komodo.execute_deployment_exec_stream({
         *   deployment: "my-deployment",
         *   shell: "bash",
         *   command: 'for i in {1..3}; do echo "$i"; sleep 1; done',
         * });
         *
         * for await (const line of stream) {
         *   console.log(line);
         * }
         * ```
         */
        execute_deployment_exec_stream,
        /**
         * Subscribes to container exec io over websocket message,
         * for use with xtermjs. Can connect to Stack service container.
         * Stack Terminal permission required.
         */
        connect_stack_exec,
        /**
         * Executes a command on a given stack service container,
         * and gives a callback to handle the output as it comes in.
         *
         * ```ts
         * await komodo.execute_stack_exec(
         *   {
         *     stack: "my-stack",
         *     service: "database"
         *     shell: "bash",
         *     command: 'for i in {1..3}; do echo "$i"; sleep 1; done',
         *   },
         *   {
         *     onLine: (line) => console.log(line),
         *     onFinish: (code) => console.log("Finished:", code),
         *   }
         * );
         * ```
         */
        execute_stack_exec,
        /**
         * Executes a command on a given stack service container,
         * and returns a stream to process the output as it comes in.
         *
         * Note. The final line of the stream will usually be
         * `__KOMODO_EXIT_CODE__:0`. The number
         * is the exit code of the command.
         *
         * If this line is NOT present, it means the stream
         * was terminated early, ie like running `exit`.
         *
         * ```ts
         * const stream = await komodo.execute_stack_exec_stream({
         *   stack: "my-stack",
         *   service: "service1",
         *   shell: "bash",
         *   command: 'for i in {1..3}; do echo "$i"; sleep 1; done',
         * });
         *
         * for await (const line of stream) {
         *   console.log(line);
         * }
         * ```
         */
        execute_stack_exec_stream,
    };
}