mirror of
https://github.com/micromdm/micromdm/
synced 2026-08-14 09:25:52 +08:00
111 lines
2.2 KiB
Go
111 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/go-kit/kit/log"
|
|
"github.com/micromdm/micromdm/core/apply"
|
|
)
|
|
|
|
type applyCommand struct {
|
|
config *ClientConfig
|
|
applysvc apply.Service
|
|
}
|
|
|
|
func (cmd *applyCommand) setup() error {
|
|
cfg, err := LoadClientConfig()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
cmd.config = cfg
|
|
logger := log.NewLogfmtLogger(os.Stderr)
|
|
applysvc, err := apply.NewClient(cfg.ServerURL, logger, cfg.APIToken)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
cmd.applysvc = applysvc
|
|
return nil
|
|
}
|
|
|
|
func (cmd *applyCommand) Run(args []string) error {
|
|
if len(args) < 1 {
|
|
cmd.Usage()
|
|
os.Exit(1)
|
|
}
|
|
if err := cmd.setup(); err != nil {
|
|
return err
|
|
}
|
|
var run func([]string) error
|
|
switch strings.ToLower(args[0]) {
|
|
case "blueprints":
|
|
run = cmd.applyBlueprint
|
|
case "dep-tokens":
|
|
run = cmd.applyDEPTokens
|
|
default:
|
|
cmd.Usage()
|
|
os.Exit(1)
|
|
}
|
|
return run(args[1:])
|
|
}
|
|
|
|
func (cmd *applyCommand) Usage() error {
|
|
const applyUsage = `
|
|
Apply a resource.
|
|
|
|
Valid resource types:
|
|
|
|
* blueprints
|
|
* dep-tokens
|
|
|
|
Examples:
|
|
# Get a list of devices
|
|
mdmctl apply blueprints -f /path/to/blueprint.json
|
|
|
|
`
|
|
fmt.Println(applyUsage)
|
|
return nil
|
|
}
|
|
|
|
func (cmd *applyCommand) applyBlueprint(args []string) error {
|
|
flagset := flag.NewFlagSet("blueprints", flag.ExitOnError)
|
|
flagset.Usage = usageFor(flagset, "mdmctl apply blueprints [flags]")
|
|
if err := flagset.Parse(args); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (cmd *applyCommand) applyDEPTokens(args []string) error {
|
|
flagset := flag.NewFlagSet("dep-tokens", flag.ExitOnError)
|
|
var (
|
|
flPublicKeyPath = flagset.String("import-token", "", "filename of p7m encrypted token file (downloaded from DEP portal)")
|
|
)
|
|
flagset.Usage = usageFor(flagset, "mdmctl apply dep-tokens [flags]")
|
|
if err := flagset.Parse(args); err != nil {
|
|
return err
|
|
}
|
|
if *flPublicKeyPath == "" {
|
|
return errors.New("must provide -import-token parameter")
|
|
}
|
|
if _, err := os.Stat(*flPublicKeyPath); os.IsNotExist(err) {
|
|
return err
|
|
}
|
|
p7mBytes, err := ioutil.ReadFile(*flPublicKeyPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
ctx := context.Background()
|
|
err = cmd.applysvc.ApplyDEPToken(ctx, p7mBytes)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fmt.Println("imported DEP token")
|
|
return nil
|
|
}
|