import { registerSync } from './reconstruct.ts';

export class Uint64Atomic {
  static readonly byteSize = 8;
  static readonly byteAlignment = 8;
  private readonly view: BigUint64Array;

  constructor(buffer?: SharedArrayBuffer, byteOffset?: number) {
    const buf = buffer ?? new SharedArrayBuffer(8);
    const offset = byteOffset ?? 0;
    this.view = new BigUint64Array(buf, offset, 1);
  }

  load(): bigint {
    return Atomics.load(this.view, 0);
  }

  store(value: bigint): void {
    Atomics.store(this.view, 0, value);
  }

  add(value: bigint): bigint {
    return Atomics.add(this.view, 0, value);
  }

  sub(value: bigint): bigint {
    return Atomics.sub(this.view, 0, value);
  }

  and(value: bigint): bigint {
    return Atomics.and(this.view, 0, value);
  }

  or(value: bigint): bigint {
    return Atomics.or(this.view, 0, value);
  }

  xor(value: bigint): bigint {
    return Atomics.xor(this.view, 0, value);
  }

  exchange(value: bigint): bigint {
    return Atomics.exchange(this.view, 0, value);
  }

  compareExchange(expected: bigint, replacement: bigint): bigint {
    return Atomics.compareExchange(this.view, 0, expected, replacement);
  }

  [Symbol.for('moroutine.shared')](): { tag: string; buffer: SharedArrayBuffer; byteOffset: number } {
    return { tag: 'Uint64Atomic', buffer: this.view.buffer as SharedArrayBuffer, byteOffset: this.view.byteOffset };
  }
}

registerSync('Uint64Atomic', Uint64Atomic);