init.go

/*
 * Copyright (c) 2013 Dario Castañé.
 * This file is part of Zas.
 *
 * Zas is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Zas is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with Zas.  If not, see <http://www.gnu.org/licenses/>.
 */
package zas

import (
	"fmt"
	"os"
	"path/filepath"

	"dario.cat/mergo"
	"github.com/melvinmt/gt"
	yaml "gopkg.in/yaml.v2"
)

/*
 * Aliasing goyaml's default map type.
 */
type ConfigSection map[interface{}]interface{}

/*
 * Loads ZAS_CONF_FILE (as defined in constants.go).
 * It must be a YAML file.
 */
func NewConfig() (ConfigSection, error) {
	data, err := os.ReadFile(ZAS_CONF_FILE)
	if err != nil {
		return nil, err
	}

	var config ConfigSection
	if err = yaml.Unmarshal(data, &config); err != nil {
		return nil, err
	}

	if err = mergo.Merge(&config, ZAS_DEFAULT_CONF); err != nil {
		return nil, err
	}

	return config, nil
}

/*
 * Loads ZAS_I18N_FILE (as defined in constants.go).
 * It must be a YAML file.
 */
func NewI18n(mainlang string) (i18n gt.Strings, err error) {
	data, err := os.ReadFile(ZAS_I18N_FILE)
	i18n = make(gt.Strings)
	if err != nil {
		if os.IsNotExist(err) {
			err = nil
			return
		}
		i18n = nil
		return
	}
	err = yaml.Unmarshal(data, &i18n)
	for k, v := range i18n {
		if _, ok := v[mainlang]; !ok {
			v[mainlang] = k
		}
	}
	return
}

/*
 * Returns a string value from current section.
 */
func (cs ConfigSection) GetString(key string) (value string) {
	raw := cs[key]
	if raw == nil {
		value = ""
	} else {
		value = raw.(string)
	}
	return
}

/*
 * Returns a subsection from current section.
 */
func (cs ConfigSection) GetSection(key string) (value ConfigSection) {
	var ok bool
	if value, ok = cs[key].(ConfigSection); !ok {
		value = ConfigSection(cs[key].(map[interface{}]interface{}))
	}
	return
}

/*
 * Returns a string value from default Zas section.
 */
func (cs ConfigSection) GetZString(key string) string {
	s := cs.GetSection(ZAS)
	return s.GetString(key)
}

type Init struct{}

func (i *Init) Run() {
	path, _ := filepath.Abs(ZAS_DIR)
	if _, err := os.Stat(ZAS_DIR); os.IsNotExist(err) {
		os.Mkdir(ZAS_DIR, os.FileMode(ZAS_DEFAULT_DIR_PERM))
		fmt.Printf("Initialized empty %s repository in %s\n", ZAS_NAME, path)
	} else {
		fmt.Printf("Reinitialized existing %s repository in %s\n", ZAS_NAME, path)
	}
	var (
		data []byte
		err  error
	)
	// If default config variable has fields, we store it as ZAS_CONF_FILE (as defined in constants.go).
	// It overwrites every time we invoke init subcommand.
	if len(ZAS_DEFAULT_CONF) > 0 {
		if data, err = yaml.Marshal(&ZAS_DEFAULT_CONF); err != nil {
			panic(err)
		}
	}
	if err := os.WriteFile(ZAS_CONF_FILE, data, os.FileMode(ZAS_DEFAULT_FILE_PERM)); err != nil {
		panic(err)
	}
}