homemaker/command.go
2015-09-04 13:34:30 +09:00

102 lines
2.6 KiB
Go

/*
* Copyright (c) 2015 Alex Yatskov <alex@foosoft.net>
* Author: Alex Yatskov <alex@foosoft.net>
*
* 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"
"os/exec"
"strings"
)
type macro struct {
Prefix []string
Suffix []string
}
func processCmdMacro(macroName string, args []string, conf *config) error {
var found bool
for _, mn := range makeVariantNames(macroName, conf.variant) {
m, ok := conf.Macros[mn]
if !ok {
continue
}
margs := appendExpEnv(nil, m.Prefix)
margs = appendExpEnv(margs, args)
margs = appendExpEnv(margs, m.Suffix)
if conf.flags&flagVerbose != 0 {
log.Printf("using macro: %s", mn)
}
return processCmd(margs, conf)
}
if !found {
return fmt.Errorf("macro or variant not found: %s", macroName)
}
return nil
}
func processCmd(params []string, conf *config) error {
args := appendExpEnv(nil, params)
if len(args) == 0 {
return fmt.Errorf("invalid command statement")
}
cmdName := args[0]
if strings.HasPrefix(cmdName, "@") {
var (
macroName = strings.TrimPrefix(cmdName, "@")
macroArgs []string
)
if len(args) > 1 {
macroArgs = args[1:]
}
return processCmdMacro(macroName, macroArgs, conf)
}
var cmdArgs []string
if len(args) > 1 {
cmdArgs = args[1:]
}
if conf.flags&flagVerbose != 0 {
log.Printf("executing command: %s %s", cmdName, strings.Join(cmdArgs, " "))
}
return try(func() error {
cmd := exec.Command(cmdName, cmdArgs...)
cmd.Dir = conf.dstDir
cmd.Stderr = os.Stderr
cmd.Stdout = os.Stdout
cmd.Stdin = os.Stdin
return cmd.Run()
})
}