files.go

package golem

import (
	"io/ioutil"
	"os"
	"path/filepath"

	"github.com/tkerber/golem/xdg"
)

// files keeps track of all files golem uses.
type files struct {
	configDir     string
	cacheDir      string
	cookies       string
	rc            string
	searchEngines string
	quickmarks    string
	bookmarks     string
	histfile      string
	downloadDir   string
	filterlistDir string
}

// configFiles is an array of all of golems config files.
var configFiles = []string{
	"golemrc",
	"searchengines",
	"quickmarks",
	"bookmarks",
}

// newFiles initializes the files golem uses.
func (g *Golem) newFiles() (*files, error) {
	downloads := xdg.GetUserDownloadDir()
	// Default to "$HOME/Downloads"
	if downloads == "" {
		downloads = filepath.Join(xdg.GetHomeDir(), "Downloads")
	}

	configDir := xdg.GetUserConfigDir()
	configDir = filepath.Join(configDir, "golem", g.profile)
	err := os.MkdirAll(configDir, 0700)
	if err != nil {
		return nil, err
	}

	filterlistDir := filepath.Join(configDir, "filterlists")
	err = os.MkdirAll(filterlistDir, 0700)
	if err != nil {
		return nil, err
	}

	cacheDir := xdg.GetUserCacheDir()
	cacheDir = filepath.Join(cacheDir, "golem", g.profile)
	err = os.MkdirAll(cacheDir, 0700)
	if err != nil {
		return nil, err
	}

	cookies := filepath.Join(configDir, "cookies")

	configFiles, err := initConfigFiles(configDir)
	if err != nil {
		return nil, err
	}

	return &files{
		configDir,
		cacheDir,
		cookies,
		configFiles[0],
		configFiles[1],
		configFiles[2],
		configFiles[3],
		filepath.Join(configDir, "history"),
		downloads,
		filterlistDir,
	}, nil
}

// rcFiles returns a list of all files golem should use as rc files.
func (fs *files) rcFiles() []string {
	return []string{fs.rc, fs.searchEngines, fs.quickmarks, fs.bookmarks}
}

// initConfigFiles ensures all config files exist in the specified config
// dir.
func initConfigFiles(configDir string) ([]string, error) {
	locations := make([]string, len(configFiles))
	for i, file := range configFiles {
		locations[i] = filepath.Join(configDir, file)
		// If the config file does not exist, we create it with the default
		// content.
		_, err := os.Stat(locations[i])
		if err != nil && os.IsNotExist(err) {
			defaultCont, err := Asset(file)
			if err != nil {
				return nil, err
			}
			err = ioutil.WriteFile(locations[i], defaultCont, 0600)
			if err != nil {
				return nil, err
			}
		} else if err != nil {
			return nil, err
		}
	}
	return locations, nil
}