@nativewrappers/client/lib/weaponComponent/WeaponComponentCollection.js

Javascript/Typescript wrapper for the FiveM natives

import { WeaponComponent } from './WeaponComponent';
import { InvalidWeaponComponent } from './InvalidWeaponComponent';
import { WeaponComponentHashesByWeaponHash } from './WeaponComponentHashesByWeaponHash';
import { ComponentAttachmentPoint } from './ComponentAttachmentPoint';
import { ComponentAttachmentPointByHash } from './ComponentAttachmentPointByHash';
/**
 * ped weapon components on weapon
 *
 */
export class WeaponComponentCollection {
    constructor(owner, weapon) {
        this.components = new Map();
        this.invalidComponent = new InvalidWeaponComponent();
        this.owner = owner;
        this.weapon = weapon;
    }
    [Symbol.iterator]() {
        let pointer = 0;
        const components = Array.from(this.components.values());
        return {
            next() {
                if (pointer < components.length) {
                    return { done: false, value: components[pointer++] };
                }
                else {
                    return { done: true, value: null };
                }
            },
        };
    }
    /**
     * get component
     *
     * @param componentHash
     */
    get(componentHash) {
        if (this.AllWeaponComponentHashes.some(x => x === componentHash)) {
            let component = this.components.get(componentHash);
            if (!component) {
                component = this.createAndAddComponent(componentHash);
            }
            return component;
        }
        return this.invalidComponent;
    }
    /**
     * get current weapon component count
     *
     * @constructor
     */
    get Count() {
        return this.components.size;
    }
    /**
     * get clip component
     *
     * @param index - index of component
     */
    getClipComponent(index) {
        return this.getAnyComponentByAttachmentPoints(index, ComponentAttachmentPoint.Clip, ComponentAttachmentPoint.Clip2);
    }
    /**
     * get clip variation count
     *
     * @constructor
     */
    get ClipVariationsCount() {
        return this.getComponentHashesByAttachmentPoints(ComponentAttachmentPoint.Clip, ComponentAttachmentPoint.Clip2).length;
    }
    /**
     * get scope component
     *
     * @param index - index of component
     */
    getScopeComponent(index) {
        return this.getAnyComponentByAttachmentPoints(index, ComponentAttachmentPoint.Scope, ComponentAttachmentPoint.Scope2);
    }
    /**
     * get scope variation count
     *
     * @constructor
     */
    get ScopeVariationsCount() {
        return this.getComponentHashesByAttachmentPoints(ComponentAttachmentPoint.Scope, ComponentAttachmentPoint.Scope2).length;
    }
    /**
     * get suppressor component
     *
     */
    getSuppressorComponent() {
        return this.getAnyComponentByAttachmentPoints(undefined, ComponentAttachmentPoint.Supp, ComponentAttachmentPoint.Supp2);
    }
    /**
     * get flash light component
     *
     */
    getFlashLightComponent() {
        return this.getAnyComponentByAttachmentPoints(undefined, ComponentAttachmentPoint.FlashLaser, ComponentAttachmentPoint.FlashLaser2);
    }
    /**
     * get luxury finish component
     *
     */
    getLuxuryFinishComponent() {
        return this.getAnyComponentByAttachmentPoints(undefined, ComponentAttachmentPoint.GunRoot);
    }
    /**
     * get Mk2 camo component
     *
     * @param index - index of component
     */
    getMk2CamoComponent(index) {
        return this.getAnyComponentByAttachmentPoints(index, ComponentAttachmentPoint.GunRoot);
    }
    /**
     * get Mk2 barrel component
     *
     * @param index - index of component
     */
    getMk2BarrelComponent(index) {
        return this.getAnyComponentByAttachmentPoints(index, ComponentAttachmentPoint.Barrel);
    }
    /**
     * Create component object and add to collection
     *
     * @param hash
     * @private
     */
    createAndAddComponent(hash) {
        const uintHash = hash >>> 0;
        console.log('createAndAdd', hash, uintHash);
        console.log('about to create', this.owner, this.weapon, uintHash);
        const component = new WeaponComponent(this.owner, this.weapon, uintHash);
        this.components.set(uintHash, component);
        return component;
    }
    /**
     * get all WeaponComponentHash belongs to weapon
     *
     * @constructor
     * @private
     */
    get AllWeaponComponentHashes() {
        // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
        return WeaponComponentHashesByWeaponHash.get(this.weapon.Hash);
    }
    /**
     * get components belongs to attachmentPoints
     *
     * @param attachmentPoints
     * @private
     */
    getComponentHashesByAttachmentPoints(...attachmentPoints) {
        return this.AllWeaponComponentHashes.filter(hash => attachmentPoints.some(attachmentPoint => ComponentAttachmentPointByHash.get(hash) === attachmentPoint));
    }
    /**
     * get component by index and attachmentPoints
     *
     * @param index - component index
     * @param attachmentPoints -  attachmentPoints to search
     * @private
     */
    getAnyComponentByAttachmentPoints(index, ...attachmentPoints) {
        const hashes = this.getComponentHashesByAttachmentPoints(...attachmentPoints);
        if (index === undefined) {
            return this.get(hashes[0]) ?? this.invalidComponent;
        }
        return 0 <= index && index <= hashes.length - 1
            ? this.get(hashes[index])
            : this.invalidComponent;
    }
}