gitpod/components/gitpod-cli/cmd/open.go

// Copyright (c) 2020 Gitpod GmbH. All rights reserved.
// Licensed under the GNU Affero General Public License (AGPL).
// See License-AGPL.txt in the project root for license information.

package cmd

import (
	"log"
	"os"
	"os/exec"
	"sync"
	"syscall"
	"time"

	"github.com/spf13/cobra"

	"github.com/gitpod-io/gitpod/gitpod-cli/pkg/theialib"
)

// initCmd represents the init command
var openCmd = &cobra.Command{
	Use:   "open <filename>",
	Short: "Opens a file in Gitpod",
	Long:  ``,
	Args:  cobra.MinimumNArgs(1),
	Run: func(cmd *cobra.Command, args []string) {
		wait, _ := cmd.Flags().GetBool("wait")

		service, err := theialib.NewServiceFromEnv()
		if err != nil {
			log.Fatal(err)
		}

		var wg sync.WaitGroup
		for _, fn := range args {
			_, err := service.OpenFile(theialib.OpenFileRequest{Path: fn})
			if err == theialib.ErrNotFound {
				// Code cannot provide cli service and returns 404. See if GP_OPEN_EDITOR is set (e.g. to 'code' CLI):
				gpOpenEditor := os.Getenv("GP_OPEN_EDITOR")
				if gpOpenEditor == "" {
					// Use 'vi' instead
					gpOpenEditor = "vi"
				}

				argv0, err := exec.LookPath(gpOpenEditor)
				if err != nil {
					log.Fatal(err)
				}
				err = syscall.Exec(argv0, append([]string{gpOpenEditor}, args...), os.Environ())
				if err != nil {
					log.Fatal(err)
				}
				return
			}
			if err != nil {
				log.Println(err)
				continue
			}

			if wait {
				wg.Add(1)
				go func(fn string) {
					defer wg.Done()

					for {
						resp, err := service.IsFileOpen(theialib.IsFileOpenRequest{Path: fn})
						if err != nil {
							log.Fatal(err)
							return
						}
						if !resp.IsOpen {
							return
						}

						time.Sleep(1 * time.Second)
					}
				}(fn)
			}
		}

		wg.Wait()
	},
}

func init() {
	rootCmd.AddCommand(openCmd)
	openCmd.Flags().BoolP("wait", "w", false, "wait until all opened files are closed again")
}