40 lines
567 B
Go
40 lines
567 B
Go
|
package multitool
|
||
|
|
|
||
|
|
import(
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"path"
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
type Tools map[string] func(args []string)
|
||
|
|
|
||
|
|
func Main(name string, m Tools) {
|
||
|
|
var(
|
||
|
|
utilName string
|
||
|
|
args []string
|
||
|
|
)
|
||
|
|
|
||
|
|
if binBase := path.Base(os.Args[0]) ; binBase != name {
|
||
|
|
utilName = binBase
|
||
|
|
args = os.Args[:]
|
||
|
|
} else {
|
||
|
|
if len(os.Args)<2 {
|
||
|
|
for k, _ := range m {
|
||
|
|
fmt.Printf("%s\n", k)
|
||
|
|
}
|
||
|
|
os.Exit(0)
|
||
|
|
}
|
||
|
|
utilName = os.Args[1]
|
||
|
|
args = os.Args[1:]
|
||
|
|
}
|
||
|
|
|
||
|
|
if _, ok := m[utilName] ; !ok {
|
||
|
|
fmt.Printf("%s: No such uitl as '%s'.\n", os.Args[0], utilName )
|
||
|
|
os.Exit(1)
|
||
|
|
}
|
||
|
|
|
||
|
|
m[utilName](args)
|
||
|
|
}
|
||
|
|
|