import mongoose from 'mongoose';
import stateMachinePlugin from '@rohan89/mongoose-state-machine';

// schema declaration
const matterSchema = new mongoose.Schema({ 
  matterState: String // field for storing state 
});

// state machine declaration
const stateMachine = {
  init: 'solid',
  transitions: [
    { name: 'melt', from: 'solid', to: 'liquid' },
    { name: 'freeze', from: 'liquid', to: 'solid' },
    { name: 'vaporize', from: 'liquid', to: 'gas' },
    { name: 'condense', from: 'gas', to: 'liquid' }
  ],
  methods: {
    onMelt() { console.log('I melted') },
    onFreeze() { console.log('I froze') },
    onVaporize() { console.log('I vaporized') },
    onCondense() { console.log('I condensed') }
  }
};

// extend schema with the plugin
matterSchema.plugin(stateMachinePlugin, { 
  fieldName: 'matterState', 
  stateMachine 
});
  
const Matter = mongoose.model('Matter', schema);

// create document
const matter = new Matter();
matter.matterState;   // solid (init state);
matter.is('solid');   // true
matter.is('liquid');  // false
matter.can('melt');   // true
matter.can('freeze'); // false

// transition
matter.melt();        // I melted!
matter.matterState;   // liquid 
await matter.save();  // save state

// retrieving from a database
const found = Matter.findById(matter.id);
found.matterState;  // liquid;
found.vaporize();   // I vaporized!
found.matterState;  // gas
found.melt();       // throw error

matter.matterState = 'solid'; // gas (no effect!)