123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232

import type { HexString } from 'https://deno.land/x/polkadot@0.2.40/util/types.ts';import type { AnyJson, Codec, CodecClass, Inspect, ISet, IU8a, Registry } from '../types/index.ts';
import { compactFromU8aLim, compactToU8a, isHex, isU8a, logger, stringify, u8aConcatStrict, u8aToHex, u8aToU8a } from 'https://deno.land/x/polkadot@0.2.40/util/mod.ts';
import { compareSet, decodeU8aVec, sortSet, typeToConstructor } from '../utils/index.ts';
const l = logger('BTreeSet');
/** @internal */function decodeSetFromU8a<V extends Codec> (registry: Registry, ValClass: CodecClass<V>, u8a: Uint8Array): [CodecClass<V>, Set<V>, number] {  const output = new Set<V>();  const [offset, count] = compactFromU8aLim(u8a);  const result = new Array<V>(count);  const [decodedLength] = decodeU8aVec(registry, result, u8a, offset, ValClass);
  for (let i = 0; i < count; i++) {    output.add(result[i] as unknown as V);  }
  return [ValClass, output, decodedLength];}
/** @internal */function decodeSetFromSet<V extends Codec> (registry: Registry, ValClass: CodecClass<V>, value: Set<any> | string[]): [CodecClass<V>, Set<V>, number] {  const output = new Set<V>();
  value.forEach((val: any) => {    try {      output.add((val instanceof ValClass) ? val : new ValClass(registry, val));    } catch (error) {      l.error('Failed to decode key or value:', (error as Error).message);
      throw error;    }  });
  return [ValClass, output, 0];}
/** * Decode input to pass into constructor. * * @param ValClass - Type of the map value * @param value - Value to decode, one of: * - null * - undefined * - hex * - Uint8Array * - Set<any>, where both key and value types are either *   constructors or decodeable values for their types. * @param jsonSet * @internal */function decodeSet<V extends Codec> (registry: Registry, valType: CodecClass<V> | string, value?: Uint8Array | string | string[] | Set<any>): [CodecClass<V>, Set<V>, number] {  const ValClass = typeToConstructor(registry, valType);
  if (!value) {    return [ValClass, new Set<V>(), 0];  } else if (isU8a(value) || isHex(value)) {    return decodeSetFromU8a<V>(registry, ValClass, u8aToU8a(value));  } else if (Array.isArray(value) || value instanceof Set) {    return decodeSetFromSet<V>(registry, ValClass, value);  }
  throw new Error('BTreeSet: cannot decode type');}
export class BTreeSet<V extends Codec = Codec> extends Set<V> implements ISet<V> {  readonly registry: Registry;
  public createdAtHash?: IU8a;  public initialU8aLength?: number;  public isStorageFallback?: boolean;
  readonly #ValClass: CodecClass<V>;
  constructor (registry: Registry, valType: CodecClass<V> | string, rawValue?: Uint8Array | string | string[] | Set<any>) {    const [ValClass, values, decodedLength] = decodeSet(registry, valType, rawValue);
    super(sortSet(values));
    this.registry = registry;    this.initialU8aLength = decodedLength;    this.#ValClass = ValClass;  }
  public static with<V extends Codec> (valType: CodecClass<V> | string): CodecClass<BTreeSet<V>> {    return class extends BTreeSet<V> {      constructor (registry: Registry, value?: Uint8Array | string | Set<any>) {        super(registry, valType, value);      }    };  }
  /**   * @description The length of the value when encoded as a Uint8Array   */  public get encodedLength (): number {    let len = compactToU8a(this.size).length;
    for (const v of this.values()) {      len += v.encodedLength;    }
    return len;  }
  /**   * @description Returns a hash of the value   */  public get hash (): IU8a {    return this.registry.hash(this.toU8a());  }
  /**   * @description Checks if the value is an empty value   */  public get isEmpty (): boolean {    return this.size === 0;  }
  /**   * @description The actual set values as a string[]   */  public get strings (): string[] {    return [...super.values()].map((v) => v.toString());  }
  /**   * @description Compares the value of the input to see if there is a match   */  public eq (other?: unknown): boolean {    return compareSet(this, other);  }
  /**   * @description Returns a breakdown of the hex encoding for this Codec   */  public inspect (): Inspect {    const inner: Inspect[] = [];
    for (const v of this.values()) {      inner.push(v.inspect());    }
    return {      inner,      outer: [compactToU8a(this.size)]    };  }
  /**   * @description Returns a hex string representation of the value. isLe returns a LE (number-only) representation   */  public toHex (): HexString {    return u8aToHex(this.toU8a());  }
  /**   * @description Converts the Object to to a human-friendly JSON, with additional fields, expansion and formatting of information   */  public toHuman (isExtended?: boolean): AnyJson {    const json: AnyJson = [];
    for (const v of this.values()) {      json.push(v.toHuman(isExtended));    }
    return json;  }
  /**   * @description Converts the Object to JSON, typically used for RPC transfers   */  public toJSON (): AnyJson {    const json: AnyJson = [];
    for (const v of this.values()) {      json.push(v.toJSON());    }
    return json;  }
  /**   * @description Returns the base runtime type name for this instance   */  public toRawType (): string {    return `BTreeSet<${this.registry.getClassName(this.#ValClass) || new this.#ValClass(this.registry).toRawType()}>`;  }
  /**   * @description Converts the value in a best-fit primitive form   */  public toPrimitive (): AnyJson {    const json: AnyJson = [];
    for (const v of this.values()) {      json.push(v.toPrimitive());    }
    return json;  }
  /**   * @description Returns the string representation of the value   */  public override toString (): string {    return stringify(this.toJSON());  }
  /**   * @description Encodes the value as a Uint8Array as per the SCALE specifications   * @param isBare true when the value has none of the type-specific prefixes (internal)   */  public toU8a (isBare?: boolean): Uint8Array {    const encoded: Uint8Array[] = [];
    if (!isBare) {      encoded.push(compactToU8a(this.size));    }
    for (const v of this.values()) {      encoded.push(v.toU8a(isBare));    }
    return u8aConcatStrict(encoded);  }}