VaultUI/src/pages/Transit/TransitEncrypt.ts

import { CopyableModal } from "../../elements/CopyableModal";
import { FileUploadInput } from "../../elements/FileUploadInput";
import { Margin } from "../../elements/Margin";
import { Page } from "../../types/Page";
import { changePage, setErrorText, setPageContent, setTitleElement } from "../../pageUtils";
import { fileToBase64, makeElement } from "../../htmlUtils";
import { pageState } from "../../globalPageState";
import { transitEncrypt } from "../../api/transit/transitEncrypt";
import UIkit from 'uikit/dist/js/uikit.min.js';
import i18next from "i18next";


export class TransitEncryptPage extends Page {
  constructor() {
    super();
  }

  goBack(): void {
    changePage("TRANSIT_VIEW_SECRET");
  }

  transitEncryptForm: HTMLFormElement;


  async render(): Promise<void> {
    setTitleElement(pageState);
    setPageContent(makeElement({
      tag: "div"
    }));
    this.transitEncryptForm = makeElement({
      tag: "form",
      children: [
        Margin(makeElement({
          tag: "textarea",
          class: ["uk-textarea", "uk-form-width-medium"],
          attributes: {
            placeholder: i18next.t("transit_encrypt_input_placeholder"),
            name: "plaintext",
          }
        })),
        Margin(FileUploadInput("plaintext_file")),
        Margin([
          makeElement({
            tag: "div",
            class: "uk-form-label",
            text: i18next.t("transit_encrypt_already_encoded_checkbox"),
          }),
          makeElement({
            tag: "div",
            class: ["uk-form-controls", "uk-form-controls-text"],
            children: makeElement({
              tag: "input",
              class: "uk-checkbox",
              attributes: {
                type: "checkbox",
                name: "base64Checkbox",
              }
            }),
          }),
        ]),
        makeElement({
          tag: "p",
          id: "errorText",
          class: "uk-text-danger"
        }),
        makeElement({
          tag: "button",
          class: ["uk-button", "uk-button-primary"],
          text: i18next.t("transit_encrypt_encrypt_btn"),
          attributes: {
            type: "submit",
          }
        })
      ]
    }) as HTMLFormElement;
    setPageContent(this.transitEncryptForm);

    this.transitEncryptForm.addEventListener("submit", async function (e) {
      e.preventDefault();
      await this.transitEncryptFormHandler();
    }.bind(this));
  }

  async transitEncryptFormHandler(): Promise<void> {
    const formData = new FormData(this.transitEncryptForm);

    const base64Checkbox = formData.get("base64Checkbox") as string;

    let plaintext = formData.get("plaintext") as string;

    const plaintext_file = formData.get("plaintext_file") as File;
    if (plaintext_file.size > 0) {
      plaintext = (await fileToBase64(plaintext_file) as string).replace("data:text/plain;base64,", "");
      plaintext = base64Checkbox == "on" ? atob(plaintext) : plaintext;
    } else {
      plaintext = base64Checkbox == "on" ? plaintext : btoa(plaintext);
    }

    try {
      let res = await transitEncrypt(
        pageState.currentBaseMount,
        pageState.currentSecret,
        { plaintext: plaintext }
      );
      const modal = CopyableModal(i18next.t("transit_encrypt_encryption_result_modal_title"), res.ciphertext);
      document.body.querySelector("#pageContent").appendChild(modal);
      UIkit.modal(modal).show();
    } catch (e) {
      setErrorText(`API Error: ${e.message}`);
    }
  }

  get titleSuffix(): string {
    return i18next.t("transit_encrypt_suffix");
  }

  get name(): string {
    return i18next.t("transit_encrypt_title");
  }
}