package cmds

import (
	"fmt"
	"io/ioutil"
	"path/filepath"
	"sort"
	"strings"

	"github.com/maximilien/i18n4go/common"
)

type MergeStrings struct {
	options common.Options

	I18nStringInfos []common.I18nStringInfo

	Recurse        bool
	SourceLanguage string
	Directory      string
}

func NewMergeStrings(options common.Options) MergeStrings {
	return MergeStrings{
		options:         options,
		I18nStringInfos: []common.I18nStringInfo{},
		Recurse:         options.RecurseFlag,
		SourceLanguage:  options.SourceLanguageFlag,
		Directory:       options.DirnameFlag,
	}
}

func (ms *MergeStrings) Options() common.Options {
	return ms.options
}

func (ms *MergeStrings) Println(a ...interface{}) (int, error) {
	if ms.options.VerboseFlag {
		return fmt.Println(a...)
	}

	return 0, nil
}

func (ms *MergeStrings) Printf(msg string, a ...interface{}) (int, error) {
	if ms.options.VerboseFlag {
		return fmt.Printf(msg, a...)
	}

	return 0, nil
}

func (ms *MergeStrings) Run() error {
	return ms.combineStringInfosPerDirectory(ms.Directory)
}

func (ms *MergeStrings) combineStringInfosPerDirectory(directory string) error {
	files, directories := getFilesAndDir(directory)
	fileList := ms.matchFileToSourceLanguage(files, ms.SourceLanguage)

	combinedMap := map[string]common.I18nStringInfo{}
	for _, file := range fileList {
		StringInfos, err := common.LoadI18nStringInfos(file)
		if err != nil {
			return err
		}

		combineStringInfo(StringInfos, combinedMap)
	}

	filePath := filepath.Join(directory, ms.SourceLanguage+".all.json")
	ms.I18nStringInfos = common.I18nStringInfoMapValues2Array(combinedMap)
	sort.Sort(ms)
	common.SaveI18nStringInfos(ms, ms.Options(), ms.I18nStringInfos, filePath)
	ms.Println("gi18n: saving combined language file: " + filePath)

	if ms.Recurse {
		for _, directory = range directories {
			err := ms.combineStringInfosPerDirectory(directory)
			if err != nil {
				return err
			}
		}
	}

	return nil
}

func getFilesAndDir(dir string) (files []string, dirs []string) {
	contents, _ := ioutil.ReadDir(dir)

	for _, fileInfo := range contents {
		if !fileInfo.IsDir() {
			files = append(files, filepath.Join(dir, fileInfo.Name()))
		} else {
			dirs = append(dirs, filepath.Join(dir, fileInfo.Name()))
		}
	}
	return
}

func (ms MergeStrings) matchFileToSourceLanguage(files []string, lang string) (list []string) {
	languageMatcher := "go." + lang + ".json"
	for _, file := range files {
		if strings.Contains(file, languageMatcher) {
			list = append(list, file)
			ms.Println("gi18n: scanning file: " + file)
		}
	}
	return
}

func combineStringInfo(stringInfoList []common.I18nStringInfo, combinedMap map[string]common.I18nStringInfo) {
	for _, stringInfo := range stringInfoList {
		if _, ok := combinedMap[stringInfo.ID]; !ok {
			combinedMap[stringInfo.ID] = stringInfo
		}
	}
}

// sort.Interface methods

func (ms *MergeStrings) Len() int {
	return len(ms.I18nStringInfos)
}

func (ms *MergeStrings) Less(i, j int) bool {
	return ms.I18nStringInfos[i].ID < ms.I18nStringInfos[j].ID
}

func (ms *MergeStrings) Swap(i, j int) {
	tmpI18nStringInfo := ms.I18nStringInfos[i]
	ms.I18nStringInfos[i] = ms.I18nStringInfos[j]
	ms.I18nStringInfos[j] = tmpI18nStringInfo
}