config.go

// See LICENSE.txt for licensing information.

package main

import (
	"encoding/json"
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"
	"time"
)

const (
	CFG_VER_1 = iota + 1
)

const CFG_VER = CFG_VER_1

type Config struct {
	Root    string  // absolute path for game directories
	Games   []*Game // slice of games
	Version int     `json:Version,omitempty` // config format version
}

func (c *Config) PrintWhole() {
	if len(c.Games) < 1 {
		fmt.Println("No games defined")
		return
	}
	if flagVerbose {
		fmt.Printf("Config version: %d\n\n", c.Version)
	}
	fmt.Printf("%3s ", "#")
	c.Games[0].PrintHeader()
	for i, g := range c.Games {
		fmt.Printf("%3d ", i+1)
		g.Print()
	}
	fmt.Println()
}

func (c *Config) GetGame(name string) *Game {
	for _, g := range c.Games {
		if g.Name == name {
			return g
		}
	}
	return nil
}

func (c *Config) AddGame(name, path string) (gm *Game, err error) {
	r := filepath.Join(c.Root, name)
	if _, err = os.Stat(r); os.IsExist(err) {
		return
	}
	if _, err = os.Stat(path); os.IsNotExist(err) {
		return
	}
	if err = os.MkdirAll(r, 0777); err != nil {
		return
	}
	gm = &Game{
		Name:  name,
		Path:  path,
		Root:  r,
		Stamp: time.Now(),
		Saves: make([]*Save, 0),
	}
	c.Games = append(c.Games, gm)
	return gm, nil
}

func (c *Config) DelGame(name string) error {
	var i int
	var g *Game
	for i, g = range c.Games {
		if g.Name == name {
			break
		}
	}
	if g.Name != name {
		return fmt.Errorf(`Couldn't find game "%s"`, name)
	}
	if flagVerbose {
		fmt.Fprintln(os.Stderr, "removing all from:", g.Root)
	}
	if err := os.RemoveAll(g.Root); err != nil {
		return err
	}
	copy(c.Games[i:], c.Games[i+1:])
	c.Games[len(c.Games)-1] = nil
	c.Games = c.Games[:len(c.Games)-1]
	return nil
}

func (c *Config) CheckVer(min_ver int, print bool) bool {
	if c.Version >= min_ver {
		return true
	}

	if print {
		fmt.Fprintf(os.Stderr, "using old config format, please [migrate] (%d changes(s) behind)\n", CFG_VER-c.Version)
	}
	return false
}

func loadConfig(path string) (*Config, error) {
	if flagVerbose {
		fmt.Fprintln(os.Stderr, "loading config from", path)
	}
	f, err := os.Open(path)
	if err != nil {
		return nil, err
	}
	defer f.Close()
	d, err := ioutil.ReadAll(f)
	if err != nil {
		return nil, err
	}
	var c *Config
	if err := json.Unmarshal(d, &c); err != nil {
		return nil, err
	}
	c.CheckVer(CFG_VER, flagVerbose)

	return c, nil
}

func (c *Config) Save(path string) error {
	if flagVerbose {
		fmt.Fprintln(os.Stderr, "saving config to", path)
	}
	f, err := os.Create(path)
	if err != nil {
		return err
	}
	defer f.Close()
	j := json.NewEncoder(f)
	return j.Encode(c)
}

func (c *Config) Migrate() error {
	if c.Version < CFG_VER_1 {
		spinner.Msg("Migrating -> 1")
		for _, g := range c.Games {
			spinner.Tick()
			for _, s := range g.Saves {
				if s.Size > 0 {
					spinner.Tick()
					continue
				}
				i, err := os.Stat(s.Path)
				if err != nil {
					return err
				}
				spinner.Tick()
				size := uint64(i.Size())
				s.Size = size
				g.Size += size
			}
		}
		c.Version = CFG_VER_1
		spinner.Finish()
	}
	fmt.Println("All done")
	return nil
}