coverage_handlers.go

package main

import (
	"github.com/datatogether/api/apiutil"
	"github.com/datatogether/core"
	"github.com/datatogether/coverage/coverage"
	"github.com/datatogether/coverage/tree"
	"net"
	"net/http"
	"net/rpc"
	"strings"
)

func CoverageHandler(w http.ResponseWriter, r *http.Request) {
	switch r.Method {
	case "OPTIONS":
		EmptyOkHandler(w, r)
	case "GET":
		CoverageTreeHandler(w, r)
	default:
		NotFoundHandler(w, r)
	}
}

func CoverageTreeHandler(w http.ResponseWriter, r *http.Request) {
	var primer *core.Primer
	patterns := strings.Split(r.FormValue("patterns"), ",")
	if r.FormValue("primer") != "" {
		primer = &core.Primer{Id: r.FormValue("primer")}
		if err := primer.Read(store); err != nil {
			apiutil.WriteErrResponse(w, http.StatusBadRequest, err)
			return
		}
		if err := primer.ReadSources(appDB); err != nil {
			apiutil.WriteErrResponse(w, http.StatusBadRequest, err)
			return
		}
		patterns = make([]string, len(primer.Sources))
		for i, s := range primer.Sources {
			patterns[i] = s.Url
		}
	}

	conn, err := net.Dial("tcp", cfg.CoverageServiceUrl)
	if err != nil {
		apiutil.WriteErrResponse(w, http.StatusInternalServerError, err)
		return
	}
	cli := rpc.NewClient(conn)

	p := coverage.CoverageTreeParams{
		Root:     r.FormValue("root"),
		Patterns: patterns,
		// Depth:    depth,
		// RepoIds:  strings.Split(r.FormValue("repositories"), ","),
	}

	if depth, err := apiutil.ReqParamInt("depth", r); err == nil {
		p.Depth = depth
	}

	// if r.FormValue("patterns") != "" {
	// 	args.Patterns = strings.Split(r.FormValue("patterns"), ",")
	// }

	if r.FormValue("repos") != "" {
		p.RepoIds = strings.Split(r.FormValue("repos"), ",")
	}

	reply := &tree.Node{}
	if err := cli.Call("CoverageRequests.Tree", p, reply); err != nil {
		apiutil.WriteErrResponse(w, http.StatusInternalServerError, err)
		return
	}

	if primer != nil {
		reply.Name = primer.Title
	}
	apiutil.WriteResponse(w, reply)
}

// func CoverageSummaryHandler(w http.ResponseWriter, r *http.Request) {
// 	conn, err := net.Dial("tcp", cfg.CoverageServiceUrl)
// 	if err != nil {
// 		apiutil.WriteErrResponse(w, http.StatusInternalServerError, err)
// 		return
// 	}
// 	cli := rpc.NewClient(conn)
// 	p := coverage.CoverageSummaryParams{
// 		Patterns: strings.Split(r.FormValue("patterns"), ","),
// 	}
// 	reply := &coverage.Summary{}
// 	if err := cli.Call("CoverageRequests.Summary", p, reply); err != nil {
// 		apiutil.WriteErrResponse(w, http.StatusInternalServerError, err)
// 		return
// 	}
// 	apiutil.WriteResponse(w, reply)
// }