bblfsh.go

package main

import (
	"github.com/bblfsh/sdk/protocol"
	"google.golang.org/grpc"
	"context"
	"os"
	"flag"
	"io/ioutil"
	"log"
	"time"
	"fmt"
)

func checkErr(err error) {
	if err != nil {
		panic(err)
	}
}

var (
	f_lang = flag.String("lang", "", "programming language")
)

func main() {
	flag.Parse()
	if flag.NArg() == 0 {
		log.Println("no files specified")
		os.Exit(1)
	}
	cc, err := grpc.Dial("localhost:9432", grpc.WithInsecure())
	checkErr(err)
	defer cc.Close()
	cli := protocol.NewProtocolServiceClient(cc)

	ctx := context.TODO()

	for _, arg := range flag.Args() {
		log.Println("processing", arg)
		data, err := ioutil.ReadFile(arg)
		checkErr(err)

		cctx, cancel := context.WithTimeout(ctx, time.Second*10)
		res, err := cli.ParseUAST(cctx, &protocol.ParseUASTRequest{
			Filename:arg,
			Language: *f_lang,
			Content: string(data),
		})
		select {
		case <-cctx.Done():
			log.Println("timeout on", arg)
			continue
		default:
			cancel()
			checkErr(err)
			if res.Status != protocol.Ok {
				checkErr(fmt.Errorf("status: %v: %v", res.Status, res.Errors))
			}
		}
	}
}