import * as hz from 'horizon/core';

type Props = { initialAmmo: number };
type State = { ammo: number };

class WeaponWithAmmo extends hz.Component<Props, State> {

  static propsDefinition = {
      initialAmmo: { type: hz.PropTypes.Number, default: 20 },
  };



  // local state that would normally be lost on ownership transfer
  private ammo: number = 0;

  start(): void {

    // set local value for ammo
    this.ammo = this.props.initialAmmo;

    // imagine there's some event here that decreases ammo over time

  };

  // this is called on the new owner's instance of the script
  // after it starts up and receives state from the previous owner.
  receiveOwnership(
    state: State | null,
    fromPlayer: hz.Player,
    toPlayer: hz.Player,
  ): void {
    if (state != null) {
      this.ammo = state.ammo;
    };
  };

  // this is called on the previous owner's instance of the script
  // to let it export its local state and send it over to the new owner.
  transferOwnership(
    fromPlayer: hz.Player,
    toPlayer: hz.Player,
  ): State {
    return { ammo: this.ammo };
  };

};

hz.Component.register(WeaponWithAmmo);