main.go

package main

import (
	"flag"
	"fmt"
	"io/ioutil"
	"os"
	"os/user"
	"path"
	"path/filepath"
	"text/template"
)

const (
	templateFileName = ".gb-sublime-project.template"
	defaultTemplate  = `{
    "folders":
    [
        {
            "path": "{{ .ProjectDir }}"
        }
    ],
    "settings": {
        "GoSublime": {
            "env": {
                "GOPATH": "{{ .ProjectDir }}:{{ .ProjectDir }}/vendor"
            }
        }
    }
}`
)

var (
	writeDefaultTemplate = flag.Bool("write-template", false, "writes default template to ~/"+templateFileName)
)

func main() {
	flag.Parse()
	wd, err := os.Getwd()
	if err != nil {
		panic(err)
	}

	u, err := user.Current()
	if err != nil {
		panic(err)
	}

	fullTemplatePath := path.Join(u.HomeDir, templateFileName)

	if *writeDefaultTemplate {
		out, err := os.Create(fullTemplatePath)
		if err != nil {
			panic(fmt.Errorf("Error opening template file for writing: %s", err.Error()))
		}
		out.WriteString(defaultTemplate)
		out.Close()
		fmt.Printf("writing template to %s...\n", fullTemplatePath)
	}

	f, err := os.Open(fullTemplatePath)
	var templateStr string
	switch {
	case os.IsNotExist(err):
		fmt.Println("template not found in home directory; using default...")
		templateStr = defaultTemplate
	case err != nil:
		panic(fmt.Sprintf("Error opening template file: %s", err.Error()))
	default:
		defer f.Close()
		b, err := ioutil.ReadAll(f)
		if err != nil {
			panic(err)
		}
		templateStr = string(b)
	}

	tmpl := template.Must(template.New("project").Parse(templateStr))

	projectName := filepath.Base(wd)
	projectFileName := projectName + ".sublime-project"

	fmt.Printf("writing %s to %s...\n", projectFileName, wd)
	out, err := os.Create(projectFileName)
	if err != nil {
		panic(err)
	}
	defer out.Close()

	err = tmpl.Execute(out, struct{ ProjectDir string }{wd})
	if err != nil {
		panic(err)
	}

}