60 lines
1.1 KiB
Go
60 lines
1.1 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"math/rand"
|
|
"net/url"
|
|
"os"
|
|
"os/signal"
|
|
"time"
|
|
|
|
"github.com/BeryJu/passbook/proxy/pkg/server"
|
|
)
|
|
|
|
const helpMessage = `passbook proxy
|
|
|
|
Required environment variables:
|
|
- PASSBOOK_HOST: URL to connect to (format "http://passbook.company")
|
|
- PASSBOOK_TOKEN: Token to authenticate with
|
|
- PASSBOOK_INSECURE: Skip SSL Certificate verification`
|
|
|
|
// RunServer main entrypoint, runs the full server
|
|
func RunServer() {
|
|
pbURL, found := os.LookupEnv("PASSBOOK_HOST")
|
|
if !found {
|
|
fmt.Println("env PASSBOOK_HOST not set!")
|
|
fmt.Println(helpMessage)
|
|
os.Exit(1)
|
|
}
|
|
pbToken, found := os.LookupEnv("PASSBOOK_TOKEN")
|
|
if !found {
|
|
fmt.Println("env PASSBOOK_TOKEN not set!")
|
|
fmt.Println(helpMessage)
|
|
os.Exit(1)
|
|
}
|
|
|
|
pbURLActual, err := url.Parse(pbURL)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
fmt.Println(helpMessage)
|
|
os.Exit(1)
|
|
}
|
|
|
|
rand.Seed(time.Now().UnixNano())
|
|
|
|
ac := server.NewAPIController(*pbURLActual, pbToken)
|
|
|
|
interrupt := make(chan os.Signal, 1)
|
|
signal.Notify(interrupt, os.Interrupt)
|
|
|
|
ac.Start()
|
|
|
|
for {
|
|
select {
|
|
case <-interrupt:
|
|
ac.Shutdown()
|
|
os.Exit(0)
|
|
}
|
|
}
|
|
}
|