summaryrefslogtreecommitdiff
path: root/protoc.go
blob: acd6ad77148f7060f0ca55c9d49958f3fcae1c26 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
package main

// auto run protoc with the correct args

import (
	"errors"
	"os"
	"path/filepath"
	"strings"

	"go.wit.com/lib/gui/shell"
	"go.wit.com/lib/protobuf/forgepb"
	"go.wit.com/log"
)

// test.pb.go: test.proto
// 	cd ~/go/src && protoc --go_out=. --proto_path=go.wit.com/apps/autogenpb/testautogen \
// 		--go_opt=Mtest.proto=go.wit.com/apps/autogenpb/testautogen \
// 		test.proto

// forgeConfig.pb.go: forgeConfig.proto
// 	cd ~/go/src && protoc --go_out=. --proto_path=go.wit.com/apps/autogenpb/testautogen \
// 		--go_opt=MforgeConfig.proto=go.wit.com/apps/autogenpb/testautogen \
// 		forgeConfig.proto

func protocBuild(names map[string]string) error {
	// read in the .proto file
	data, err := os.ReadFile(names["protofile"])
	if err != nil {
		// log.Info("open config file :", err)
		return err
	}

	// have to figure out how to run protoc so initialize forge
	forge = forgepb.Init()
	// forge.ConfigPrintTable()
	os.Setenv("REPO_WORK_PATH", forge.GetGoSrc())

	log.Info("")
	if shell.Exists(names["protoc"]) {
		log.Info("protoc file already created", names["protoc"])
		// return nil
	}
	log.Info("make protoc file:", names["protoc"])
	log.Info("go src", forge.GetGoSrc())
	pwd, _ := os.Getwd()
	log.Info("go.Getwd()", pwd)
	if !strings.HasPrefix(pwd, forge.GetGoSrc()) {
		return errors.New("paths don't match")
	}
	gopath := strings.TrimPrefix(pwd, forge.GetGoSrc())
	gopath = strings.Trim(gopath, "/")
	log.Info("gopath", gopath)
	cmd := []string{"protoc", "--go_out=."}
	cmd = append(cmd, "--proto_path="+gopath)
	cmd = append(cmd, "--go_opt=M"+names["protofile"]+"="+gopath)

	// look for included proto files
	lines := strings.Split(string(data), "\n")
	for _, line := range lines {
		// log.Info("line:", line)
		if strings.HasPrefix(line, "import ") {
			parts := strings.Split(line, "\"")
			protofile := parts[1]
			if shell.Exists(protofile) {
				log.Info("adding import", protofile)
				cmd = append(cmd, "--go_opt=M"+protofile+"="+gopath)
			} else {
				basepath, pname := filepath.Split(protofile)
				if basepath == "" {
					log.Warn("import line:", line, "missing proto file:", pname)
					log.Warn("protoc will probably fail here")
				} else {
					log.Warn("need to check basepath here", basepath, pname)
				}
			}
		}
		/*
			start := "type " + names["Bases"] + " struct {"
			if strings.HasSuffix(line, start) {
				found = true
				log.Info("FOUND line:", line)
				fmt.Fprintln(w, line)
				fmt.Fprintln(w, "\tsync.RWMutex // auto-added by go.wit.com/apps/autogenpb")
				fmt.Fprintln(w, "")
			} else {
				fmt.Fprintln(w, line)
			}
		*/
	}

	cmd = append(cmd, names["protofile"])
	log.Info("\tpwd", forge.GetGoSrc())
	for i, s := range cmd {
		log.Info("\t", i, s)
	}
	shell.PathRun(forge.GetGoSrc(), cmd)
	return nil
}