/**
 * This code was AUTOGENERATED using the Codama library.
 * Please DO NOT EDIT THIS FILE, instead use visitors
 * to add features, then rerun Codama to update it.
 *
 * @see https://github.com/codama-idl/codama
 */

import {
  combineCodec,
  getDiscriminatedUnionDecoder,
  getDiscriminatedUnionEncoder,
  getStructDecoder,
  getStructEncoder,
  getTupleDecoder,
  getTupleEncoder,
  getUnitDecoder,
  getUnitEncoder,
  type Codec,
  type Decoder,
  type Encoder,
  type GetDiscriminatedUnionVariant,
  type GetDiscriminatedUnionVariantContent,
} from '@solana/kit';
import {
  getFactionDecoder,
  getFactionEncoder,
  type Faction,
  type FactionArgs,
} from '@staratlas/dev-profile-faction';

export type RiskZone =
  | { __kind: 'LowRiskZone'; fields: readonly [Faction] }
  | { __kind: 'MediumRiskZone' }
  | { __kind: 'HighRiskZone' };

export type RiskZoneArgs =
  | { __kind: 'LowRiskZone'; fields: readonly [FactionArgs] }
  | { __kind: 'MediumRiskZone' }
  | { __kind: 'HighRiskZone' };

export function getRiskZoneEncoder(): Encoder<RiskZoneArgs> {
  return getDiscriminatedUnionEncoder([
    [
      'LowRiskZone',
      getStructEncoder([['fields', getTupleEncoder([getFactionEncoder()])]]),
    ],
    ['MediumRiskZone', getUnitEncoder()],
    ['HighRiskZone', getUnitEncoder()],
  ]);
}

export function getRiskZoneDecoder(): Decoder<RiskZone> {
  return getDiscriminatedUnionDecoder([
    [
      'LowRiskZone',
      getStructDecoder([['fields', getTupleDecoder([getFactionDecoder()])]]),
    ],
    ['MediumRiskZone', getUnitDecoder()],
    ['HighRiskZone', getUnitDecoder()],
  ]);
}

export function getRiskZoneCodec(): Codec<RiskZoneArgs, RiskZone> {
  return combineCodec(getRiskZoneEncoder(), getRiskZoneDecoder());
}

// Data Enum Helpers.
export function riskZone(
  kind: 'LowRiskZone',
  data: GetDiscriminatedUnionVariantContent<
    RiskZoneArgs,
    '__kind',
    'LowRiskZone'
  >['fields'],
): GetDiscriminatedUnionVariant<RiskZoneArgs, '__kind', 'LowRiskZone'>;
export function riskZone(
  kind: 'MediumRiskZone',
): GetDiscriminatedUnionVariant<RiskZoneArgs, '__kind', 'MediumRiskZone'>;
export function riskZone(
  kind: 'HighRiskZone',
): GetDiscriminatedUnionVariant<RiskZoneArgs, '__kind', 'HighRiskZone'>;
export function riskZone<K extends RiskZoneArgs['__kind'], Data>(
  kind: K,
  data?: Data,
) {
  return Array.isArray(data)
    ? { __kind: kind, fields: data }
    : { __kind: kind, ...(data ?? {}) };
}

export function isRiskZone<K extends RiskZone['__kind']>(
  kind: K,
  value: RiskZone,
): value is RiskZone & { __kind: K } {
  return value.__kind === kind;
}