import { AString, AUint8Array } from '../common';

declare namespace __AspectTransientStorageApi__ {
  function get(aspectId: i32, key: i32): i32;

  function set(key: i32, value: i32): void;
}

export class AspectTransientStorageApi {
  private static _instance: AspectTransientStorageApi | null;

  private constructor() {}

  public static instance(): AspectTransientStorageApi {
    if (!this._instance) {
      this._instance = new AspectTransientStorageApi();
    }
    return this._instance!;
  }

  public get(key: string, aspectId: Uint8Array = new Uint8Array(0)): Uint8Array {
    const inputAspectId = new AUint8Array();
    inputAspectId.set(aspectId);
    const aspectIdPtr = inputAspectId.store();
    const inputKey = new AString();
    inputKey.set(key);
    const inPtr = inputKey.store();
    const ret = __AspectTransientStorageApi__.get(aspectIdPtr, inPtr);
    const bytes = new AUint8Array();
    bytes.load(ret);
    return bytes.get();
  }

  public set(key: string, value: Uint8Array): void {
    const inputKey = new AString();
    inputKey.set(key);
    const keyPtr = inputKey.store();
    const inputValue = new AUint8Array();
    inputValue.set(value);
    const valPtr = inputValue.store();

    __AspectTransientStorageApi__.set(keyPtr, valPtr);
  }
}