main.go

package main

import (
	"crypto/tls"
	"fmt"
	log "github.com/Sirupsen/logrus"
	"github.com/marthjod/ansible-one-inventory/config"
	"github.com/marthjod/ansible-one-inventory/discovery"
	"github.com/marthjod/ansible-one-inventory/filter"
	"github.com/marthjod/ansible-one-inventory/hostnameextractor"
	"github.com/marthjod/gocart/api"
	"github.com/marthjod/gocart/vmpool"
	flag "github.com/ogier/pflag"
	"net/http"
	"os"
	"path"
)

const (
	configFile = "opennebula-inventory.yaml"
)

func main() {
	var (
		host      = flag.String("host", "", "")
		list      = flag.Bool("list", true, "")
		debug     = flag.Bool("debug", false, "Enable debug output.")
		extractor hostnameextractor.HostnameExtractor
	)

	flag.Parse()

	if *debug {
		log.SetLevel(log.DebugLevel)
	}

	confPath := path.Dir(os.Args[0]) + "/" + configFile
	log.Debug("Loading config ", confPath)
	f, err := os.Open(confPath)
	if err != nil {
		log.Fatal(err.Error())
	}
	defer f.Close()
	conf, err := config.FromFile(f)
	if err != nil {
		log.Error(err.Error())
		log.Fatalf("Config file %s missing (expected in same directory).", configFile)
	}

	log.Debug("Accessing API at ", conf.Url)
	apiClient, err := api.NewClient(conf.Url, conf.Username, conf.Password, &http.Transport{
		TLSClientConfig: &tls.Config{InsecureSkipVerify: conf.SslSkipVerify},
	})
	if err != nil {
		log.Fatal(err.Error())
	}

	log.Debug("Fetching VM pool")
	vmPool := vmpool.NewVmPool()
	if err := apiClient.Call(vmPool); err != nil {
		log.Fatal(err.Error())
	}

	extractor = &hostnameextractor.VmNameExtractor{}
	if conf.HostnameFieldInUserTemplate != "" {
		extractor = &hostnameextractor.UserTemplateExtractor{
			Field: conf.HostnameFieldInUserTemplate,
		}
	}

	hostNames := discovery.GetHostnames(vmPool, extractor)
	log.Debugf("Hostnames: %v", hostNames)

	groupFilters := conf.StaticGroupFilters

	if conf.DynamicGroupFilters.Pattern != "" {
		distinctPatterns := vmPool.GetDistinctVmNamePatternsExtractHostname(
			conf.DynamicGroupFilters.Pattern, conf.DynamicGroupFilters.Prefix, conf.DynamicGroupFilters.Infix,
			conf.DynamicGroupFilters.Suffix, extractor.Extract)

		log.Debugf("Distinct patterns: %+v", distinctPatterns)

		for pattern, _ := range distinctPatterns {
			groupFilters[filter.AdjustPatternName(pattern, conf.DynamicGroupFilters.PatternReplace)] = pattern
		}
	}

	log.Debugf("Group filters: %+v", groupFilters)
	inventory := discovery.GetInventoryGroups(hostNames, groupFilters)

	if *host != "" {
		log.Error("--host option not implemented yet")
		os.Exit(1)
	} else if *list {
		log.Debug("Writing inventory")
		fmt.Println(inventory.Json())
	}

}