accounted/lib/onboarding-journey/reducer.ts

import type { CompanySettings, EntityType, MomsPeriod } from '@/types'
import type { CompanyLookupResult } from '@/lib/company-lookup/types'
import type { CompanyLookupOutcome } from '@/lib/company-lookup/fetch-company-lookup'
import { mapEntityType } from '@/lib/company-lookup/entity-type-map'
import { deriveSwedishVatNumber } from '@/lib/vat/vat-number'

/**
 * Pure state machine for the journey onboarding
 * (dev_docs/onboarding_migration_plan.md). The component renders `step`,
 * dispatches actions, and performs the side effects (the single TIC lookup
 * via fetchCompanyLookup, the createCompanyFromOnboarding call); the reducer
 * owns every transition and every settings write.
 *
 * Invariants encoded here:
 * - `settings` accumulates the exact CompanySettings partial today's wizard
 *   sends to createCompanyFromOnboarding: nothing less.
 * - `lookupRan` is true only when TIC answered with data for the CURRENT
 *   orgnr. Facts (name/address/F-skatt/moms/räkenskapsår) may be presented
 *   as facts only then; otherwise they are asked as questions. BankID
 *   prefill without a successful lookup is the degraded ask-questions path.
 * - `vat_registered` is never silently defaulted: it is either lookup data
 *   (registration.vat === true) or an explicit answer (ML 17 kap 24 §).
 * - History stores each step's ENTRY snapshot, so Back rolls both answers
 *   and stations to how they were when the step began.
 */

export type JourneyStep =
  | 'orgnr'
  | 'notfound'
  | 'ceased'
  | 'form'
  | 'name'
  | 'address'
  | 'fskatt'
  | 'fy'
  | 'fymonth'
  | 'fystart'
  | 'fyend'
  | 'momsyn'
  | 'moms'
  | 'method'
  | 'done'

export type JourneyStation = 0 | 1 | 2 | 3 | 4

const STATION_OF: Record<JourneyStep, JourneyStation> = {
  orgnr: 0,
  notfound: 0,
  ceased: 0,
  form: 0,
  name: 0,
  address: 0,
  fskatt: 0,
  fy: 1,
  fymonth: 1,
  fystart: 1,
  fyend: 1,
  momsyn: 2,
  moms: 2,
  method: 3,
  done: 4,
}

export function stationOfStep(step: JourneyStep): JourneyStation {
  return STATION_OF[step]
}

export type JourneyServerError =
  | 'org_number_invalid'
  | 'period_invalid'
  | 'generic'
  | null

/** The slice of state a step's entry snapshot preserves for Back. */
interface JourneySnapshot {
  step: JourneyStep
  settings: Partial<CompanySettings>
  ticLookup: CompanyLookupResult | null
  lookupRan: boolean
  lookupNote: 'none' | 'error'
  addressAsked: boolean
  /** EF only: the verksamhetsnamn question was explicitly answered. */
  nameConfirmedForEf: boolean
}

export interface JourneyState extends JourneySnapshot {
  /** Entry snapshot of the CURRENT step (what Back from a later step restores). */
  entry: JourneySnapshot
  history: JourneySnapshot[]
  /** Component fires the lookup while this is true; reducer set on ORG_SUBMITTED. */
  lookupPending: boolean
  /** BankID CompanyRoles prefill present (name/entity trusted without lookup). */
  viaPrefill: boolean
  mode: 'first' | 'add'
  submitting: boolean
  serverError: JourneyServerError
}

export interface JourneyInit {
  mode?: 'first' | 'add'
  /** ?org_number= deep link. The component auto-submits it on mount, which
   *  triggers the same single lookup as manual entry (2026-07-24 addendum:
   *  no preverified suppression in the journey). */
  initialOrgNumber?: string
  initialEntityType?: EntityType
  initialLegalName?: string
}

