|
|
@@ -1,96 +0,0 @@
|
|
|
-// +build generate
|
|
|
-
|
|
|
-package main
|
|
|
-
|
|
|
-import (
|
|
|
- "bytes"
|
|
|
- "fmt"
|
|
|
- "io/ioutil"
|
|
|
- "os"
|
|
|
- "os/exec"
|
|
|
- "path/filepath"
|
|
|
- "runtime"
|
|
|
- "strings"
|
|
|
-
|
|
|
- "v2ray.com/core/common"
|
|
|
-)
|
|
|
-
|
|
|
-var protocMap = map[string]string{
|
|
|
- "windows": filepath.Join(os.Getenv("GOPATH"), "src", "v2ray.com", "core", ".dev", "protoc", "windows", "protoc.exe"),
|
|
|
- "darwin": filepath.Join(os.Getenv("GOPATH"), "src", "v2ray.com", "core", ".dev", "protoc", "macos", "protoc"),
|
|
|
- "linux": filepath.Join(os.Getenv("GOPATH"), "src", "v2ray.com", "core", ".dev", "protoc", "linux", "protoc"),
|
|
|
-}
|
|
|
-
|
|
|
-func sdkPath(lang string) string {
|
|
|
- path := filepath.Join(os.Getenv("GOPATH"), "src", "github.com", "v2ray", "sdk-"+lang, "proto")
|
|
|
- os.MkdirAll(path, os.ModePerm)
|
|
|
- return path
|
|
|
-}
|
|
|
-
|
|
|
-func main() {
|
|
|
- protofiles := make(map[string][]string)
|
|
|
- protoc := protocMap[runtime.GOOS]
|
|
|
- gosrc := filepath.Join(os.Getenv("GOPATH"), "src")
|
|
|
- reporoot := filepath.Join(gosrc, "v2ray.com", "core")
|
|
|
-
|
|
|
- filepath.Walk(reporoot, func(path string, info os.FileInfo, err error) error {
|
|
|
- if err != nil {
|
|
|
- fmt.Println(err)
|
|
|
- return err
|
|
|
- }
|
|
|
-
|
|
|
- if info.IsDir() {
|
|
|
- return nil
|
|
|
- }
|
|
|
-
|
|
|
- dir := filepath.Dir(path)
|
|
|
- filename := filepath.Base(path)
|
|
|
- if strings.HasSuffix(filename, ".proto") {
|
|
|
- protofiles[dir] = append(protofiles[dir], path)
|
|
|
- }
|
|
|
-
|
|
|
- return nil
|
|
|
- })
|
|
|
-
|
|
|
- for _, files := range protofiles {
|
|
|
- args := []string{"--proto_path", gosrc, "--go_out", gosrc, "--java_out", sdkPath("java")}
|
|
|
- args = append(args, files...)
|
|
|
- cmd := exec.Command(protoc, args...)
|
|
|
- cmd.Env = append(cmd.Env, os.Environ()...)
|
|
|
- output, err := cmd.CombinedOutput()
|
|
|
- if len(output) > 0 {
|
|
|
- fmt.Println(string(output))
|
|
|
- }
|
|
|
- if err != nil {
|
|
|
- fmt.Println(err)
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- common.Must(filepath.Walk(reporoot, func(path string, info os.FileInfo, err error) error {
|
|
|
- if err != nil {
|
|
|
- fmt.Println(err)
|
|
|
- return err
|
|
|
- }
|
|
|
-
|
|
|
- if info.IsDir() {
|
|
|
- return nil
|
|
|
- }
|
|
|
-
|
|
|
- if !strings.HasSuffix(info.Name(), ".pb.go") {
|
|
|
- return nil
|
|
|
- }
|
|
|
-
|
|
|
- content, err := ioutil.ReadFile(path)
|
|
|
- if err != nil {
|
|
|
- return err
|
|
|
- }
|
|
|
- pos := bytes.Index(content, []byte("\npackage"))
|
|
|
- if pos > 0 {
|
|
|
- if err := ioutil.WriteFile(path, content[pos+1:], info.Mode()); err != nil {
|
|
|
- return err
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- return nil
|
|
|
- }))
|
|
|
-}
|