| Giorgi Lekveishvili | 12850ee | 2023-06-22 13:11:17 +0400 | [diff] [blame] | 1 | package welcome |
| 2 | |
| 3 | import ( |
| 4 | "embed" |
| 5 | "encoding/json" |
| 6 | "fmt" |
| 7 | "log" |
| 8 | "net/http" |
| Giorgi Lekveishvili | 6b887be | 2023-06-22 14:38:19 +0400 | [diff] [blame] | 9 | "net/url" |
| Giorgi Lekveishvili | 12850ee | 2023-06-22 13:11:17 +0400 | [diff] [blame] | 10 | |
| 11 | "github.com/labstack/echo/v4" |
| 12 | |
| 13 | "github.com/giolekva/pcloud/core/installer" |
| 14 | ) |
| 15 | |
| Giorgi Lekveishvili | b4a9c98 | 2023-06-22 15:17:02 +0400 | [diff] [blame] | 16 | //go:embed create-admin-account.html |
| Giorgi Lekveishvili | 12850ee | 2023-06-22 13:11:17 +0400 | [diff] [blame] | 17 | var indexHtml string |
| 18 | |
| 19 | //go:embed static/* |
| 20 | var staticAssets embed.FS |
| 21 | |
| 22 | type Server struct { |
| Giorgi Lekveishvili | 7fb28bf | 2023-06-24 19:51:16 +0400 | [diff] [blame^] | 23 | port int |
| 24 | repo installer.RepoIO |
| 25 | nsCreator installer.NamespaceCreator |
| Giorgi Lekveishvili | 12850ee | 2023-06-22 13:11:17 +0400 | [diff] [blame] | 26 | } |
| 27 | |
| Giorgi Lekveishvili | 7fb28bf | 2023-06-24 19:51:16 +0400 | [diff] [blame^] | 28 | func NewServer(port int, repo installer.RepoIO, nsCreator installer.NamespaceCreator) *Server { |
| Giorgi Lekveishvili | 12850ee | 2023-06-22 13:11:17 +0400 | [diff] [blame] | 29 | return &Server{ |
| 30 | port, |
| 31 | repo, |
| Giorgi Lekveishvili | 7fb28bf | 2023-06-24 19:51:16 +0400 | [diff] [blame^] | 32 | nsCreator, |
| Giorgi Lekveishvili | 12850ee | 2023-06-22 13:11:17 +0400 | [diff] [blame] | 33 | } |
| 34 | } |
| 35 | |
| 36 | func (s *Server) Start() { |
| 37 | e := echo.New() |
| 38 | e.StaticFS("/static", echo.MustSubFS(staticAssets, "static")) |
| 39 | e.POST("/create-admin-account", s.createAdminAccount) |
| 40 | e.GET("/", s.createAdminAccountForm) |
| 41 | log.Fatal(e.Start(fmt.Sprintf(":%d", s.port))) |
| 42 | } |
| 43 | |
| 44 | func (s *Server) createAdminAccountForm(c echo.Context) error { |
| 45 | return c.HTML(http.StatusOK, indexHtml) |
| 46 | } |
| 47 | |
| 48 | type createAdminAccountReq struct { |
| 49 | Username string `json:"username,omitempty"` |
| 50 | Password string `json:"password,omitempty"` // TODO(giolekva): actually use this |
| 51 | GandiAPIToken string `json:"gandiAPIToken,omitempty"` |
| 52 | SecretToken string `json:"secretToken,omitempty"` |
| 53 | } |
| 54 | |
| Giorgi Lekveishvili | 6b887be | 2023-06-22 14:38:19 +0400 | [diff] [blame] | 55 | func getFormValue(v url.Values, name string) (string, error) { |
| 56 | items, ok := v[name] |
| 57 | if !ok || len(items) != 1 { |
| 58 | return "", fmt.Errorf("%s not found", name) |
| 59 | } |
| 60 | return items[0], nil |
| 61 | } |
| 62 | |
| Giorgi Lekveishvili | 12850ee | 2023-06-22 13:11:17 +0400 | [diff] [blame] | 63 | func (s *Server) createAdminAccount(c echo.Context) error { |
| 64 | var req createAdminAccountReq |
| Giorgi Lekveishvili | 6b887be | 2023-06-22 14:38:19 +0400 | [diff] [blame] | 65 | if err := func() error { |
| 66 | var err error |
| 67 | f, err := c.FormParams() |
| 68 | if err != nil { |
| 69 | return err |
| 70 | } |
| 71 | if req.Username, err = getFormValue(f, "username"); err != nil { |
| 72 | return err |
| 73 | } |
| 74 | if req.Password, err = getFormValue(f, "password"); err != nil { |
| 75 | return err |
| 76 | } |
| 77 | if req.GandiAPIToken, err = getFormValue(f, "gandi-api-token"); err != nil { |
| 78 | return err |
| 79 | } |
| 80 | if req.SecretToken, err = getFormValue(f, "secret-token"); err != nil { |
| 81 | return err |
| 82 | } |
| 83 | return nil |
| 84 | }(); err != nil { |
| 85 | if err := json.NewDecoder(c.Request().Body).Decode(&req); err != nil { |
| 86 | return err |
| 87 | } |
| Giorgi Lekveishvili | 12850ee | 2023-06-22 13:11:17 +0400 | [diff] [blame] | 88 | } |
| 89 | // TODO(giolekva): accounts-ui create user req |
| 90 | { |
| Giorgi Lekveishvili | 7fb28bf | 2023-06-24 19:51:16 +0400 | [diff] [blame^] | 91 | config, err := s.repo.ReadConfig() |
| 92 | if err != nil { |
| 93 | return err |
| 94 | } |
| 95 | if err != nil { |
| 96 | return err |
| 97 | } |
| 98 | nsGen := installer.NewPrefixGenerator(config.Values.Id + "-") |
| 99 | appManager, err := installer.NewAppManager(s.repo, s.nsCreator) |
| Giorgi Lekveishvili | 12850ee | 2023-06-22 13:11:17 +0400 | [diff] [blame] | 100 | if err != nil { |
| 101 | return err |
| 102 | } |
| 103 | appsRepo := installer.NewInMemoryAppRepository(installer.CreateAllApps()) |
| 104 | { |
| 105 | app, err := appsRepo.Find("certificate-issuer-private") |
| 106 | if err != nil { |
| 107 | return err |
| 108 | } |
| Giorgi Lekveishvili | 7fb28bf | 2023-06-24 19:51:16 +0400 | [diff] [blame^] | 109 | if err := appManager.Install(*app, nsGen, map[string]any{ |
| Giorgi Lekveishvili | 12850ee | 2023-06-22 13:11:17 +0400 | [diff] [blame] | 110 | "GandiAPIToken": req.GandiAPIToken, |
| 111 | }); err != nil { |
| 112 | return err |
| 113 | } |
| 114 | } |
| 115 | { |
| 116 | app, err := appsRepo.Find("tailscale-proxy") |
| 117 | if err != nil { |
| 118 | return err |
| 119 | } |
| Giorgi Lekveishvili | 7fb28bf | 2023-06-24 19:51:16 +0400 | [diff] [blame^] | 120 | if err := appManager.Install(*app, nsGen, map[string]any{ |
| Giorgi Lekveishvili | 12850ee | 2023-06-22 13:11:17 +0400 | [diff] [blame] | 121 | "Username": req.Username, |
| 122 | "IPSubnet": "10.1.0.0/24", // TODO(giolekva): this should be taken from the config generated during new env creation |
| 123 | }); err != nil { |
| 124 | return err |
| 125 | } |
| 126 | // TODO(giolekva): headscale accept routes |
| 127 | } |
| 128 | } |
| 129 | return c.String(http.StatusOK, "OK") |
| 130 | } |