main.go

package main

import (
	"context"
	"flag"
	"fmt"
	"log"
	"os"
	"os/signal"
	"syscall"

	"kamatera/shell-script-csi-driver/driver"
)

func main() {
	var (
		endpoint   = flag.String("endpoint", "unix:///var/lib/kubelet/plugins/"+driver.DefaultDriverName+"/csi.sock", "CSI endpoint")
		driverName = flag.String("driver-name", driver.DefaultDriverName, "Name for the driver")
		version    = flag.Bool("version", false, "Print the version and exit.")
		workdir    = flag.String("workdir", "/var/run/kamatera/shell-script-csi-driver", "Workdir for runtime files")
		nodeId     = flag.String("node-id", "", "The Node ID")
	)
	flag.Parse()

	if *version {
		fmt.Printf("%s - %s (%s)\n", driver.GetVersion(), driver.GetCommit(), driver.GetTreeState())
		os.Exit(0)
	}

	if len(*nodeId) < 1 {
		fmt.Println("node-id is required")
		os.Exit(1)
	}

	drv, err := driver.NewDriver(*endpoint, *driverName, *workdir, *nodeId)
	if err != nil {
		log.Fatalln(err)
	}

	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	c := make(chan os.Signal, 1)
	signal.Notify(c, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
	go func() {
		<-c
		cancel()
	}()

	if err := drv.Run(ctx); err != nil {
		log.Fatalln(err)
	}
}