gomu/command.go

package main

import (

	"github.com/rivo/tview"
	"github.com/spf13/viper"
	"github.com/ztrue/tracerr"
)

type Command struct {
	commands map[string]func()
}

func newCommand() Command {
	return Command{
		commands: make(map[string]func()),
	}
}

func (c *Command) define(name string, callBack func()) {
	c.commands[name] = callBack
}

func (c Command) getFn(name string) (func(), error) {
	fn, ok := c.commands[name]
	if !ok {
		return nil, tracerr.New("command not found")
	}
	return fn, nil
}

func (c Command) defineCommands() {

	/* Playlist */

	c.define("create_playlist", func() {
		name, _ := gomu.pages.GetFrontPage()
		if name != "mkdir-popup" {
			createPlaylistPopup()
		}
	})

	c.define("delete_playlist", func() {
		audioFile := gomu.playlist.getCurrentFile()
		err := gomu.playlist.deletePlaylist(audioFile)
		if err != nil {
			logError(err)
		}
	})

	c.define("delete_file", func() {
		audioFile := gomu.playlist.getCurrentFile()
		// prevent from deleting a directory
		if !audioFile.isAudioFile {
			return
		}

		err := gomu.playlist.deleteSong(audioFile)
		if err != nil {
			logError(err)
		}
	})

	c.define("download_audio", func() {

		audioFile := gomu.playlist.getCurrentFile()
		currNode := gomu.playlist.GetCurrentNode()
		if gomu.pages.HasPage("download-popup") {
			gomu.pages.RemovePage("download-popup")
			return
		}
		// this ensures it downloads to
		// the correct dir
		if audioFile.isAudioFile {
			downloadMusicPopup(audioFile.parent)
		} else {
			downloadMusicPopup(currNode)
		}
	})

	c.define("add_queue", func() {
		audioFile := gomu.playlist.getCurrentFile()
		currNode := gomu.playlist.GetCurrentNode()
		if audioFile.isAudioFile {
			gomu.queue.enqueue(audioFile)
		} else {
			currNode.SetExpanded(true)
		}
	})

	c.define("close_node", func() {
		audioFile := gomu.playlist.getCurrentFile()
		currNode := gomu.playlist.GetCurrentNode()
		// if closing node with no children
		// close the node's parent
		// remove the color of the node

		if audioFile.isAudioFile {
			parent := audioFile.parent
			gomu.playlist.setHighlight(parent)
			parent.SetExpanded(false)
		}
		currNode.Collapse()
	})

	c.define("bulk_add", func() {
		currNode := gomu.playlist.GetCurrentNode()

		if !viper.GetBool("general.confirm_bulk_add") {
			gomu.playlist.addAllToQueue(currNode)
			return
		}

		confirmationPopup(
			"Are you sure to add this whole directory into queue?",
			func(_ int, label string) {

				if label == "yes" {
					gomu.playlist.addAllToQueue(currNode)
				}

			})
	})

	c.define("refresh", func() {
		gomu.playlist.refresh()
	})

	c.define("rename", func() {
		audioFile := gomu.playlist.getCurrentFile()
		renamePopup(audioFile)
	})

	c.define("playlist_search", func() {

		if viper.GetBool("general.fzf") {
			err := gomu.playlist.fuzzyFind()
			if err != nil {
				logError(err)
			}
			return
		}

		files := make([]string, len(gomu.playlist.getAudioFiles()))

		for i, file := range gomu.playlist.getAudioFiles() {
			files[i] = file.name
		}

		searchPopup(files, func(text string) {

			audio, err := gomu.playlist.findAudioFile(sha1Hex(text))
			if err != nil {
				logError(err)
			}

			gomu.playlist.setHighlight(audio.node)
		})
	})

	/* Queue */

	c.define("move_down", func() {
		gomu.queue.next()
	})

	c.define("move_up", func() {
		gomu.queue.prev()
	})

	c.define("delete_item", func() {
		gomu.queue.deleteItem(gomu.queue.GetCurrentItem())
	})

	c.define("clear_queue", func() {
		confirmationPopup("Are you sure to clear the queue?",
			func(_ int, label string) {
				if label == "yes" {
					gomu.queue.clearQueue()
				}
			})
	})

	c.define("play_selected", func() {
    if (gomu.queue.GetItemCount() != 0 && gomu.queue.GetCurrentItem()!= -1) {
      a, err := gomu.queue.deleteItem(gomu.queue.GetCurrentItem())
      if err != nil {
        logError(err)
      }

      gomu.queue.pushFront(a)
      gomu.player.skip()
    }
	})

	c.define("toggle_loop", func() {
		gomu.queue.isLoop = gomu.player.toggleLoop()
    gomu.queue.updateTitle()
	})

	c.define("shuffle_queue", func() {
		gomu.queue.shuffle()
	})

	c.define("queue_search", func() {

		queue := gomu.queue

		if viper.GetBool("general.fzf") {
			gomu.suspend()
			if err := queue.fuzzyFind(); err != nil {
				logError(err)
			}
			gomu.unsuspend()
			return
		}

		audios := make([]string, 0, len(queue.items))
		for _, file := range queue.items {
			audios = append(audios, file.name)
		}

		searchPopup(audios, func(selected string) {

			index := 0
			for i, v := range queue.items {
				if v.name == selected {
					index = i
				}
			}

			queue.SetCurrentItem(index)
		})
	})

	/* Global */
	c.define("quit", func() {
		if !viper.GetBool("general.confirm_on_exit") {
			err := gomu.quit(gomu.args)
			if err != nil {
				logError(err)
			}
		}
		exitConfirmation(gomu.args)
	})

	c.define("toggle_pause", func() {
		gomu.player.togglePause()
	})

	c.define("volume_up", func() {
		v := volToHuman(gomu.player.volume)
		if v < 100 {
			vol := gomu.player.setVolume(0.5)
			volumePopup(vol)
		}
	})

	c.define("volume_down", func() {
		v := volToHuman(gomu.player.volume)
		if v > 0 {
			vol := gomu.player.setVolume(-0.5)
			volumePopup(vol)
		}
	})

	c.define("skip", func() {
		gomu.player.skip()
	})

	c.define("toggle_help", func() {
		name, _ := gomu.pages.GetFrontPage()

		if name == "help-page" {
			gomu.pages.RemovePage(name)
			gomu.app.SetFocus(gomu.prevPanel.(tview.Primitive))
		} else {
			helpPopup(gomu.prevPanel)
		}
	})

	c.define("command_search", func() {

		names := make([]string, 0, len(c.commands))
		for commandName := range c.commands {
			names = append(names, commandName)
		}
		searchPopup(names, func(selected string) {

			for name, fn := range c.commands {
				if name == selected {
					fn()
				}
			}
		})
	})

  c.define("forward", func() {
    if gomu.player.isRunning && ! gomu.player.ctrl.Paused {
      position := gomu.playingBar._progress + 10
      if (position < gomu.playingBar.full) {
        gomu.player.seek(position)
        gomu.playingBar._progress = position - 1
      }
    }
  })

  c.define("rewind", func() {
    if gomu.player.isRunning && ! gomu.player.ctrl.Paused {
      position := gomu.playingBar._progress - 10
      if (position - 1 > 0 ) {
        gomu.player.seek(position)
        gomu.playingBar._progress = position -1
      } else {
        gomu.player.seek(0)
        gomu.playingBar._progress = 0
      }
    }
  })

  c.define("forward_fast", func() {
    if gomu.player.isRunning && ! gomu.player.ctrl.Paused {
      position := gomu.playingBar._progress + 60
      if (position < gomu.playingBar.full) {
        gomu.player.seek(position)
        gomu.playingBar._progress = position - 1
      }
    }
  })

  c.define("rewind_fast", func() {
    if gomu.player.isRunning && ! gomu.player.ctrl.Paused {
      position := gomu.playingBar._progress - 60
      if (position -1 > 0 ) {
        gomu.player.seek(position)
        gomu.playingBar._progress = position - 1
      } else {
        gomu.player.seek(0)
        gomu.playingBar._progress = 0
      }
    }
  })

  c.define("yank", func() {
    err := gomu.playlist.yank()
		if err != nil {
			logError(err)
		}
  })
  
  c.define("paste", func() {
    err := gomu.playlist.paste()
    if err != nil {
      logError(err)
    }
  })

}