Ejemplo n.º 1
0
Archivo: main.go Proyecto: tcard/sgo
func handleMsg(msg msgType) {
	c := msg.c
	if c == nil {
		log.Println("c shouldn't be nil")
		return
	}
	switch msg.Type {
	case "format":
		resp := &msgType{
			Type: "format",
		}
		func() {
			defer func() {
				if r := recover(); r != nil {
					value := fmt.Sprintln(r)
					stack := make([]byte, 99999)
					runtime.Stack(stack, false)
					value += string(stack)
					resp.Value = value
				}
			}()
			formatted, err := format.Source([]byte(msg.Value.(string)))
			if err == nil {
				resp.Value = string(formatted)
			}
		}()
		c.WriteJSON(resp)
	case "translate":
		resp := &msgType{
			Type: "translate",
		}
		func() {
			defer func() {
				if r := recover(); r != nil {
					value := fmt.Sprintln(r)
					stack := make([]byte, 99999)
					runtime.Stack(stack, false)
					value += string(stack)
					resp.Value = value
				}
			}()
			w := &bytes.Buffer{}
			errs := sgo.TranslateFile(func() (io.Writer, error) { return w, nil }, strings.NewReader(msg.Value.(string)), "name")
			if errs != nil {
				var errMsgs []string
				for _, err := range errs {
					if errs, ok := err.(scanner.ErrorList); ok {
						for _, err := range errs {
							errMsgs = append(errMsgs, err.Error())
						}
					} else {
						errMsgs = append(errMsgs, err.Error())
					}
				}
				resp.Value = strings.Join(errMsgs, "\n")
			} else {
				resp.Value = w.String()
			}
		}()
		c.WriteJSON(resp)
	case "execute":
		resp := &msgType{
			Type: "execute",
		}
		body := url.Values{}
		body.Add("version", "2")
		var errs []error
		w := &bytes.Buffer{}
		func() {
			defer func() {
				if r := recover(); r != nil {
					value := fmt.Sprintln(r)
					stack := make([]byte, 99999)
					runtime.Stack(stack, false)
					value += string(stack)
					errs = append(errs, errors.New(value))
				}
			}()

			errs = sgo.TranslateFile(func() (io.Writer, error) { return w, nil }, strings.NewReader(msg.Value.(string)), "name")
		}()
		if errs != nil {
			var errMsgs []string
			for _, err := range errs {
				if errs, ok := err.(scanner.ErrorList); ok {
					for _, err := range errs {
						errMsgs = append(errMsgs, err.Error())
					}
				} else {
					errMsgs = append(errMsgs, err.Error())
				}
			}
			resp.Value = strings.Join(errMsgs, "\n")
		} else {
			body.Add("body", w.String())
			postResp, err := http.PostForm("https://play.golang.org/compile", body)
			if err != nil {
				resp.Value = err.Error()
			} else {
				var v interface{}
				err := json.NewDecoder(postResp.Body).Decode(&v)
				postResp.Body.Close()
				if err != nil {
					resp.Value = err.Error()
				} else {
					resp.Value = v
				}
			}
		}
		c.WriteJSON(resp)
	}
}
Ejemplo n.º 2
0
Archivo: main.go Proyecto: tcard/sgo
func main() {
	if len(os.Args) == 1 {
		fmt.Print(helpMsg)
		return
	}

	var buildFlags []string
	var extraArgs []string
	for i, arg := range os.Args[2:] {
		if arg[0] == '-' {
			buildFlags = append(buildFlags, arg)
		} else {
			extraArgs = os.Args[i+2:]
			break
		}
	}

	switch os.Args[1] {
	case "version":
		fmt.Println("sgo version 0.7 (compatible with go1.7)")
		return
	case "run":
		if len(extraArgs) == 0 {
			fmt.Fprintln(os.Stderr, "sgo run: no files listed")
			os.Exit(1)
		}
		created, errs := sgo.TranslateFilePaths(extraArgs...)
		reportErrs(errs...)
		if len(errs) > 0 {
			os.Exit(1)
		}
		runGoCommand("run", buildFlags, created...)
		return
	case "help":
		if len(extraArgs) == 0 {
			fmt.Print(helpMsg)
		} else {
			switch extraArgs[0] {
			case "translate":
				fmt.Print(translateHelpMsg)
				return
			case "version":
				fmt.Print(versionHelpMsg)
				return
			}
			runGoCommand("help", buildFlags, extraArgs...)
		}
		return
	case "translate":
		errs := sgo.TranslateFile(func() (io.Writer, error) { return os.Stdout, nil }, os.Stdin, "stdin.sgo")
		if len(errs) > 0 {
			reportErrs(errs...)
			os.Exit(1)
		}
		return
	}

	if len(extraArgs) == 0 {
		extraArgs = append(extraArgs, ".")
	}
	_, warnings, errs := sgo.TranslatePaths(extraArgs)
	reportErrs(warnings...)
	reportErrs(errs...)
	if len(errs) > 0 {
		os.Exit(1)
	}

	runGoCommand(os.Args[1], buildFlags, extraArgs...)
}