121 lines
2.2 KiB
Go
121 lines
2.2 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
// +build ignore
|
|
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
var (
|
|
mainpkg = flag.String("pkg", "storj.io/storj/private/multinodepb", "main package name")
|
|
protoc = flag.String("protoc", "protoc", "protoc compiler")
|
|
)
|
|
|
|
var ignoreProto = map[string]bool{
|
|
"gogo.proto": true,
|
|
}
|
|
|
|
func ignore(files []string) []string {
|
|
xs := []string{}
|
|
for _, file := range files {
|
|
if !ignoreProto[file] {
|
|
xs = append(xs, file)
|
|
}
|
|
}
|
|
return xs
|
|
}
|
|
|
|
// Programs needed for code generation:
|
|
//
|
|
// github.com/ckaznocha/protoc-gen-lint
|
|
// storj.io/drpc/cmd/protoc-gen-drpc
|
|
// github.com/nilslice/protolock/cmd/protolock
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
// TODO: protolock
|
|
|
|
{
|
|
// cleanup previous files
|
|
localfiles, err := filepath.Glob("*.pb.go")
|
|
check(err)
|
|
|
|
all := []string{}
|
|
all = append(all, localfiles...)
|
|
for _, match := range all {
|
|
_ = os.Remove(match)
|
|
}
|
|
}
|
|
|
|
{
|
|
protofiles, err := filepath.Glob("*.proto")
|
|
check(err)
|
|
|
|
protofiles = ignore(protofiles)
|
|
|
|
overrideImports := ",Mgoogle/protobuf/timestamp.proto=" + *mainpkg
|
|
args := []string{
|
|
"--lint_out=.",
|
|
"--drpc_out=plugins=drpc,paths=source_relative" + overrideImports + ":.",
|
|
"-I=.",
|
|
}
|
|
args = append(args, protofiles...)
|
|
|
|
// generate new code
|
|
cmd := exec.Command(*protoc, args...)
|
|
fmt.Println(strings.Join(cmd.Args, " "))
|
|
out, err := cmd.CombinedOutput()
|
|
if len(out) > 0 {
|
|
fmt.Println(string(out))
|
|
}
|
|
check(err)
|
|
}
|
|
|
|
{
|
|
files, err := filepath.Glob("*.pb.go")
|
|
check(err)
|
|
for _, file := range files {
|
|
process(file)
|
|
}
|
|
}
|
|
|
|
{
|
|
// format code to get rid of extra imports
|
|
out, err := exec.Command("goimports", "-local", "storj.io", "-w", ".").CombinedOutput()
|
|
if len(out) > 0 {
|
|
fmt.Println(string(out))
|
|
}
|
|
check(err)
|
|
}
|
|
}
|
|
|
|
func process(file string) {
|
|
data, err := ioutil.ReadFile(file)
|
|
check(err)
|
|
|
|
source := string(data)
|
|
|
|
// When generating code to the same path as proto, it will
|
|
// end up generating an `import _ "."`, the following replace removes it.
|
|
source = strings.Replace(source, `_ "."`, "", -1)
|
|
|
|
err = ioutil.WriteFile(file, []byte(source), 0644)
|
|
check(err)
|
|
}
|
|
|
|
func check(err error) {
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|