|
| 1 | +package main |
| 2 | + |
| 3 | +import ( |
| 4 | + "flag" |
| 5 | + "fmt" |
| 6 | + "io" |
| 7 | + "log" |
| 8 | + "os" |
| 9 | + "os/exec" |
| 10 | + "runtime" |
| 11 | + "time" |
| 12 | + |
| 13 | + "github.com/cespare/window" |
| 14 | + "github.com/jeanfric/goembed/countingwriter" |
| 15 | +) |
| 16 | + |
| 17 | +const Version = "0.1.0" |
| 18 | + |
| 19 | +var ( |
| 20 | + toFlag = flag.String("to", "", "email address to send output to") |
| 21 | + subjectFlag = flag.String("s", "", "subject of email (optional)") |
| 22 | + maxFlag = flag.Int("max", 10000, "max bytes to capture for stdout/stderr") |
| 23 | +) |
| 24 | + |
| 25 | +func usageAndExit(s string) { |
| 26 | + fmt.Printf("!! %s\n", s) |
| 27 | + flag.Usage() |
| 28 | + fmt.Println() |
| 29 | + fmt.Printf("%s version: %s (%s on %s/%s; %s)\n", os.Args[0], Version, runtime.Version(), runtime.GOOS, runtime.GOARCH, runtime.Compiler) |
| 30 | + os.Exit(1) |
| 31 | +} |
| 32 | + |
| 33 | +func init() { |
| 34 | + runtime.GOMAXPROCS(1) |
| 35 | + runtime.LockOSThread() |
| 36 | + |
| 37 | + flag.Usage = func() { |
| 38 | + fmt.Fprint(os.Stderr, "usage: email-me [flags] [command]\n") |
| 39 | + flag.PrintDefaults() |
| 40 | + } |
| 41 | + flag.Parse() |
| 42 | +} |
| 43 | + |
| 44 | +func main() { |
| 45 | + if *toFlag == "" { |
| 46 | + usageAndExit("missing -to=[address]") |
| 47 | + } |
| 48 | + |
| 49 | + args := flag.Args() |
| 50 | + if len(args) == 0 { |
| 51 | + usageAndExit("missing [command]") |
| 52 | + } |
| 53 | + |
| 54 | + truncStdout := window.NewWriter(*maxFlag) |
| 55 | + truncStderr := window.NewWriter(*maxFlag) |
| 56 | + countedStdout := countingwriter.New(truncStdout) |
| 57 | + countedStderr := countingwriter.New(truncStderr) |
| 58 | + child := exec.Command(args[0], args[1:]...) |
| 59 | + child.Stdout = io.MultiWriter(os.Stdout, countedStdout) |
| 60 | + child.Stderr = io.MultiWriter(os.Stderr, countedStderr) |
| 61 | + |
| 62 | + start := time.Now() |
| 63 | + err := child.Run() |
| 64 | + end := time.Now() |
| 65 | + |
| 66 | + stdout := truncStdout.Bytes() |
| 67 | + stderr := truncStderr.Bytes() |
| 68 | + |
| 69 | + r := &Result{ |
| 70 | + Cmd: child, |
| 71 | + Error: err, |
| 72 | + Start: start, |
| 73 | + End: end, |
| 74 | + Duration: end.Sub(start), |
| 75 | + Stdout: string(stdout), |
| 76 | + Stderr: string(stderr), |
| 77 | + StdoutExtra: countedStdout.BytesWritten() - len(stdout), |
| 78 | + StderrExtra: countedStderr.BytesWritten() - len(stderr), |
| 79 | + } |
| 80 | + |
| 81 | + me := identity() |
| 82 | + subject := *subjectFlag |
| 83 | + if subject == "" { |
| 84 | + subject = fmt.Sprintf("%s", child.Args) |
| 85 | + } |
| 86 | + |
| 87 | + m := &Message{ |
| 88 | + To: *toFlag, |
| 89 | + From: me, |
| 90 | + Subject: subject, |
| 91 | + Result: r, |
| 92 | + } |
| 93 | + |
| 94 | + if err := findMailer().Send([]string{*toFlag}, me, m.Bytes()); err != nil { |
| 95 | + log.Fatal(err) |
| 96 | + } |
| 97 | +} |
0 commit comments