/
plgo.go
93 lines (88 loc) · 1.79 KB
/
plgo.go
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
package main
import (
"flag"
"fmt"
"log"
"os"
"os/exec"
"path/filepath"
"runtime"
)
func printUsage() {
fmt.Println(`Usage: plgo [-v] [path/to/package]`)
flag.PrintDefaults()
}
func buildPackage(buildPath, packageName string) error {
if err := os.Setenv("CGO_LDFLAGS_ALLOW", "-shared"); err != nil {
return err
}
switchx := "-v" // substitutor
if verbose {
switchx = "-x"
}
fileExt := ".so"
if runtime.GOOS == "windows" {
fileExt = ".dll"
}
goBuild := exec.Command("go", "build", switchx,
"-buildmode=c-shared",
"-o", filepath.Join("build", packageName+fileExt),
filepath.Join(buildPath, "package.go"),
filepath.Join(buildPath, "methods.go"),
filepath.Join(buildPath, "pl.go"),
)
goBuild.Stdout = os.Stdout
goBuild.Stderr = os.Stderr
if err := goBuild.Run(); err != nil {
return fmt.Errorf("Cannot build package: %s", err)
}
return nil
}
var verbose bool
func main() {
flag.BoolVar(&verbose, "v", false, "be verbose, 'go build -x'")
flag.Parse()
packagePath := "."
if len(flag.Args()) == 1 {
packagePath = flag.Arg(0)
}
moduleWriter, err := NewModuleWriter(packagePath)
if err != nil {
fmt.Println(err)
printUsage()
return
}
tempPackagePath, err := moduleWriter.WriteModule()
if err != nil {
fmt.Println(err)
return
}
log.Println(tempPackagePath)
if _, err = os.Stat("build"); os.IsNotExist(err) {
err = os.Mkdir("build", 0744)
if err != nil {
fmt.Println(err)
return
}
}
err = buildPackage(tempPackagePath, moduleWriter.PackageName)
if err != nil {
fmt.Println(err)
return
}
err = moduleWriter.WriteSQL("build")
if err != nil {
fmt.Println(err)
return
}
err = moduleWriter.WriteControl("build")
if err != nil {
fmt.Println(err)
return
}
err = moduleWriter.WriteMakefile("build")
if err != nil {
fmt.Println(err)
return
}
}