yaak/src-web/lib/encryption.ts

import { parseTemplate } from '@yaakapp-internal/templates';
import { activeEnvironmentIdAtom } from '../hooks/useActiveEnvironment';
import { activeWorkspaceIdAtom } from '../hooks/useActiveWorkspace';
import { jotaiStore } from './jotai';
import { invokeCmd } from './tauri';

export function analyzeTemplate(template: string): 'global_secured' | 'local_secured' | 'insecure' {
  let secureTags = 0;
  let insecureTags = 0;
  let totalTags = 0;
  for (const t of parseTemplate(template).tokens) {
    if (t.type === 'eof') continue;

    totalTags++;
    if (t.type === 'tag' && t.val.type === 'fn' && t.val.name === 'secure') {
      secureTags++;
    } else if (t.type === 'tag' && t.val.type === 'var') {
      // Variables are secure
    } else if (t.type === 'tag' && t.val.type === 'bool') {
      // Booleans are secure
    } else {
      insecureTags++;
    }
  }

  if (secureTags === 1 && totalTags === 1) {
    return 'global_secured';
  }
  if (insecureTags === 0) {
    return 'local_secured';
  }
  return 'insecure';
}

export async function convertTemplateToInsecure(template: string) {
  if (template === '') {
    return '';
  }

  const workspaceId = jotaiStore.get(activeWorkspaceIdAtom) ?? 'n/a';
  const environmentId = jotaiStore.get(activeEnvironmentIdAtom) ?? null;
  return invokeCmd<string>('cmd_decrypt_template', { template, workspaceId, environmentId });
}

export async function convertTemplateToSecure(template: string): Promise<string> {
  if (template === '') {
    return '';
  }

  if (analyzeTemplate(template) === 'global_secured') {
    return template;
  }

  const workspaceId = jotaiStore.get(activeWorkspaceIdAtom) ?? 'n/a';
  const environmentId = jotaiStore.get(activeEnvironmentIdAtom) ?? null;
  return invokeCmd<string>('cmd_secure_template', { template, workspaceId, environmentId });
}