import { describe, expect, it } from 'vitest';

import { StarRecord } from '../../shared/models/star.model';
import { catalogueCensus, describingCatalogue, starReadouts } from './star-readouts';

/** Three kinds of star the catalogue holds, each as the decoder gives it back. */
const HYG_STAR: StarRecord = {
  id: 32263,
  name: 'Sirius',
  x: -0.49,
  y: 2.47,
  z: -0.75,
  magnitude: -1.44,
  magnitudeBand: 'V',
  spectralType: 'A0m',
  colorIndex: 0.009,
  colorSystem: 'B-V',
  distanceError: 0.004,
  distanceFromGaia: false,
  source: 'hyg'
};
// A star Gaia alone knows, at 117 pc and good to a tenth.
const GAIA_STAR: StarRecord = {
  id: 1000000001,
  name: 'Gaia DR3 5612323414549657984',
  x: 117,
  y: 0,
  z: 0,
  magnitude: 11.2,
  magnitudeBand: 'G',
  spectralType: 'Unknown',
  colorIndex: 1.43,
  colorSystem: 'BP-RP',
  distanceError: 0.1,
  distanceFromGaia: true,
  source: 'gaia'
};
// HYG's description of Proxima, at the place and distance Gaia measures.
const PLACED_BY_GAIA: StarRecord = { ...HYG_STAR, name: 'Proxima Centauri', magnitude: 11.01, colorIndex: 1.807, spectralType: 'M5Ve', distanceFromGaia: true, source: 'gaia' };

function value(readouts: { label: string; value: string }[], label: string): string | undefined {
  return readouts.find((readout) => readout.label === label)?.value;
}

describe('starReadouts', () => {
  it('names the band of the magnitude, and which colour the colour index is', () => {
    expect(value(starReadouts(HYG_STAR, null), 'Magnitude')).toBe('V -1.44');
    expect(value(starReadouts(GAIA_STAR, null), 'Magnitude')).toBe('G 11.20');
    expect(value(starReadouts(HYG_STAR, null), 'Colour')).toBe('B−V 0.01');
    expect(value(starReadouts(GAIA_STAR, null), 'Colour')).toBe('BP−RP 1.43');
  });

  it('says a stand-in magnitude was not measured, and leaves out a colour there is none of', () => {
    const unmeasured = starReadouts({ ...GAIA_STAR, magnitude: 12, magnitudeBand: undefined, colorIndex: null, colorSystem: undefined }, null);
    expect(value(unmeasured, 'Magnitude')).toBe('Not measured');
    expect(value(unmeasured, 'Colour')).toBeUndefined();
  });

  it('gives the distance with its uncertainty', () => {
    expect(value(starReadouts(GAIA_STAR, null), 'Distance')).toBe('117 ± 12 pc');
  });

  it('names the catalogue a star comes from, and whose distance it has', () => {
    expect(value(starReadouts(GAIA_STAR, null), 'Source')).toBe('Gaia DR3');
    expect(value(starReadouts(HYG_STAR, null), 'Source')).toBe('HYG');
    expect(value(starReadouts(PLACED_BY_GAIA, null), 'Source')).toBe('HYG, Gaia DR3 distance');
    expect(value(starReadouts({ ...HYG_STAR, source: 'exoplanet-archive' }, null), 'Source')).toBe('NASA Exoplanet Archive');
  });

  it('marks the luminosity as derived', () => {
    expect(starReadouts(HYG_STAR, 25.4).find((readout) => readout.label === 'Luminosity')).toEqual({ label: 'Luminosity', value: '25.40 L☉', derived: true });
  });
});

describe('catalogueCensus', () => {
  it('counts the stars by the catalogue describing them, not by whose position they have', () => {
    expect(describingCatalogue(PLACED_BY_GAIA)).toBe('HYG');
    expect(catalogueCensus([GAIA_STAR, GAIA_STAR, GAIA_STAR, HYG_STAR, PLACED_BY_GAIA])).toBe('Gaia DR3 3 · HYG 2');
  });
});