okrun.go

package main

import (
	"bufio"
	"bytes"
	"log"
	"os"
	"os/exec"
)

func main() {
	rerunTargetFile := true

	var errorBytes bytes.Buffer
	captureErr := bufio.NewWriter(&errorBytes)

	if len(os.Args) < 2 {
		log.Fatal("Error: please provide a target gofile path.\nUsage: okrun <path/to/file.go> [arg]*")
	}
	commandArgs := []string{"run"}
	commandArgs = append(commandArgs, os.Args[1:]...)

	for rerunTargetFile {
		if !rerunTargetFile {
			return
		}

		errorBytes.Reset()

		cmd := exec.Command("go", commandArgs...)
		cmd.Stdout = os.Stdout
		cmd.Stderr = captureErr
		err := cmd.Start()
		if err != nil {
			panic(err)
		}
		cmd.Wait()
		captureErr.Flush()

		handleableErrors, unhandleableLines := handleErrors(errorBytes.String())

		if len(unhandleableLines) > 0 {
			errorMsg := unhandleableLinesMessage(unhandleableLines)
			log.Fatal(errorMsg)
		} else if len(handleableErrors) > 0 {
			if err = fixErrors(handleableErrors); err != nil {
				log.Fatal("Error: okrun encountered an error while fixing errors.", err)
			}
		} else {
			rerunTargetFile = false
		}
	}
}

func unhandleableLinesMessage(lines []string) string {
	errorMsg := "Error: the following errors cannot be automatically corrected by okrun to run your file:\n"
	for i := 0; i < len(lines); i++ {
		errorMsg = errorMsg + "* " + lines[i] + "\n"
	}
	errorMsg = errorMsg + "Please fix the error(s) listed above and then use okrun."
	return errorMsg
}