import moment from 'moment'
import FileSaver from 'file-saver'
import types from '../types'

const liveInitialState = {
  loading: false,
  error: null,
  opt: {},
  checkpoints: [],
  epochs: ['latest'],
  sequences: [],
  fps: 0,
  playing: false,
  frame: { i: 0, sequence_i: 0, sequence_len: '1' }
}

const liveReducer = (state = liveInitialState, action) => {
  let results;

  switch(action.type) {
    case types.player.load_params:
      if (! action.opt || ! Object.keys(action.opt).length) {
        return state
      }
      return {
        ...state,
        loading: false,
        error: null,
        opt: action.opt,
      }

    case types.player.set_param:
      return {
        ...state,
        opt: {
          ...state.opt,
          [action.key]: action.value,
        }
      }

    case types.player.list_checkpoints:
      return {
        ...state,
        checkpoints: action.checkpoints,
        epochs: [],
      }

    case types.player.list_epochs:
      return {
        ...state,
        epochs: (action.epochs || []).map(a => [ a == 'latest' ? Infinity : a, a ])
                      .sort((a,b) => a[0] - b[0])
                      .map(a => a[1])
    }

    case types.player.list_sequences:
      return {
        ...state,
        sequences: action.sequences,
      }

    case types.player.set_fps:
      return {
        ...state,
        fps: action.fps,
      }

    case types.player.current_frame:
      return action.meta ? {
        ...state,
        frame: action.meta
      } : state

    case types.player.pausing:
      return {
        ...state,
        playing: false,
      }
    
    case types.player.playing:
      return {
        ...state,
        playing: true,
      }

    case types.player.start_recording:
      return {
        ...state,
        opt: {
          ...state.opt,
          recording: true,
        }
      }
    case types.player.add_record_frame:
      return {
        ...state,
        opt: {
          ...state.opt,
          recordFrames: (state.opt.recordFrames || 0) + 1,
        }
      }
    case types.player.save_frame:
      FileSaver.saveAs(
        action.blob,
        state.opt.checkpoint_name + "_" +
        state.opt.sequence + "_" +
        moment().format("YYYYMMDD_HHmm") + ".png"
      )
      return state

    case types.player.saving_video:
      return {
        ...state,
        opt: {
          ...state.opt,
          savingVideo: true,
        }
      }
    case types.player.save_video:
      FileSaver.saveAs(
        action.blob,
        state.opt.checkpoint_name + "_" +
        state.opt.sequence + "_" +
        moment().format("YYYYMMDD_HHmm") + ".webm"
      )
      return {
        ...state,
        opt: {
          ...state.opt,
          recording: false,
          savingVideo: false,
          recordFrames: 0,
        }
      }

    default:
      return state
  }
}

export default liveReducer