/* * Copyright (c) 2015 Alex Yatskov * Author: Alex Yatskov , Lucas Bremgartner * * Permission is hereby granted, free of charge, to any person obtaining a copy of * this software and associated documentation files (the "Software"), to deal in * the Software without restriction, including without limitation the rights to * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of * the Software, and to permit persons to whom the Software is furnished to do so, * subject to the following conditions: * * The above copyright notice and this permission notice shall be included in all * copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS * FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER * IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ package main import ( "fmt" "log" "os" "path" "strconv" "strings" "text/template" ) type context struct { } func (c *context) Env() map[string]string { env := make(map[string]string) for _, i := range os.Environ() { sep := strings.Index(i, "=") env[i[0:sep]] = i[sep+1:] } return env } func parseTemplate(params []string) (srcPath, dstPath string, mode os.FileMode, err error) { length := len(params) if length < 1 || length > 3 { err = fmt.Errorf("invalid template statement") return } if length > 2 { var parsed uint64 parsed, err = strconv.ParseUint(params[2], 0, 64) if err != nil { return } mode = os.FileMode(parsed) } else { mode = 0755 } dstPath = os.ExpandEnv(params[0]) srcPath = dstPath if length > 1 { srcPath = os.ExpandEnv(params[1]) } return } func processTemplate(params []string, conf *config) (err error) { srcPath, dstPath, mode, err := parseTemplate(params) if err != nil { return err } srcPathAbs := srcPath if !path.IsAbs(srcPathAbs) { srcPathAbs = path.Join(conf.srcDir, srcPath) } dstPathAbs := dstPath if !path.IsAbs(dstPathAbs) { dstPathAbs = path.Join(conf.dstDir, dstPath) } if _, err = os.Stat(srcPathAbs); os.IsNotExist(err) { return fmt.Errorf("source path %s does not exist in filesystem", srcPathAbs) } if err = try(func() error { return createPath(dstPathAbs, conf.flags, mode) }); err != nil { return err } if err = try(func() error { return cleanPath(dstPathAbs, conf.flags) }); err != nil { return err } if conf.flags&flagVerbose != 0 { log.Printf("process template %s to %s", srcPathAbs, dstPathAbs) } t, err := template.New(srcPath).ParseFiles(srcPathAbs) if err != nil { return err } f, err := os.Create(dstPathAbs) if err != nil { return err } defer func() { err = f.Close() }() return try(func() error { return t.Execute(f, &context{}) }) }