cmd/llgo
This commit is contained in:
xushiwei
114
cmd/internal/base/base.go
Normal file
114
cmd/internal/base/base.go
Normal file
@@ -0,0 +1,114 @@
|
||||
/*
|
||||
* Copyright (c) 2023 The GoPlus Authors (goplus.org). All rights reserved.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
// Package base defines shared basic pieces of the llgo command,
|
||||
// in particular logging and the Command structure.
|
||||
package base
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// A Command is an implementation of a gop command
|
||||
// like gop export or gop install.
|
||||
type Command struct {
|
||||
// Run runs the command.
|
||||
// The args are the arguments after the command name.
|
||||
Run func(cmd *Command, args []string)
|
||||
|
||||
// UsageLine is the one-line usage message.
|
||||
// The words between "gop" and the first flag or argument in the line are taken to be the command name.
|
||||
UsageLine string
|
||||
|
||||
// Short is the short description shown in the 'gop help' output.
|
||||
Short string
|
||||
|
||||
// Flag is a set of flags specific to this command.
|
||||
Flag flag.FlagSet
|
||||
|
||||
// Commands lists the available commands and help topics.
|
||||
// The order here is the order in which they are printed by 'gop help'.
|
||||
// Note that subcommands are in general best avoided.
|
||||
Commands []*Command
|
||||
}
|
||||
|
||||
// Llgo command
|
||||
var Llgo = &Command{
|
||||
UsageLine: "llgo",
|
||||
Short: `llgo is a Go compiler based on LLVM in order to better integrate Go with the C ecosystem.`,
|
||||
// Commands initialized in package main
|
||||
}
|
||||
|
||||
// LongName returns the command's long name: all the words in the usage line between "gop" and a flag or argument,
|
||||
func (c *Command) LongName() string {
|
||||
name := c.UsageLine
|
||||
if i := strings.Index(name, " ["); i >= 0 {
|
||||
name = name[:i]
|
||||
}
|
||||
if name == "llgo" {
|
||||
return ""
|
||||
}
|
||||
return strings.TrimPrefix(name, "llgo ")
|
||||
}
|
||||
|
||||
// Name returns the command's short name: the last word in the usage line before a flag or argument.
|
||||
func (c *Command) Name() string {
|
||||
name := c.LongName()
|
||||
if i := strings.LastIndex(name, " "); i >= 0 {
|
||||
name = name[i+1:]
|
||||
}
|
||||
return name
|
||||
}
|
||||
|
||||
// Usage show the command usage.
|
||||
func (c *Command) Usage(w io.Writer) {
|
||||
fmt.Fprintf(w, "%s\n\nUsage: %s\n", c.Short, c.UsageLine)
|
||||
|
||||
// restore output of flag
|
||||
defer c.Flag.SetOutput(c.Flag.Output())
|
||||
|
||||
c.Flag.SetOutput(w)
|
||||
c.Flag.PrintDefaults()
|
||||
fmt.Fprintln(w)
|
||||
os.Exit(2)
|
||||
}
|
||||
|
||||
// Runnable reports whether the command can be run; otherwise
|
||||
// it is a documentation pseudo-command.
|
||||
func (c *Command) Runnable() bool {
|
||||
return c.Run != nil
|
||||
}
|
||||
|
||||
// Usage is the usage-reporting function, filled in by package main
|
||||
// but here for reference by other packages.
|
||||
//
|
||||
// flag.Usage func()
|
||||
|
||||
// CmdName - "build", "install", "list", "mod tidy", etc.
|
||||
var CmdName string
|
||||
|
||||
// Main runs a command.
|
||||
func Main(c *Command, app string, args []string) {
|
||||
name := c.UsageLine
|
||||
if i := strings.Index(name, " ["); i >= 0 {
|
||||
c.UsageLine = app + name[i:]
|
||||
}
|
||||
c.Run(c, args)
|
||||
}
|
||||
106
cmd/internal/build/build.go
Normal file
106
cmd/internal/build/build.go
Normal file
@@ -0,0 +1,106 @@
|
||||
/*
|
||||
* Copyright (c) 2023 The GoPlus Authors (goplus.org). All rights reserved.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
// Package build implements the “llgo build” command.
|
||||
package build
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"reflect"
|
||||
|
||||
"github.com/goplus/gop"
|
||||
"github.com/goplus/gop/x/gopprojs"
|
||||
"github.com/goplus/llgo"
|
||||
"github.com/goplus/llgo/cmd/internal/base"
|
||||
"github.com/goplus/llgo/x/gocmd"
|
||||
)
|
||||
|
||||
// llgo build
|
||||
var Cmd = &base.Command{
|
||||
UsageLine: "llog build [flags] [packages]",
|
||||
Short: "Build Go files",
|
||||
}
|
||||
|
||||
var (
|
||||
flagOutput = flag.String("o", "", "build output file")
|
||||
_ = flag.Bool("v", false, "print verbose information")
|
||||
flag = &Cmd.Flag
|
||||
)
|
||||
|
||||
func init() {
|
||||
Cmd.Run = runCmd
|
||||
}
|
||||
|
||||
func runCmd(cmd *base.Command, args []string) {
|
||||
err := flag.Parse(args)
|
||||
if err != nil {
|
||||
log.Panicln("parse input arguments failed:", err)
|
||||
}
|
||||
|
||||
args = flag.Args()
|
||||
if len(args) == 0 {
|
||||
args = []string{"."}
|
||||
}
|
||||
|
||||
proj, args, err := gopprojs.ParseOne(args...)
|
||||
if err != nil {
|
||||
log.Panicln(err)
|
||||
}
|
||||
if len(args) != 0 {
|
||||
log.Panicln("too many arguments:", args)
|
||||
}
|
||||
|
||||
conf := &llgo.Config{}
|
||||
confCmd := &gocmd.BuildConfig{}
|
||||
if *flagOutput != "" {
|
||||
output, err := filepath.Abs(*flagOutput)
|
||||
if err != nil {
|
||||
log.Panicln(err)
|
||||
}
|
||||
confCmd.Output = output
|
||||
}
|
||||
build(proj, conf, confCmd)
|
||||
}
|
||||
|
||||
func build(proj gopprojs.Proj, conf *llgo.Config, build *gocmd.BuildConfig) {
|
||||
var obj string
|
||||
var err error
|
||||
switch v := proj.(type) {
|
||||
case *gopprojs.DirProj:
|
||||
obj = v.Dir
|
||||
err = llgo.BuildDir(obj, conf, build)
|
||||
case *gopprojs.PkgPathProj:
|
||||
obj = v.Path
|
||||
err = llgo.BuildPkgPath("", v.Path, conf, build)
|
||||
case *gopprojs.FilesProj:
|
||||
err = llgo.BuildFiles(v.Files, conf, build)
|
||||
default:
|
||||
log.Panicln("`llgo build` doesn't support", reflect.TypeOf(v))
|
||||
}
|
||||
if gop.NotFound(err) {
|
||||
fmt.Fprintf(os.Stderr, "llgo build %v: not found\n", obj)
|
||||
} else if err != nil {
|
||||
fmt.Fprintln(os.Stderr, err)
|
||||
} else {
|
||||
return
|
||||
}
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// -----------------------------------------------------------------------------
|
||||
123
cmd/internal/help/help.go
Normal file
123
cmd/internal/help/help.go
Normal file
@@ -0,0 +1,123 @@
|
||||
/*
|
||||
* Copyright (c) 2023 The GoPlus Authors (goplus.org). All rights reserved.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
// Package help implements the “llgo help” command.
|
||||
package help
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"os"
|
||||
"strings"
|
||||
"text/template"
|
||||
"unicode"
|
||||
"unicode/utf8"
|
||||
|
||||
"github.com/goplus/llgo/cmd/internal/base"
|
||||
)
|
||||
|
||||
// Help implements the 'help' command.
|
||||
func Help(w io.Writer, args []string) {
|
||||
cmd := base.Llgo
|
||||
Args:
|
||||
for i, arg := range args {
|
||||
for _, sub := range cmd.Commands {
|
||||
if sub.Name() == arg {
|
||||
cmd = sub
|
||||
continue Args
|
||||
}
|
||||
}
|
||||
|
||||
// helpSuccess is the help command using as many args as possible that would succeed.
|
||||
helpSuccess := "llgo help"
|
||||
if i > 0 {
|
||||
helpSuccess += " " + strings.Join(args[:i], " ")
|
||||
}
|
||||
fmt.Fprintf(os.Stderr, "llgo help %s: unknown help topic. Run '%s'.\n", strings.Join(args, " "), helpSuccess)
|
||||
os.Exit(2)
|
||||
}
|
||||
|
||||
if len(cmd.Commands) > 0 {
|
||||
PrintUsage(w, cmd)
|
||||
} else {
|
||||
cmd.Usage(w)
|
||||
}
|
||||
// not exit 2: succeeded at 'llgo help cmd'.
|
||||
}
|
||||
|
||||
var usageTemplate = `{{.Short | trim}}
|
||||
|
||||
Usage:
|
||||
|
||||
{{.UsageLine}} <command> [arguments]
|
||||
|
||||
The commands are:
|
||||
{{range .Commands}}{{if or (.Runnable) .Commands}}
|
||||
{{.Name | printf "%-11s"}} {{.Short}}{{end}}{{end}}
|
||||
|
||||
Use "llgo help{{with .LongName}} {{.}}{{end}} <command>" for more information about a command.
|
||||
|
||||
`
|
||||
|
||||
// An errWriter wraps a writer, recording whether a write error occurred.
|
||||
type errWriter struct {
|
||||
w io.Writer
|
||||
err error
|
||||
}
|
||||
|
||||
func (w *errWriter) Write(b []byte) (int, error) {
|
||||
n, err := w.w.Write(b)
|
||||
if err != nil {
|
||||
w.err = err
|
||||
}
|
||||
return n, err
|
||||
}
|
||||
|
||||
// tmpl executes the given template text on data, writing the result to w.
|
||||
func tmpl(w io.Writer, text string, data interface{}) {
|
||||
t := template.New("top")
|
||||
t.Funcs(template.FuncMap{"trim": strings.TrimSpace, "capitalize": capitalize})
|
||||
template.Must(t.Parse(text))
|
||||
ew := &errWriter{w: w}
|
||||
err := t.Execute(ew, data)
|
||||
if ew.err != nil {
|
||||
// I/O error writing. Ignore write on closed pipe.
|
||||
if strings.Contains(ew.err.Error(), "pipe") {
|
||||
os.Exit(1)
|
||||
}
|
||||
log.Fatalf("writing output: %v", ew.err)
|
||||
}
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
|
||||
func capitalize(s string) string {
|
||||
if s == "" {
|
||||
return s
|
||||
}
|
||||
r, n := utf8.DecodeRuneInString(s)
|
||||
return string(unicode.ToTitle(r)) + s[n:]
|
||||
}
|
||||
|
||||
// PrintUsage prints usage information.
|
||||
func PrintUsage(w io.Writer, cmd *base.Command) {
|
||||
bw := bufio.NewWriter(w)
|
||||
tmpl(bw, usageTemplate, cmd)
|
||||
bw.Flush()
|
||||
}
|
||||
91
cmd/llgo/llgo.go
Normal file
91
cmd/llgo/llgo.go
Normal file
@@ -0,0 +1,91 @@
|
||||
/*
|
||||
* Copyright (c) 2023 The GoPlus Authors (goplus.org). All rights reserved.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"os"
|
||||
"strings"
|
||||
|
||||
"github.com/qiniu/x/log"
|
||||
|
||||
"github.com/goplus/llgo/cmd/internal/base"
|
||||
"github.com/goplus/llgo/cmd/internal/build"
|
||||
"github.com/goplus/llgo/cmd/internal/help"
|
||||
)
|
||||
|
||||
func mainUsage() {
|
||||
help.PrintUsage(os.Stderr, base.Llgo)
|
||||
os.Exit(2)
|
||||
}
|
||||
|
||||
func init() {
|
||||
flag.Usage = mainUsage
|
||||
base.Llgo.Commands = []*base.Command{
|
||||
build.Cmd,
|
||||
}
|
||||
}
|
||||
|
||||
func main() {
|
||||
flag.Parse()
|
||||
args := flag.Args()
|
||||
if len(args) < 1 {
|
||||
flag.Usage()
|
||||
}
|
||||
log.SetFlags(log.Ldefault &^ log.LstdFlags)
|
||||
|
||||
base.CmdName = args[0] // for error messages
|
||||
if args[0] == "help" {
|
||||
help.Help(os.Stderr, args[1:])
|
||||
return
|
||||
}
|
||||
|
||||
BigCmdLoop:
|
||||
for bigCmd := base.Llgo; ; {
|
||||
for _, cmd := range bigCmd.Commands {
|
||||
if cmd.Name() != args[0] {
|
||||
continue
|
||||
}
|
||||
args = args[1:]
|
||||
if len(cmd.Commands) > 0 {
|
||||
bigCmd = cmd
|
||||
if len(args) == 0 {
|
||||
help.PrintUsage(os.Stderr, bigCmd)
|
||||
os.Exit(2)
|
||||
}
|
||||
if args[0] == "help" {
|
||||
help.Help(os.Stderr, append(strings.Split(base.CmdName, " "), args[1:]...))
|
||||
return
|
||||
}
|
||||
base.CmdName += " " + args[0]
|
||||
continue BigCmdLoop
|
||||
}
|
||||
if !cmd.Runnable() {
|
||||
continue
|
||||
}
|
||||
cmd.Run(cmd, args)
|
||||
return
|
||||
}
|
||||
helpArg := ""
|
||||
if i := strings.LastIndex(base.CmdName, " "); i >= 0 {
|
||||
helpArg = " " + base.CmdName[:i]
|
||||
}
|
||||
fmt.Fprintf(os.Stderr, "llgo %s: unknown command\nRun 'llgo help%s' for usage.\n", base.CmdName, helpArg)
|
||||
os.Exit(2)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user