import {
  ActionName,
  PythGovernanceAction,
  PythGovernanceActionImpl,
  PythGovernanceHeader,
} from "./PythGovernanceAction";
import * as BufferLayout from "@solana/buffer-layout";
import { ChainName } from "../chains";

/**
 * Authorize transferring the governance data source from the sender's emitter address to another emitter.
 * The receiving emitter address is the sender of claimVaa, which must be a RequestGovernanceDataSourceTransfer message.
 */
export class AuthorizeGovernanceDataSourceTransfer
  implements PythGovernanceAction
{
  readonly actionName: ActionName;
  readonly claimVaa: Buffer;

  constructor(readonly targetChainId: ChainName, vaa: Buffer) {
    this.actionName = "AuthorizeGovernanceDataSourceTransfer";
    this.claimVaa = new Buffer(vaa);
  }

  static decode(
    data: Buffer
  ): AuthorizeGovernanceDataSourceTransfer | undefined {
    const header = PythGovernanceHeader.decode(data);
    if (!header || header.action !== "AuthorizeGovernanceDataSourceTransfer") {
      return undefined;
    }

    const payload = data.subarray(PythGovernanceHeader.span, data.length);

    return new AuthorizeGovernanceDataSourceTransfer(
      header.targetChainId,
      payload
    );
  }

  encode(): Buffer {
    const headerBuffer = new PythGovernanceHeader(
      this.targetChainId,
      this.actionName
    ).encode();
    return Buffer.concat([headerBuffer, this.claimVaa]);
  }
}

/**
 * Request a transfer of the governance data source to the emitter of this message.
 */
export class RequestGovernanceDataSourceTransfer extends PythGovernanceActionImpl {
  static layout: BufferLayout.Structure<
    Readonly<{ governanceDataSourceIndex: number }>
  > = BufferLayout.struct([BufferLayout.u32be("governanceDataSourceIndex")]);

  constructor(
    targetChainId: ChainName,
    readonly governanceDataSourceIndex: number
  ) {
    super(targetChainId, "RequestGovernanceDataSourceTransfer");
  }

  static decode(data: Buffer): RequestGovernanceDataSourceTransfer | undefined {
    const decoded = PythGovernanceActionImpl.decodeWithPayload(
      data,
      "RequestGovernanceDataSourceTransfer",
      RequestGovernanceDataSourceTransfer.layout
    );
    if (!decoded) return undefined;

    return new RequestGovernanceDataSourceTransfer(
      decoded[0].targetChainId,
      decoded[1].governanceDataSourceIndex
    );
  }

  encode(): Buffer {
    return super.encodeWithPayload(RequestGovernanceDataSourceTransfer.layout, {
      governanceDataSourceIndex: this.governanceDataSourceIndex,
    });
  }
}