go.chromium.org/luci@v0.0.0-20240309015107-7cdc2e660f33/tools/internal/apigen/service.go

// Copyright 2015 The LUCI Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package apigen

import (
	"context"
	"errors"
	"fmt"
	"io/ioutil"
	"net/url"
	"os"
	"os/exec"
	"path/filepath"

	log "go.chromium.org/luci/common/logging"

	"gopkg.in/yaml.v2"
)

// appYAML is a subset of the contents of an AppEngine application's "app.yaml"
// descriptor needed by this service.
type appYAML struct {
	Runtime string `yaml:"runtime"`
	VM      bool   `yaml:"vm"`
}

type service interface {
	run(context.Context, serviceRunFunc) error
}

type serviceRunFunc func(c context.Context, u url.URL) error

// loadService is a generic service loader routine. It attempts to:
// 1) Identify the filesystem path of the service being described.
// 2) Analyze its "app.yaml" to determine its runtime parameters.
// 3) Construct and return a `service` instance for the result.
//
// "path" is decoded as:
// - A discovery base URL
// - A filesystem path, pointing to an "app.yaml" file.
// - A Go package path containing an "app.yaml" file.
func loadService(c context.Context, path string) (service, error) {
	url, err := url.Parse(path)
	if err == nil && url.Scheme != "" {
		log.Fields{
			"url": path,
		}.Infof(c, "Identified path as service URL.")
		return &remoteDiscoveryService{
			url: *url,
		}, nil
	}
	log.Fields{
		log.ErrorKey: err,
		"value":      path,
	}.Debugf(c, "Path did not parse as URL. Trying local filesystem options.")

	yamlPath := ""
	st, err := os.Stat(path)
	switch {
	case os.IsNotExist(err):
		log.Fields{
			"path": path,
		}.Debugf(c, "Path does not exist. Maybe it's a Go path?")

		// Not a filesysem path. Perhaps it's a Go package on GOPATH?
		pkgPath, err := getPackagePath(path)
		if err != nil {
			log.Fields{
				"path": path,
			}.Debugf(c, "Could not resolve package path.")
			return nil, fmt.Errorf("could not resolve path [%s]", path)
		}
		path = pkgPath

	case err != nil:
		return nil, fmt.Errorf("failed to stat [%s]: %s", path, err)

	case st.IsDir():
		break

	default:
		// "path" is a path to a non-directory. Use its parent directory.
		yamlPath, err = filepath.Abs(path)
		if err != nil {
			return nil, fmt.Errorf("could not get absolute path for YAML config [%s]: %s", path, err)
		}
		path = filepath.Dir(path)
	}

	// "path" is a directory. Does its `app.yaml` exist?
	if yamlPath == "" {
		yamlPath = filepath.Join(path, "app.yaml")
	}

	if _, err = os.Stat(yamlPath); err != nil {
		return nil, fmt.Errorf("unable to stat YAML config at [%s]: %s", yamlPath, err)
	}

	configData, err := os.ReadFile(yamlPath)
	if err != nil {
		return nil, fmt.Errorf("failed to read YAML config at [%s]: %s", yamlPath, err)
	}

	config := appYAML{}
	if err := yaml.Unmarshal(configData, &config); err != nil {
		return nil, fmt.Errorf("failed to Unmarshal YAML config from [%s]: %s", yamlPath, err)
	}

	switch config.Runtime {
	case "go":
		if config.VM {
			return &discoveryTranslateService{
				dir: path,
			}, nil
		}
		return &devAppserverService{
			prerun: func(c context.Context) error {
				return checkBuild(c, path)
			},
			args: []string{"goapp", "serve", yamlPath},
		}, nil

	case "python27":
		return &devAppserverService{
			args: []string{"dev_appserver.py", yamlPath},
		}, nil

	default:
		return nil, fmt.Errorf("don't know how to load service runtime [%s]", config.Runtime)
	}
}

type remoteDiscoveryService struct {
	url url.URL
}

func (s *remoteDiscoveryService) run(c context.Context, f serviceRunFunc) error {
	return f(c, s.url)
}

type devAppserverService struct {
	prerun func(context.Context) error
	args   []string
}

func (s *devAppserverService) run(c context.Context, f serviceRunFunc) error {
	if s.prerun != nil {
		if err := s.prerun(c); err != nil {
			return err
		}
	}

	log.Fields{
		"args": s.args,
	}.Infof(c, "Executing service.")

	if len(s.args) == 0 {
		return errors.New("no command configured")
	}

	// Execute `dev_appserver`.
	cmd := &killableCommand{
		Cmd: exec.Command(s.args[0], s.args[1:]...),
	}
	if err := cmd.Start(); err != nil {
		return err
	}
	defer cmd.kill(c)

	return f(c, url.URL{
		Scheme: "http",
		Host:   "localhost:8080",
	})
}

// discoveryTranslateService is a service that loads a backend discovery
// document, translates it to a frontend directory list, then hosts its own
// frontend server to expose the translated data.
type discoveryTranslateService struct {
	dir string
}

func (s *discoveryTranslateService) run(c context.Context, f serviceRunFunc) error {
	// Build the Go Managed VM service application.
	p, err := filepath.Abs(s.dir)
	if err != nil {
		return fmt.Errorf("failed to get absolute path [%s]: %s", s.dir, err)
	}

	d, err := ioutil.TempDir(p, "apigen_service")
	if err != nil {
		return err
	}
	defer os.RemoveAll(d)

	svcPath := filepath.Join(d, "service")
	cmd := exec.Command("go", "build", "-o", svcPath, ".")
	cmd.Dir = p
	log.Fields{
		"args": cmd.Args,
		"wd":   cmd.Dir,
	}.Debugf(c, "Executing `go build` command.")
	if out, err := cmd.CombinedOutput(); err != nil {
		log.Fields{
			log.ErrorKey: err,
			"dst":        svcPath,
			"wd":         cmd.Dir,
		}.Errorf(c, "Failed to build package:\n%s", string(out))
		return fmt.Errorf("failed to build package: %s", err)
	}

	// Execute the service.
	svc := &killableCommand{
		Cmd: exec.Command(svcPath),
	}
	svc.Env = append(os.Environ(), "LUCI_GO_APPENGINE_APIGEN=1")
	if err := svc.Start(); err != nil {
		return err
	}
	defer svc.kill(c)

	return f(c, url.URL{
		Scheme: "http",
		Host:   "localhost:8080",
	})
}

func checkBuild(c context.Context, dir string) error {
	d, err := ioutil.TempDir(dir, "apigen_service")
	if err != nil {
		return err
	}
	defer os.RemoveAll(d)

	cmd := exec.Command("go", "build", "-o", filepath.Join(filepath.Base(d), "service"), ".")
	cmd.Dir = dir
	log.Fields{
		"args": cmd.Args,
		"wd":   cmd.Dir,
	}.Debugf(c, "Executing `go build` command.")
	if out, err := cmd.CombinedOutput(); err != nil {
		log.Fields{
			log.ErrorKey: err,
			"wd":         cmd.Dir,
		}.Errorf(c, "Failed to build package:\n%s", string(out))
		return fmt.Errorf("failed to build package: %s", err)
	}
	return nil
}