Merge pull request #42 from hairyhenderson/switch-to-codegangsta-cli

Switching argument parsing to codegangsta/cli

1 files changed, 16 insertions, 14 deletions

diff --git a/main.go b/main.go
index 007de5c0..38f1c5f3 100644
--- a/main.go
+++ b/main.go
@@ -1,27 +1,18 @@
 package main
 
 import (
-	"flag"
-	"fmt"
 	"io"
 	"io/ioutil"
 	"log"
 	"os"
 
-	"github.com/hairyhenderson/gomplate/aws"
-	"github.com/hairyhenderson/gomplate/version"
 	"strings"
 	"text/template"
-)
 
-func init() {
-	ver := flag.Bool("v", false, "Print version and exit")
-	flag.Parse()
-	if *ver {
-		fmt.Println(version.Version)
-		os.Exit(0)
-	}
-}
+	"github.com/codegangsta/cli"
+	"github.com/hairyhenderson/gomplate/aws"
+	"github.com/hairyhenderson/gomplate/version"
+)
 
 func (g *Gomplate) createTemplate() *template.Template {
 	return template.New("template").Funcs(g.funcMap).Option("missingkey=error")
@@ -75,7 +66,18 @@ func NewGomplate() *Gomplate {
 	}
 }
 
-func main() {
+func runTemplate(c *cli.Context) error {
 	g := NewGomplate()
 	g.RunTemplate(os.Stdin, os.Stdout)
+	return nil
+}
+
+func main() {
+	app := cli.NewApp()
+	app.Name = "gomplate"
+	app.Usage = "Process text files with Go templates"
+	app.Version = version.Version
+	app.Action = runTemplate
+
+	app.Run(os.Args)
 }