open-hardware-monitor-client/lib/app/app.go

package app

import (
	"fmt"
	"io"
	"log"
	"net/http"
	"net/url"
	"os"

	"git.xbudex.com/buddy/open-hardware-monitor/lib/client"
	"git.xbudex.com/buddy/open-hardware-monitor/lib/metrics"
	"github.com/prometheus/client_golang/prometheus"
	"github.com/prometheus/client_golang/prometheus/promhttp"
	"github.com/urfave/cli"
)

// App context
type App struct {
	cli    *cli.App
	client *client.Client
	out    io.Writer
}

var (
	// FlagHost flag for host
	FlagHost = cli.StringFlag{
		Name:   "host",
		EnvVar: "OHWM_HOST",
		Value:  "127.0.0.1:8085",
		Usage:  "open hardware monitor host [OSHW_HOST]",
	}

	// FlagScheme flag for host
	FlagScheme = cli.StringFlag{
		Name:   "scheme",
		EnvVar: "OHWM_SCHEME",
		Value:  "http",
		Usage:  "open hardware monitor scheme (http/https)",
	}

	// FlagPath flag for path url
	FlagPath = cli.StringFlag{
		Name:   "path",
		EnvVar: "OHWM_PATH",
		Value:  "/data.json",
		Usage:  "open hardware monitor path (ie /data.json)",
	}

	// FlagInterface flag for host
	FlagInterface = cli.StringFlag{
		Name:   "http",
		EnvVar: "OHWM_HTTP",
		Value:  ":9200",
		Usage:  "interface to serve http requests",
	}
)

// New returns instance of an app
func New() *App {
	cliapp := cli.NewApp()
	app := &App{
		out: os.Stdout,
		cli: cliapp,
	}

	app.cli.Name = "ohwm-client"
	app.cli.Description = "Open Hardware Manager Client"
	app.cli.UsageText = "ohwm-client [flags] <command> [flags]"
	app.cli.HideVersion = true

	app.cli.Flags = []cli.Flag{
		FlagHost,
		FlagScheme,
		FlagPath,
	}

	app.cli.Commands = []cli.Command{
		cli.Command{
			Name:   "print",
			Usage:  "pretty prints api result",
			Action: app.actionPrint,
		},
		cli.Command{
			Name:   "export",
			Usage:  "runs http exporter for prometheus",
			Action: app.actionMetrics,
			Flags: []cli.Flag{
				FlagInterface,
			},
		},
	}

	app.cli.Before = app.before
	return app
}

// Run runs cli action
func (a *App) Run(args []string) error {
	return a.cli.Run(args)
}

// Before sets up app
func (a *App) before(ctx *cli.Context) error {
	a.client = &client.Client{
		URL: url.URL{
			Scheme: ctx.String("scheme"),
			Host:   ctx.String("host"),
			Path:   ctx.String("path"),
		},
	}
	return nil
}

func (a *App) actionPrint(_ *cli.Context) error {
	node, err := a.client.Fetch()
	if err != nil {
		return err
	}
	fmt.Fprint(a.out, node.Stringify())
	return nil
}

func (a *App) actionMetrics(ctx *cli.Context) error {
	addr := ctx.String("http")
	http.HandleFunc("/metrics", func(resp http.ResponseWriter, req *http.Request) {
		log.Println("GET /metrics")
		root, err := a.client.Fetch()
		if err != nil {
			return
		}
		vals, err := root.Values()
		if err != nil {
			return
		}
		collection := metrics.FromVals(vals)
		registry := prometheus.NewRegistry()
		if err := registry.Register(collection); err != nil {
			return
		}

		handler := promhttp.HandlerFor(registry, promhttp.HandlerOpts{
			Registry: registry,
		})

		handler.ServeHTTP(resp, req)
		registry.Unregister(collection)
	})
	log.Println("listen on: " + addr)
	return http.ListenAndServe(addr, nil)
}