export type JourneyAction =
  | { type: 'ORG_SUBMITTED'; orgNumber: string }
  | { type: 'LOOKUP_RESULT'; outcome: CompanyLookupOutcome }
  | { type: 'NOTFOUND_CONTINUE' }
  | { type: 'NOTFOUND_EDIT' }
  | { type: 'CEASED_CONTINUE' }
  | { type: 'CEASED_EDIT' }
  | { type: 'ENTITY_PICKED'; entityType: EntityType }
  | { type: 'NAME_SUBMITTED'; name: string }
  | { type: 'ADDRESS_SUBMITTED'; addressLine1?: string; postalCode?: string; city?: string }
  | { type: 'FSKATT_ANSWERED'; fskatt: boolean }
  | { type: 'FY_CALENDAR_CONFIRMED' }
  | { type: 'FY_OTHER_SELECTED' }
  | { type: 'FY_FIRST_SELECTED' }
  | { type: 'FY_END_MONTH_PICKED'; endMonth: number }
  | { type: 'FY_START_PICKED'; date: string }
  | { type: 'FY_END_PICKED'; date: string }
  | { type: 'VAT_ANSWERED'; registered: boolean }
  | { type: 'MOMS_PERIOD_PICKED'; period: MomsPeriod }
  | { type: 'METHOD_PICKED'; method: 'accrual' | 'cash' }
  | { type: 'SUBMIT_SUCCEEDED' }
  | { type: 'SUBMIT_FAILED'; code: 'org_number_invalid' | 'period_invalid' | 'generic' }
  | { type: 'BACK' }
  | { type: 'STATION_JUMP'; station: 0 | 1 | 2 | 3 }

function snapshotOf(s: JourneySnapshot): JourneySnapshot {
  return {
    step: s.step,
    settings: s.settings,
    ticLookup: s.ticLookup,
    lookupRan: s.lookupRan,
    lookupNote: s.lookupNote,
    addressAsked: s.addressAsked,
    nameConfirmedForEf: s.nameConfirmedForEf,
  }
}

export function initJourney(init: JourneyInit = {}): JourneyState {
  const settings: Partial<CompanySettings> = {}
  if (init.initialOrgNumber) settings.org_number = init.initialOrgNumber
  if (init.initialEntityType) settings.entity_type = init.initialEntityType
  if (init.initialLegalName) settings.company_name = init.initialLegalName
  const base: JourneySnapshot = {
    step: 'orgnr',
    settings,
    ticLookup: null,
    lookupRan: false,
    lookupNote: 'none',
    addressAsked: false,
    nameConfirmedForEf: false,
  }
  return {
    ...base,
    entry: snapshotOf(base),
    history: [],
    lookupPending: false,
    viaPrefill: Boolean(init.initialOrgNumber && (init.initialEntityType || init.initialLegalName)),
    mode: init.mode ?? 'first',
    submitting: false,
    serverError: null,
  }
}

/** Transition to `next`, pushing the current step's entry snapshot.
 *  `patch` wins over the defaults (a transition may carry a serverError). */
function go(state: JourneyState, next: JourneyStep, patch?: Partial<JourneyState>): JourneyState {
  const moved: JourneyState = {
    ...state,
    lookupPending: false,
    serverError: null,
    ...patch,
    step: next,
    history: [...state.history, state.entry],
  }
  return { ...moved, entry: snapshotOf(moved) }
}

/** Update within the current step (no history push, entry unchanged). */
function stay(state: JourneyState, patch: Partial<JourneyState>): JourneyState {
  return { ...state, ...patch }
}

/**
 * The Företaget station asks only what is still unknown, then hands over to
 * the fiscal-year station. Order: name → address → F-skatt.
 * - AB with a known company_name (lookup or BankID roles) skips the name
 *   question; EF always confirms the verksamhetsnamn (it defaults to the
 *   person's name but is freely choosable, same as the wizard).
 * - Address is asked only when the lookup did not provide one.
 * - F-skatt is asked whenever it is not lookup data.
 */
function nextCompanyStep(state: JourneyState): JourneyStep {
  const s = state.settings
  const nameKnown =
    s.entity_type === 'aktiebolag'
      ? Boolean(s.company_name)
      : Boolean(s.company_name) && state.nameConfirmedForEf === true
  if (!nameKnown) return 'name'
  if (!state.lookupRan && !state.addressAsked) return 'address'
  if (s.f_skatt === undefined) return 'fskatt'
  return 'fy'
}

/** After the fiscal-year station: skip the moms question only when the
 *  lookup POSITIVELY says the company is VAT registered. A negative or
 *  missing registration is always asked (never defaulted). */
function afterFiscalYear(state: JourneyState): JourneyState {
  if (state.lookupRan && state.ticLookup?.registration.vat === true) {
    const settings = {
      ...state.settings,
      vat_registered: true,
      vat_number: deriveSwedishVatNumber(state.settings.org_number),
    }
    return go(stay(state, { settings }), 'moms')
  }
  return go(state, 'momsyn')
}

/** Downstream answers invalidated by an entity-type change. */
function wipeDownstream(settings: Partial<CompanySettings>): Partial<CompanySettings> {
  const next = { ...settings }
  delete next.fiscal_year_start_month
  delete next.is_first_fiscal_year
  delete next.first_year_start
  delete next.first_year_end
  delete next.vat_registered
  delete next.vat_number
  delete next.moms_period
  delete next.accounting_method
  return next
}

