authd.go

package main

import (
	"flag"
	"log"
	"net"

	"github.com/cockroachdb/cmux"
	"github.com/dhtech/authservice/audit"
	"github.com/dhtech/authservice/rpc"
	"github.com/dhtech/authservice/sign"
	"github.com/dhtech/authservice/verify"
	"github.com/dhtech/authservice/webui"
)

var (
	listenAddress = flag.String("listen", ":1214", "Address to listen to")
)

func main() {
	flag.Parse()

	s, err := net.Listen("tcp", *listenAddress)
	if err != nil {
		log.Fatalf("failed to listen: %v", err)
	}

	mux := cmux.New(s)
	sg := mux.Match(cmux.HTTP2HeaderField("content-type", "application/grpc"))
	sh := mux.Match(cmux.Any())

	a := audit.New()
	si := sign.New(a)
	w := webui.New()
	v := verify.New(w, si)
	r := rpc.New(v)

	go r.Serve(sg)
	go w.Serve(sh)

	err = mux.Serve()
	if err != nil {
		log.Fatalf("failed to serve mux: %v", err)
	}
}