| Giorgi Lekveishvili | 0ccd148 | 2023-06-21 15:02:24 +0400 | [diff] [blame] | 1 | package main |
| 2 | |
| 3 | import ( |
| Giorgi Lekveishvili | 0ccd148 | 2023-06-21 15:02:24 +0400 | [diff] [blame] | 4 | "log" |
| Giorgi Lekveishvili | 94cda9d | 2023-07-20 10:16:09 +0400 | [diff] [blame^] | 5 | "net/netip" |
| Giorgi Lekveishvili | 0ccd148 | 2023-06-21 15:02:24 +0400 | [diff] [blame] | 6 | "os" |
| Giorgi Lekveishvili | 0ccd148 | 2023-06-21 15:02:24 +0400 | [diff] [blame] | 7 | |
| Giorgi Lekveishvili | 0ccd148 | 2023-06-21 15:02:24 +0400 | [diff] [blame] | 8 | "github.com/spf13/cobra" |
| 9 | |
| 10 | "github.com/giolekva/pcloud/core/installer" |
| 11 | "github.com/giolekva/pcloud/core/installer/soft" |
| Giorgi Lekveishvili | b4a9c98 | 2023-06-22 15:17:02 +0400 | [diff] [blame] | 12 | "github.com/giolekva/pcloud/core/installer/welcome" |
| Giorgi Lekveishvili | 0ccd148 | 2023-06-21 15:02:24 +0400 | [diff] [blame] | 13 | ) |
| 14 | |
| 15 | var envManagerFlags struct { |
| Giorgi Lekveishvili | 94cda9d | 2023-07-20 10:16:09 +0400 | [diff] [blame^] | 16 | repoAddr string |
| 17 | repoName string |
| Giorgi Lekveishvili | 0ccd148 | 2023-06-21 15:02:24 +0400 | [diff] [blame] | 18 | sshKey string |
| 19 | port int |
| 20 | } |
| 21 | |
| 22 | func envManagerCmd() *cobra.Command { |
| 23 | cmd := &cobra.Command{ |
| 24 | Use: "envmanager", |
| 25 | RunE: envManagerCmdRun, |
| 26 | } |
| 27 | cmd.Flags().StringVar( |
| Giorgi Lekveishvili | 94cda9d | 2023-07-20 10:16:09 +0400 | [diff] [blame^] | 28 | &envManagerFlags.repoAddr, |
| 29 | "repo-addr", |
| Giorgi Lekveishvili | 0ccd148 | 2023-06-21 15:02:24 +0400 | [diff] [blame] | 30 | "", |
| 31 | "", |
| 32 | ) |
| Giorgi Lekveishvili | 94cda9d | 2023-07-20 10:16:09 +0400 | [diff] [blame^] | 33 | cmd.Flags().StringVar( |
| 34 | &envManagerFlags.repoName, |
| 35 | "repo-name", |
| 36 | "", |
| Giorgi Lekveishvili | 0ccd148 | 2023-06-21 15:02:24 +0400 | [diff] [blame] | 37 | "", |
| 38 | ) |
| 39 | cmd.Flags().StringVar( |
| 40 | &envManagerFlags.sshKey, |
| 41 | "ssh-key", |
| 42 | "", |
| 43 | "", |
| 44 | ) |
| 45 | cmd.Flags().IntVar( |
| 46 | &envManagerFlags.port, |
| 47 | "port", |
| 48 | 8080, |
| 49 | "", |
| 50 | ) |
| 51 | return cmd |
| 52 | } |
| 53 | |
| 54 | func envManagerCmdRun(cmd *cobra.Command, args []string) error { |
| 55 | sshKey, err := os.ReadFile(envManagerFlags.sshKey) |
| 56 | if err != nil { |
| 57 | return err |
| 58 | } |
| Giorgi Lekveishvili | 94cda9d | 2023-07-20 10:16:09 +0400 | [diff] [blame^] | 59 | repoAddr, err := netip.ParseAddrPort(envManagerFlags.repoAddr) |
| Giorgi Lekveishvili | 0ccd148 | 2023-06-21 15:02:24 +0400 | [diff] [blame] | 60 | if err != nil { |
| 61 | return err |
| 62 | } |
| Giorgi Lekveishvili | 94cda9d | 2023-07-20 10:16:09 +0400 | [diff] [blame^] | 63 | ss, err := soft.NewClient(repoAddr, sshKey, log.Default()) |
| 64 | if err != nil { |
| 65 | return err |
| 66 | } |
| 67 | repo, err := ss.GetRepo(envManagerFlags.repoName) |
| Giorgi Lekveishvili | 0ccd148 | 2023-06-21 15:02:24 +0400 | [diff] [blame] | 68 | if err != nil { |
| 69 | return err |
| 70 | } |
| Giorgi Lekveishvili | 0ccd148 | 2023-06-21 15:02:24 +0400 | [diff] [blame] | 71 | repoIO := installer.NewRepoIO(repo, ss.Signer) |
| Giorgi Lekveishvili | 7fb28bf | 2023-06-24 19:51:16 +0400 | [diff] [blame] | 72 | nsCreator, err := newNSCreator() |
| 73 | if err != nil { |
| 74 | return err |
| 75 | } |
| Giorgi Lekveishvili | b4a9c98 | 2023-06-22 15:17:02 +0400 | [diff] [blame] | 76 | s := welcome.NewEnvServer( |
| 77 | envManagerFlags.port, |
| 78 | ss, |
| 79 | repoIO, |
| Giorgi Lekveishvili | 7fb28bf | 2023-06-24 19:51:16 +0400 | [diff] [blame] | 80 | nsCreator, |
| Giorgi Lekveishvili | b4a9c98 | 2023-06-22 15:17:02 +0400 | [diff] [blame] | 81 | ) |
| 82 | s.Start() |
| Giorgi Lekveishvili | 0ccd148 | 2023-06-21 15:02:24 +0400 | [diff] [blame] | 83 | return nil |
| 84 | } |