export function journeyReducer(state: JourneyState, action: JourneyAction): JourneyState {
  switch (action.type) {
    case 'ORG_SUBMITTED': {
      if (state.submitting) return state
      // Fresh orgnr invalidates any previous lookup facts.
      return stay(state, {
        settings: { ...state.settings, org_number: action.orgNumber },
        ticLookup: null,
        lookupRan: false,
        lookupNote: 'none',
        lookupPending: true,
        serverError: null,
      })
    }

    case 'LOOKUP_RESULT': {
      if (!state.lookupPending) return state
      const cleared = stay(state, { lookupPending: false })
      const outcome = action.outcome

      if (outcome.status === 'aborted') return cleared

      if (outcome.status === 'found') {
        const lookup = outcome.result
        const mapped = mapEntityType(lookup.legalEntityType)
        const settings: Partial<CompanySettings> = {
          ...state.settings,
          entity_type: mapped ?? state.settings.entity_type,
          company_name: lookup.companyName || state.settings.company_name,
          address_line1: lookup.address?.street ?? state.settings.address_line1,
          postal_code: lookup.address?.postalCode ?? state.settings.postal_code,
          city: lookup.address?.city ?? state.settings.city,
          f_skatt: lookup.registration.fTax,
        }
        const enriched = stay(cleared, {
          settings,
          ticLookup: lookup,
          lookupRan: true,
          lookupNote: 'none' as const,
        })
        if (lookup.isCeased) return go(enriched, 'ceased')
        if (!settings.entity_type) return go(enriched, 'form')
        return go(enriched, nextCompanyStep(enriched))
      }

      if (outcome.status === 'not_found') {
        return go(cleared, 'notfound')
      }

      // disabled: silent manual path. error: manual path + advisory note.
      const noted = stay(cleared, {
        lookupNote: outcome.status === 'error' ? ('error' as const) : ('none' as const),
      })
      if (noted.settings.entity_type) {
        // BankID prefill (or re-run after entity known): degraded ask-
        // questions path; entity/name from CompanyRoles survive as prefill.
        return go(noted, nextCompanyStep(noted))
      }
      return go(noted, 'form')
    }

    case 'NOTFOUND_CONTINUE': {
      if (state.settings.entity_type) return go(state, nextCompanyStep(state))
      return go(state, 'form')
    }

    case 'NOTFOUND_EDIT':
    case 'CEASED_EDIT': {
      // Back to the orgnr question; the fresh submit re-runs the single lookup.
      return go(state, 'orgnr', {
        settings: { ...state.settings, org_number: undefined },
        ticLookup: null,
        lookupRan: false,
        lookupNote: 'none',
      })
    }

    case 'CEASED_CONTINUE': {
      // Proceed with the (ceased) lookup facts: same as wizard, which lets
      // the user continue after the inline warning.
      if (!state.settings.entity_type) return go(state, 'form')
      return go(state, nextCompanyStep(state))
    }

    case 'ENTITY_PICKED': {
      const prev = state.settings.entity_type
      let settings: Partial<CompanySettings> = { ...state.settings, entity_type: action.entityType }
      let next = state
      if (prev && prev !== action.entityType) {
        // Same guard as the wizard's step-1 wipe, but broader per the plan:
        // a changed entity invalidates org/name and every downstream answer.
        settings = wipeDownstream({
          ...settings,
          org_number: undefined,
          company_name: undefined,
        })
        next = stay(state, { ticLookup: null, lookupRan: false, nameConfirmedForEf: false })
        return go(stay(next, { settings }), 'orgnr')
      }
      return go(stay(next, { settings }), nextCompanyStep(stay(next, { settings })))
    }

    case 'NAME_SUBMITTED': {
      const trimmed = action.name.trim()
      if (!trimmed) return state
      const patched = stay(state, {
        settings: { ...state.settings, company_name: trimmed },
        nameConfirmedForEf: true,
      })
      return go(patched, nextCompanyStep(patched))
    }

    case 'ADDRESS_SUBMITTED': {
      const patched = stay(state, {
        settings: {
          ...state.settings,
          address_line1: action.addressLine1 || state.settings.address_line1,
          postal_code: action.postalCode || state.settings.postal_code,
          city: action.city || state.settings.city,
        },
        addressAsked: true,
      })
      return go(patched, nextCompanyStep(patched))
    }

    case 'FSKATT_ANSWERED': {
      const patched = stay(state, {
        settings: { ...state.settings, f_skatt: action.fskatt },
      })
      return go(patched, nextCompanyStep(patched))
    }

    case 'FY_CALENDAR_CONFIRMED': {
      const patched = stay(state, {
        settings: {
          ...state.settings,
          fiscal_year_start_month: 1,
          is_first_fiscal_year: false,
          first_year_start: undefined,
          first_year_end: undefined,
        },
      })
      return afterFiscalYear(patched)
    }

    case 'FY_OTHER_SELECTED':
      return go(state, 'fymonth')

    case 'FY_FIRST_SELECTED':
      return go(state, 'fystart')

    case 'FY_END_MONTH_PICKED': {
      const m = action.endMonth
      if (!Number.isInteger(m) || m < 1 || m > 12) return state
      const patched = stay(state, {
        settings: {
          ...state.settings,
          fiscal_year_start_month: m === 12 ? 1 : m + 1,
          is_first_fiscal_year: false,
          first_year_start: undefined,
          first_year_end: undefined,
        },
      })
      return afterFiscalYear(patched)
    }

    case 'FY_START_PICKED': {
      const patched = stay(state, {
        settings: {
          ...state.settings,
          is_first_fiscal_year: true,
          first_year_start: action.date,
        },
      })
      return go(patched, 'fyend')
    }

    case 'FY_END_PICKED': {
      const endMonth = Number(action.date.split('-')[1])
      const patched = stay(state, {
        settings: {
          ...state.settings,
          is_first_fiscal_year: true,
          first_year_end: action.date,
          // The ongoing fiscal year starts the month after the first year
          // ends: same derivation as the wizard's Step 3.
          fiscal_year_start_month:
            Number.isInteger(endMonth) && endMonth >= 1 && endMonth <= 12
              ? endMonth === 12
                ? 1
                : endMonth + 1
              : 1,
        },
      })
      return afterFiscalYear(patched)
    }

    case 'VAT_ANSWERED': {
      if (action.registered) {
        const patched = stay(state, {
          settings: {
            ...state.settings,
            vat_registered: true,
            vat_number: deriveSwedishVatNumber(state.settings.org_number),
          },
        })
        return go(patched, 'moms')
      }
      const patched = stay(state, {
        settings: {
          ...state.settings,
          vat_registered: false,
          vat_number: null,
          moms_period: null,
        },
      })
      return go(patched, 'method')
    }

    case 'MOMS_PERIOD_PICKED': {
      const patched = stay(state, {
        settings: { ...state.settings, moms_period: action.period },
      })
      return go(patched, 'method')
    }

    case 'METHOD_PICKED': {
      return stay(state, {
        settings: { ...state.settings, accounting_method: action.method },
        submitting: true,
        serverError: null,
      })
    }

    case 'SUBMIT_SUCCEEDED':
      return go(stay(state, { submitting: false }), 'done')

    case 'SUBMIT_FAILED': {
      const cleared = stay(state, { submitting: false })
      if (action.code === 'org_number_invalid') {
        // The server rejected the orgnr: travel back to the Företaget
        // station. Answers are kept; the fresh orgnr re-runs the lookup and
        // the flow walks forward again.
        return go(cleared, 'orgnr', {
          ticLookup: null,
          lookupRan: false,
          serverError: 'org_number_invalid',
        })
      }
      if (action.code === 'period_invalid') {
        return go(cleared, 'fy', { serverError: 'period_invalid' })
      }
      return stay(cleared, { serverError: 'generic' })
    }

    case 'BACK': {
      if (state.submitting || state.history.length === 0) return state
      const history = [...state.history]
      const snap = history.pop() as JourneySnapshot
      return {
        ...state,
        ...snap,
        entry: snap,
        history,
        lookupPending: false,
        submitting: false,
        serverError: null,
      }
    }

    case 'STATION_JUMP': {
      if (state.submitting) return state
      const target = action.station
      if (stationOfStep(state.step) <= target) return state
      const history = [...state.history]
      let snap: JourneySnapshot | null = null
      while (history.length > 0) {
        const top = history[history.length - 1]
        const st = stationOfStep(top.step)
        if (st > target) {
          history.pop()
          continue
        }
        if (st === target) {
          snap = history.pop() as JourneySnapshot
          // Rewind to the station's FIRST step, not its last.
          while (
            history.length > 0 &&
            stationOfStep(history[history.length - 1].step) === target
          ) {
            snap = history.pop() as JourneySnapshot
          }
        }
        break
      }
      if (!snap) return state
      return {
        ...state,
        ...snap,
        entry: snap,
        history,
        lookupPending: false,
        submitting: false,
        serverError: null,
      }
    }

    default:
      return state
  }
}