import { type BerReader, type BerWriter } from '../ber/index.js';
import { type ProtocolOperationValues } from '../ProtocolOperation.js';
import { ProtocolOperation } from '../ProtocolOperation.js';

import { type MessageOptions } from './Message.js';
import { Message } from './Message.js';

export interface ExtendedRequestMessageOptions extends MessageOptions {
  oid?: string;
  value?: Buffer | string;
}

export class ExtendedRequest extends Message {
  public protocolOperation: ProtocolOperationValues;

  public oid: string;

  public value: Buffer | string;

  public constructor(options: ExtendedRequestMessageOptions) {
    super(options);
    this.protocolOperation = ProtocolOperation.LDAP_REQ_EXTENSION;

    this.oid = options.oid ?? '';
    this.value = options.value ?? '';
  }

  public override writeMessage(writer: BerWriter): void {
    writer.writeString(this.oid, 0x80);
    if (Buffer.isBuffer(this.value)) {
      writer.writeBuffer(this.value, 0x81);
    } else if (this.value) {
      writer.writeString(this.value, 0x81);
    }
  }

  public override parseMessage(reader: BerReader): void {
    this.oid = reader.readString(0x80) ?? '';
    if (reader.peek() === 0x81) {
      try {
        this.value = reader.readString(0x81) ?? '';
      } catch {
        this.value = reader.readString(0x81, true) ?? Buffer.alloc(0);
      }
    }
  }
}