solana-program-examples/tokens/pda-mint-authority/steel/tests/instructions.ts

import { BN } from 'bn.js';

class Assignable {
  constructor(properties) {
    for (const [key, value] of Object.entries(properties)) {
      this[key] = value;
    }
  }
}

// Helper function to pad strings to fixed length buffers
function strToBytes(str: string, length: number): Buffer {
  const buffer = Buffer.alloc(length);
  buffer.write(str);
  return buffer;
}

export enum PDAMintAuthorityInstruction {
  Init = 0,
  Create = 1,
  Mint = 2,
}

export class InitArgs {
  instruction: number;

  constructor() {
    this.instruction = PDAMintAuthorityInstruction.Init;
  }

  toBuffer(): Buffer {
    // Only need 1 byte for the instruction as there are no other fields
    const buffer = Buffer.alloc(1);
    buffer.writeUInt8(this.instruction, 0);
    return buffer;
  }
}

export class CreateTokenArgs {
  instruction: number;
  name: Buffer;
  symbol: Buffer;
  uri: Buffer;

  constructor(name: string, symbol: string, uri: string) {
    this.instruction = PDAMintAuthorityInstruction.Create;
    this.name = strToBytes(name, 32);
    this.symbol = strToBytes(symbol, 8);
    this.uri = strToBytes(uri, 64);
  }

  toBuffer(): Buffer {
    const buffer = Buffer.alloc(1 + 32 + 8 + 64);
    let offset = 0;

    buffer.writeUInt8(this.instruction, offset);
    offset += 1;

    this.name.copy(buffer, offset);
    offset += 32;
    this.symbol.copy(buffer, offset);
    offset += 8;
    this.uri.copy(buffer, offset);

    return buffer;
  }
}

export class MintToArgs {
  instruction: number;
  quantity: BN;

  constructor(quantity: number) {
    this.instruction = PDAMintAuthorityInstruction.Mint;
    this.quantity = new BN(quantity);
  }

  toBuffer(): Buffer {
    const buffer = Buffer.alloc(9); // 1 byte for instruction + 8 bytes for u64 quantity
    let offset = 0;

    // Write instruction
    buffer.writeUInt8(this.instruction, offset);
    offset += 1;

    // Write quantity as u64 LE (8 bytes)
    this.quantity.toBuffer('le', 8).copy(buffer, offset);

    return buffer;
  }
}