jeol-t330a/succbone/succd/main.go

package main

import (
	"context"
	"flag"
	"net/http"
	"os"
	"os/signal"

	"k8s.io/klog"
)

var (
	flagFake                   bool
	flagListenHTTP             string
	flagPressureThresholdRough = ScientificNotationValue(1e-1)
	flagPressureThresholdHigh  = ScientificNotationValue(1e-4)
)

func main() {
	flag.BoolVar(&flagFake, "fake", false, "Enable fake mode which allows to run succd for tests outside the succbone")
	flag.StringVar(&flagListenHTTP, "listen_http", ":8080", "Address at which to listen for HTTP requests")
	flag.TextVar(&flagPressureThresholdRough, "pressure_threshold_rough", &flagPressureThresholdRough, "Threshold for opening up diffusion pump (mbar)")
	flag.TextVar(&flagPressureThresholdHigh, "pressure_threshold_high", &flagPressureThresholdHigh, "Threshold for enabling high voltage circuits (mbar)")
	flag.Parse()

	ctx, _ := signal.NotifyContext(context.Background(), os.Interrupt)

	d := daemon{}
	d.daemonState.rpOn = true

	d.aboveRough.threshold = float64(flagPressureThresholdRough)
	d.aboveHigh.threshold = float64(flagPressureThresholdHigh)
	if flagFake {
		klog.Infof("Starting with fake peripherals")
		d.adcPirani = &fakeADC{}
		d.gpioRoughingPump = &fakeGPIO{desc: "rp"}
		d.gpioDiffusionPump = &fakeGPIO{desc: "~dp"}
		d.gpioBtnPumpDown = &fakeGPIO{desc: "~pd"}
		d.gpioBtnVent = &fakeGPIO{desc: "~vent"}
		d.gpioBelowRough = &fakeGPIO{desc: "~rough"}
		d.gpioBelowHigh = &fakeGPIO{desc: "~high"}
	} else {
		adc, err := newBBADC(0)
		if err != nil {
			klog.Exitf("Failed to setup Pirani ADC: %v", err)
		}
		d.adcPirani = adc

		for _, c := range []struct {
			out *gpio
			num int
		}{
			{&d.gpioRoughingPump, 115},
			{&d.gpioDiffusionPump, 49},
			{&d.gpioBtnPumpDown, 48},
			{&d.gpioBtnVent, 60},
			{&d.gpioBelowRough, 30},
			{&d.gpioBelowHigh, 7},
		} {
			*c.out, err = newBBGPIO(c.num, true)
			if err != nil {
				klog.Exitf("Failed to setup GPIO: %v", err)
			}
		}
	}

	web := webServer{
		d: &d,
	}
	web.setupViews()

	klog.Infof("Listening for HTTP at %s", flagListenHTTP)
	go func() {
		if err := http.ListenAndServe(flagListenHTTP, nil); err != nil {
			klog.Errorf("HTTP listen failed: %v", err)
		}
	}()

	go d.process(ctx)
	<-ctx.Done()
}