From eb377aeb2a287eab43cecef9dcae73f98365114e Mon Sep 17 00:00:00 2001
From: b4b4r07 <b4b4r07@gmail.com>
Date: Thu, 19 Apr 2018 19:32:02 +0900
Subject: [PATCH] Add shell struct to make shell command run

---
 main.go | 37 +++++++++++++++++++++++++++++++------
 1 file changed, 31 insertions(+), 6 deletions(-)

diff --git a/main.go b/main.go
index 2e1d04b..8abbd2c 100644
--- a/main.go
+++ b/main.go
@@ -133,7 +133,8 @@ func (c CLI) run() int {
 	)
 	args = append(args, url)
 
-	return c.exit(runCommand(env.Binary, args))
+	s := newShell(env.Binary, args)
+	return c.exit(s.run())
 }
 
 func (c CLI) debug(a ...interface{}) {
@@ -147,11 +148,32 @@ func (c CLI) getURL() string {
 	return c.urls[0].String()
 }
 
-func runCommand(command string, args []string) error {
+type shell struct {
+	stdin   io.Reader
+	stdout  io.Writer
+	stderr  io.Writer
+	env     map[string]string
+	command string
+	args    []string
+}
+
+func newShell(command string, args []string) shell {
+	return shell{
+		stdin:   os.Stdin,
+		stdout:  os.Stdout,
+		stderr:  os.Stderr,
+		env:     map[string]string{},
+		command: command,
+		args:    args,
+	}
+}
+
+func (s shell) run() error {
+	command := s.command
 	if _, err := exec.LookPath(command); err != nil {
 		return err
 	}
-	for _, arg := range args {
+	for _, arg := range s.args {
 		command += " " + arg
 	}
 	var cmd *exec.Cmd
@@ -160,8 +182,11 @@ func runCommand(command string, args []string) error {
 	} else {
 		cmd = exec.Command("sh", "-c", command)
 	}
-	cmd.Stderr = os.Stderr
-	cmd.Stdout = os.Stdout
-	cmd.Stdin = os.Stdin
+	cmd.Stderr = s.stderr
+	cmd.Stdout = s.stdout
+	cmd.Stdin = s.stdin
+	for k, v := range s.env {
+		cmd.Env = append(os.Environ(), fmt.Sprintf("%s=%s", k, v))
+	}
 	return cmd.Run()
 }