1d3f2dbe3c
As it turns out, there are other helper packages out there. I wrote this before I realized that, and it didn't take me that long anyways.
126 lines
3.7 KiB
Go
126 lines
3.7 KiB
Go
// Mgmt
|
|
// Copyright (C) 2013-2024+ James Shubin and the project contributors
|
|
// Written by James Shubin <james@shubin.ca> and the project contributors
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
//
|
|
// Additional permission under GNU GPL version 3 section 7
|
|
//
|
|
// If you modify this program, or any covered work, by linking or combining it
|
|
// with embedded mcl code and modules (and that the embedded mcl code and
|
|
// modules which link with this program, contain a copy of their source code in
|
|
// the authoritative form) containing parts covered by the terms of any other
|
|
// license, the licensors of this program grant you additional permission to
|
|
// convey the resulting work. Furthermore, the licensors of this program grant
|
|
// the original author, James Shubin, additional permission to update this
|
|
// additional permission if he deems it necessary to achieve the goals of this
|
|
// additional permission.
|
|
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
_ "embed"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/purpleidea/mgmt/cli"
|
|
cliUtil "github.com/purpleidea/mgmt/cli/util"
|
|
"github.com/purpleidea/mgmt/entry"
|
|
_ "github.com/purpleidea/mgmt/gapi/empty" // import so the gapi registers
|
|
_ "github.com/purpleidea/mgmt/lang/gapi" // import so the gapi registers
|
|
_ "github.com/purpleidea/mgmt/puppet" // import so the gapi registers
|
|
_ "github.com/purpleidea/mgmt/puppet/langpuppet" // import so the gapi registers
|
|
"github.com/purpleidea/mgmt/util/pprof"
|
|
_ "github.com/purpleidea/mgmt/yamlgraph" // import so the gapi registers
|
|
"go.etcd.io/etcd/server/v3/etcdmain"
|
|
)
|
|
|
|
// These constants are some global variables that are used throughout the code.
|
|
const (
|
|
tagline = "next generation config management"
|
|
debug = false // add additional log messages
|
|
)
|
|
|
|
// set at compile time
|
|
var (
|
|
program string
|
|
version string
|
|
)
|
|
|
|
//go:embed COPYING
|
|
var copying string
|
|
|
|
func main() {
|
|
// embed etcd completely
|
|
if len(os.Args) >= 2 && os.Args[1] == "etcd" {
|
|
args := []string{}
|
|
for i, s := range os.Args {
|
|
if i == 0 { // pop off our argv[0] and let `etcd` be it
|
|
continue
|
|
}
|
|
args = append(args, s)
|
|
}
|
|
etcdmain.Main(args) // this will os.Exit
|
|
return // for safety
|
|
}
|
|
|
|
// Run profiling if it's activated.
|
|
// TODO: Should we pass a logger into this?
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
if err := pprof.Run(ctx); err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
//return // redundant
|
|
}
|
|
defer cancel()
|
|
|
|
cliUtil.LogSetup(debug)
|
|
data := &cliUtil.Data{
|
|
Program: program,
|
|
Version: version,
|
|
Copying: copying,
|
|
Tagline: tagline,
|
|
Flags: cliUtil.Flags{
|
|
Debug: debug,
|
|
Logf: func(format string, v ...interface{}) {
|
|
log.Printf(format, v...) // the top-level log!
|
|
},
|
|
},
|
|
Args: os.Args,
|
|
}
|
|
|
|
name := ""
|
|
if len(data.Args) > 1 {
|
|
name = data.Args[1]
|
|
}
|
|
// is there an alternate entry point for the cli?
|
|
if cli, err := entry.Lookup(name); err == nil && cli.Name() == name {
|
|
data.Args = data.Args[1:] // pop off "argv[0]"
|
|
if err := cli.CLI(context.Background(), data); err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
//return // redundant
|
|
}
|
|
return
|
|
}
|
|
|
|
if err := cli.CLI(context.Background(), data); err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
return
|
|
}
|
|
}
|