// deno-lint-ignore-file no-explicit-any import { colors, log, setClient } from "./deps.ts"; import * as wmill from "./gen/services.gen.ts"; import { GlobalUserInfo } from "./gen/types.gen.ts"; import { loginInteractive, tryGetLoginInfo } from "./login.ts"; import { GlobalOptions } from "./types.ts"; import { getHeaders } from "./utils.ts"; import { addWorkspace, getActiveWorkspace, getWorkspaceByName, removeWorkspace, Workspace, } from "./workspace.ts"; export type Context = { workspace: string; baseUrl: string; urlStore: string; token: string; }; async function tryResolveWorkspace( opts: GlobalOptions ): Promise< { isError: false; value: Workspace } | { isError: true; error: string } > { const cache = (opts as any).__secret_workspace; if (cache) return { isError: false, value: cache }; if (opts.workspace) { const e = await getWorkspaceByName(opts.workspace); if (!e) { return { isError: true, error: colors.red.underline("Given workspace does not exist."), }; } (opts as any).__secret_workspace = e; return { isError: false, value: e }; } const defaultWorkspace = await getActiveWorkspace(opts); if (!defaultWorkspace) { return { isError: true, error: colors.red.underline("No workspace given and no default set."), }; } return { isError: false, value: defaultWorkspace }; } export async function resolveWorkspace( opts: GlobalOptions ): Promise { if (opts.baseUrl) { if (opts.workspace && opts.token) { return { remote: new URL(opts.baseUrl).toString(), // add trailing slash if not present workspaceId: opts.workspace, name: opts.workspace, token: opts.token, }; } else { log.info( colors.red( "If you specify a base URL with --base-url, you must also specify a workspace (--workspace) and token (--token)." ) ); return Deno.exit(-1); } } const res = await tryResolveWorkspace(opts); if (res.isError) { log.info(colors.red.bold(res.error)); return Deno.exit(-1); } else { return res.value; } } export async function requireLogin( opts: GlobalOptions ): Promise { const workspace = await resolveWorkspace(opts); let token = await tryGetLoginInfo(opts); if (!token) { token = workspace.token; } setClient(token, workspace.remote.substring(0, workspace.remote.length - 1)); try { return await wmill.globalWhoami(); } catch { log.info( "! Could not reach API given existing credentials. Attempting to reauth..." ); const newToken = await loginInteractive(workspace.remote); if (!newToken) { throw new Error("Could not reauth"); } removeWorkspace(workspace.name, false, opts); workspace.token = newToken; addWorkspace(workspace, opts); setClient( token, workspace.remote.substring(0, workspace.remote.length - 1) ); return await wmill.globalWhoami(); } } export async function fetchVersion(baseUrl: string): Promise { const requestHeaders = new Headers(); const extraHeaders = getHeaders(); if (extraHeaders) { for (const [key, value] of Object.entries(extraHeaders)) { // @ts-ignore requestHeaders.set(key, value); } } const response = await fetch( new URL(new URL(baseUrl).origin + "/api/version"), { headers: requestHeaders, method: "GET" } ); return await response.text(); } export async function tryResolveVersion( opts: GlobalOptions ): Promise { if ((opts as any).__cache_version) { return (opts as any).__cache_version; } const workspaceRes = await tryResolveWorkspace(opts); if (workspaceRes.isError) return undefined; const version = await fetchVersion(workspaceRes.value.remote); try { return Number.parseInt( version.split("-", 1)[0].replaceAll(".", "").replace("v", "") ); } catch { return undefined; } } export function validatePath(path: string): boolean { if (!(path.startsWith("g") || path.startsWith("u") || path.startsWith("f"))) { log.info( colors.red( "Given remote path looks invalid. Remote paths are typically of the form //..." ) ); return false; } return true; }