solana-dex-parser/src/instruction-classifier.ts

Solana Dex Transaction Parser

import { SYSTEM_PROGRAMS } from './constants';
import { TransactionAdapter } from './transaction-adapter';
import { ClassifiedInstruction } from './types/common';

export class InstructionClassifier {
  private instructionMap: Map<string, ClassifiedInstruction[]> = new Map();

  constructor(private adapter: TransactionAdapter) {
    this.classifyInstructions();
  }

  private classifyInstructions() {
    // outer instructions
    this.adapter.instructions.forEach((instruction: any, outerIndex: any) => {
      const programId = this.adapter.getInstructionProgramId(instruction);
      this.addInstruction({
        instruction,
        programId,
        outerIndex,
      });
    });

    // innerInstructions
    const innerInstructions = this.adapter.innerInstructions;
    if (innerInstructions) {
      innerInstructions.forEach((set) => {
        set.instructions.forEach((instruction, innerIndex) => {
          const programId = this.adapter.getInstructionProgramId(instruction);
          this.addInstruction({
            instruction,
            programId,
            outerIndex: set.index,
            innerIndex,
          });
        });
      });
    }
  }

  private addInstruction(classified: ClassifiedInstruction) {
    if (!classified.programId) return;

    const instructions = this.instructionMap.get(classified.programId) || [];
    instructions.push(classified);
    this.instructionMap.set(classified.programId, instructions);
  }

  public getInstructions(programId: string): ClassifiedInstruction[] {
    return this.instructionMap.get(programId) || [];
  }

  public getAllProgramIds(): string[] {
    return Array.from(this.instructionMap.keys()).filter((it) => !SYSTEM_PROGRAMS.includes(it));
  }
}