pages/internal/commands/host.go

// Copyright (C) 2022  The pages authors
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.
//

package commands

import (
	"context"
	"mime"
	"net/http"
	"path"
	"time"

	"github.com/urfave/cli/v2"
	"go.uber.org/zap"
	"golang.org/x/sync/errgroup"

	"code.pitz.tech/mya/pages/internal"
	"code.pitz.tech/mya/pages/internal/git"

	"github.com/mjpitz/myago/flagset"
	"github.com/mjpitz/myago/zaputil"
)

type HostConfig struct {
	internal.ServerConfig
	Git git.Config `json:"git"`
}

var (
	hostConfig = &HostConfig{
		ServerConfig: internal.ServerConfig{
			Public:  internal.BindConfig{Address: "0.0.0.0:8080"},
			Private: internal.BindConfig{Address: "0.0.0.0:8081"},
		},
	}

	Host = &cli.Command{
		Name:      "host",
		Usage:     "Host the web page content",
		UsageText: "pages host",
		Flags:     flagset.ExtractPrefix("pages", hostConfig),
		Action: func(ctx *cli.Context) error {
			log := zaputil.Extract(ctx.Context)

			// additional mime-types that need to be explicitly registered
			_ = mime.AddExtensionType(".woff2", "application/font-woff2")
			_ = mime.AddExtensionType(".woff", "application/font-woff")
			_ = mime.AddExtensionType(".ttf", "font/ttf")
			_ = mime.AddExtensionType(".yaml", "application/yaml")
			_ = mime.AddExtensionType(".yml", "application/yaml")
			_ = mime.AddExtensionType(".json", "application/json")

			gitService, err := git.NewService(hostConfig.Git)
			if err != nil {
				return err
			}

			err = gitService.Load(ctx.Context)
			if err != nil {
				return err
			}

			server, err := internal.NewServer(ctx.Context, hostConfig.ServerConfig)
			if err != nil {
				return err
			}

			{
				server.AdminMux.
					HandleFunc("/sync", func(w http.ResponseWriter, r *http.Request) {
						err = gitService.Sync(r.Context())
						if err != nil {
							log.Error(err.Error())
							http.Error(w, "", http.StatusInternalServerError)
							return
						}
					}).
					Methods(http.MethodPost)
			}

			httpfs := http.FileServer(git.HTTP(gitService.FS))
			server.PublicMux.PathPrefix("/").HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				values := r.URL.Query()

				if values.Get("go-get") == "1" {
					// peak for go-get requests
					_, name := path.Split(r.URL.Path)
					index := path.Join(r.URL.Path, "index.html")

					// if index.html exists, then use that
					info, err := gitService.FS.Stat(index)
					if err == nil {
						file, err := gitService.FS.Open(index)
						if err != nil {
							http.Error(w, "", http.StatusInternalServerError)
							return
						}

						defer file.Close()

						http.ServeContent(w, r, name, info.ModTime(), file)
						return
					}
				}

				httpfs.ServeHTTP(w, r)
				return
			}).Methods(http.MethodGet)

			log.Info("serving",
				zap.String("public", hostConfig.Public.Address),
				zap.String("private", hostConfig.Private.Address))

			group, c := errgroup.WithContext(ctx.Context)
			group.Go(server.ListenAndServe)
			group.Go(func() error {
				timer := time.NewTimer(hostConfig.Git.SyncInterval)
				defer timer.Stop()

				for {
					select {
					case <-c.Done():
						return nil
					case <-timer.C:
						_ = gitService.Sync(ctx.Context)

						timer.Reset(hostConfig.Git.SyncInterval)
					}
				}
			})

			<-c.Done()

			_ = server.Shutdown(context.Background())
			_ = group.Wait()

			err = c.Err()
			if err != context.Canceled {
				return err
			}

			return nil
		},
		HideHelpCommand: true,
	}
)