main.go

// Copyright (c) 2016 Pulcy.
//
// 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 (
	"fmt"
	"net/url"
	"os"

	"github.com/op/go-logging"
	"github.com/spf13/cobra"

	"github.com/pulcy/tranquility/fleet"
)

const (
	projectName = "tranquility"

	defaultLogLevel      = "info"
	defaultFleetEndpoint = "unix:///var/run/fleet.sock"
)

var (
	projectVersion = "dev"
	projectBuild   = "dev"
)

var (
	cmdMain = &cobra.Command{
		Use: projectName,
		Run: showUsage,
		PersistentPreRun: func(*cobra.Command, []string) {
			setLogLevel(globalFlags.logLevel, defaultLogLevel, projectName)
		},
	}
	globalFlags struct {
		logLevel      string
		fleetEndpoint string
		dryRun        bool
	}
	log *logging.Logger
)

func init() {
	log = logging.MustGetLogger(projectName)
	cmdMain.PersistentFlags().StringVar(&globalFlags.logLevel, "log-level", defaultLogLevel, "Log level (debug|info|warning|error)")
	cmdMain.PersistentFlags().StringVar(&globalFlags.fleetEndpoint, "fleet-endpoint", defaultFleetEndpoint, "URL used to connect to fleet")
	cmdMain.PersistentFlags().BoolVar(&globalFlags.dryRun, "dryrun", false, "Do not make changed")
}

func main() {
	cmdMain.AddCommand(checkCmd)
	cmdMain.AddCommand(fixCmd)

	cmdMain.Execute()
}

func showUsage(cmd *cobra.Command, args []string) {
	cmd.Usage()
}

func Exitf(format string, args ...interface{}) {
	fmt.Printf(format, args...)
	fmt.Println()
	os.Exit(1)
}

func assert(err error) {
	if err != nil {
		Exitf("Assertion failed: %#v", err)
	}
}

func setLogLevel(logLevel, defaultLogLevel, logName string) {
	if logLevel == "" {
		logLevel = defaultLogLevel
	}
	level, err := logging.LogLevel(logLevel)
	if err != nil {
		Exitf("Invalid log-level '%s': %#v", logLevel, err)
	}
	logging.SetLevel(level, logName)
}

func newFleet() (*fleet.Fleet, error) {
	ep, err := url.Parse(globalFlags.fleetEndpoint)
	if err != nil {
		return nil, maskAny(err)
	}
	result, err := fleet.NewFleet(*ep, log)
	if err != nil {
		return nil, maskAny(err)
	}
	return result, nil
}