filemap.go

// Copyright 2018 Mathew Robinson <chasinglogic@gmail.com>. All rights reserved.
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program 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 General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.


package filemap

import (
	"fmt"
	"os"
	"path/filepath"
	"regexp"
	"strings"
)

// Mappings is a list of Mapping's
type Mappings []Mapping

// Mapping is a way for DFM to map custom locations to custom locations
type Mapping struct {
	Match  string `yaml:"match"`
	Dest   string `yaml:"dest"`
	Regexp bool   `yaml:"regexp"`
	IsDir  bool   `yaml:"is_dir"`
	Skip   bool   `yaml:"skip"`
}

func New() Mapping {
	return Mapping{
		Match:  "",
		Dest:   "",
		Regexp: false,
		IsDir:  true,
		Skip:   false,
	}
}

func (m Mapping) Matches(filename string) bool {
	if m.Regexp {
		rg, err := regexp.Compile(m.Match)
		if err != nil {
			fmt.Println("ERROR compiling match regex:", err.Error())
			return false
		}

		return rg.Match([]byte(filename))
	}

	return strings.HasPrefix(filename, m.Match)
}

func (m Mappings) Matches(filename string) *Mapping {
	for i := range m {
		if m[i].Matches(filename) {
			return &m[i]
		}
	}

	return nil
}

func DefaultMappings() Mappings {
	xdg := os.Getenv("XDG_CONFIG_HOME")
	if xdg == "" {
		xdg = filepath.Join(os.Getenv("HOME"), ".config")
	}

	xdgconfig := New()
	xdgconfig.Match = "^[.]?config"
	xdgconfig.Regexp = true
	xdgconfig.Dest = xdg

	return Mappings{
		xdgconfig,
		{
			Match:  "^[.]?ggitignore",
			IsDir:  false,
			Regexp: true,
			Dest:   "gitignore",
		},
		{
			Match:  "^\\.git",
			IsDir:  true,
			Regexp: true,
			Skip:   true,
		},
		{
			Match:  "^\\.gitignore$",
			IsDir:  false,
			Regexp: true,
			Skip:   true,
		},
		{
			Match:  "^LICENSE(\\.md)?$",
			IsDir:  false,
			Regexp: true,
			Skip:   true,
		},
		{
			Match:  "^\\.dfm\\.yml$",
			IsDir:  false,
			Regexp: true,
			Skip:   true,
		},
		{
			Match:  "^README(\\.md)?$",
			IsDir:  false,
			Regexp: true,
			Skip:   true,
		},
		{
			Match:  "^\\.modules",
			IsDir:  true,
			Skip:   true,
			Regexp: true,
		},
	}
}