827bbf1143
made to parser. This was missed due to the previously mentioned disparity in import paths not causing install to break as would be expected.
116 lines
2.7 KiB
Go
116 lines
2.7 KiB
Go
// Copyright 2014 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// Server is an http server that provides an alternative way of generating code
|
|
// based on int types and the constants defined with it.
|
|
//
|
|
// Use the http flag to change the address on which the server will listen for
|
|
// requests. The default is 127.0.0.1:8080.
|
|
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"text/template"
|
|
|
|
"go/format"
|
|
|
|
"github.com/campoy/jsonenums/parser"
|
|
)
|
|
|
|
func init() {
|
|
http.Handle("/generate", handler(generateHandler))
|
|
}
|
|
|
|
func generateHandler(w http.ResponseWriter, r *http.Request) error {
|
|
if r.Method != "GET" {
|
|
return codeError{fmt.Errorf("only GET accepted"), http.StatusMethodNotAllowed}
|
|
}
|
|
code := r.FormValue("code")
|
|
if code == "" {
|
|
return codeError{fmt.Errorf("no code to be parsed"), http.StatusBadRequest}
|
|
}
|
|
typ := r.FormValue("type")
|
|
if typ == "" {
|
|
return codeError{fmt.Errorf("no type to be analyzed"), http.StatusBadRequest}
|
|
}
|
|
|
|
dir, err := createDir(code)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer os.RemoveAll(dir)
|
|
|
|
pkg, err := parser.ParsePackage(dir, "", "")
|
|
if err != nil {
|
|
return fmt.Errorf("parse package: %v", err)
|
|
}
|
|
|
|
values, err := pkg.ValuesOfType(typ)
|
|
if err != nil {
|
|
return fmt.Errorf("find values: %v", err)
|
|
}
|
|
|
|
t, err := template.New("code").Parse(r.FormValue("template"))
|
|
if err != nil {
|
|
return codeError{fmt.Errorf("parse template: %v", err), http.StatusBadRequest}
|
|
}
|
|
var data = struct {
|
|
PackageName string
|
|
TypeName string
|
|
Values []string
|
|
}{pkg.Name, typ, values}
|
|
|
|
var buf bytes.Buffer
|
|
if err := t.Execute(&buf, data); err != nil {
|
|
return codeError{fmt.Errorf("execute template: %v", err), http.StatusBadRequest}
|
|
}
|
|
src, err := format.Source(buf.Bytes())
|
|
if err != nil {
|
|
return codeError{fmt.Errorf("code generated is not valid: %v\n%v", err, buf.String()), http.StatusBadRequest}
|
|
}
|
|
w.Write(src)
|
|
return nil
|
|
}
|
|
|
|
func createDir(content string) (string, error) {
|
|
dir, err := ioutil.TempDir("", "jsonenums")
|
|
if err != nil {
|
|
return "", fmt.Errorf("create tmp dir: %v", err)
|
|
}
|
|
f, err := os.Create(filepath.Join(dir, "main.go"))
|
|
if err != nil {
|
|
os.RemoveAll(dir)
|
|
return "", fmt.Errorf("create tmp file: %v", err)
|
|
}
|
|
f.WriteString(content)
|
|
f.Close()
|
|
return dir, err
|
|
}
|
|
|
|
type handler func(http.ResponseWriter, *http.Request) error
|
|
|
|
func (h handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
err := h(w, r)
|
|
if err != nil {
|
|
code := http.StatusInternalServerError
|
|
if cErr, ok := err.(codeError); ok {
|
|
code = cErr.code
|
|
} else {
|
|
log.Printf("%v: %v", r.URL, code)
|
|
}
|
|
http.Error(w, err.Error(), code)
|
|
}
|
|
}
|
|
|
|
type codeError struct {
|
|
error
|
|
code int
|
|
}
|