import { setup, types } from 'xstate';

const baseDef = setup({
  schemas: {
    actions: {
      track: { params: types<{ key: string }>() }
    },
    guards: {
      hasAccess: { params: types<{ role: string }>() }
    }
  }
})


const extendedDef = baseDef.extend({
  schemas: {
    actions: {
      notify: { params: types<{ message: string }>() }
    },
    guards: {
      canReset: { params: types<{ reason: string }>() }
    }
  }
})

// Both base and extended actions/guards are available
const machine = extendedDef.createMachine({
  entry: ({ actions }, enq) => {
    enq(actions.track({ key: 'init' }));
    enq(actions.notify({ message: 'started' }));
  },
  on: {
    RESET: ({ guards }) => {
      if (
        guards.hasAccess({ role: 'admin' }) &&
        guards.canReset({ reason: 'manual' })
      ) {
        return { target: '.idle' };
      }
    }
  }
});