main.go 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778
  1. package main
  2. import (
  3. "fmt"
  4. "goseg/config"
  5. "goseg/docker"
  6. "goseg/ws"
  7. "log/slog"
  8. "net/http"
  9. "os"
  10. "github.com/gorilla/mux"
  11. )
  12. var (
  13. logger = slog.New(slog.NewJSONHandler(os.Stdout, nil))
  14. devMode = false
  15. )
  16. func loadService(loadFunc func() error, errMsg string) {
  17. go func() {
  18. if err := loadFunc(); err != nil {
  19. logger.Error(fmt.Sprintf("%s %v", errMsg, err))
  20. }
  21. }()
  22. }
  23. func main() {
  24. for _, arg := range os.Args[1:] {
  25. // trigger this with `./groundseg dev`
  26. if arg == "dev" {
  27. logger.Info("Starting GroundSeg in debug mode")
  28. devMode = true
  29. }
  30. }
  31. logger.Info("Starting GroundSeg")
  32. logger.Info("Urbit is love <3")
  33. // global SysConfig var is managed through config package
  34. conf := config.Conf()
  35. internetAvailable := config.NetCheck("1.1.1.1:53")
  36. availMsg := fmt.Sprintf("Internet available: %t", internetAvailable)
  37. logger.Info(availMsg)
  38. // async operation to retrieve version info
  39. versionUpdateChannel := make(chan bool)
  40. if conf.UpdateMode == "auto" {
  41. go func() {
  42. _, versionUpdate := config.CheckVersion()
  43. if versionUpdate {
  44. logger.Info("Version info retrieved")
  45. }
  46. versionUpdateChannel <- versionUpdate
  47. }()
  48. }
  49. // just making sure we can parse (debug)
  50. var pierList string
  51. for _, pier := range conf.Piers {
  52. pierList = pierList + ", " + pier
  53. }
  54. logger.Info(pierList)
  55. // block until returns (debug)
  56. <-versionUpdateChannel
  57. // Load Netdata
  58. loadService(docker.LoadNetdata, "Unable to load Netdata!")
  59. // Load Wireguard
  60. loadService(docker.LoadWireguard, "Unable to load Wireguard!")
  61. // Load Urbits
  62. loadService(docker.LoadUrbits, "Unable to load Urbit ships!")
  63. // Load MC
  64. loadService(docker.LoadMC, "Unable to load MinIO Client!")
  65. // Load MinIOs
  66. loadService(docker.LoadMinIOs, "Unable to load MinIO containers!")
  67. // Websocket
  68. r := mux.NewRouter()
  69. r.HandleFunc("/ws", ws.WsHandler)
  70. http.ListenAndServe(":3000", r)
  71. }