gmitohtml-upstream/main.go

package main

import (
	"flag"
	"fmt"
	"io/ioutil"
	"log"
	"net/url"
	"os"
	"os/exec"
	"path"
	"runtime"

	"gitlab.com/tslocum/gmitohtml/pkg/gmitohtml"
)

func openBrowser(url string) {
	var err error
	switch runtime.GOOS {
	case "linux":
		err = exec.Command("xdg-open", url).Start()
	case "windows":
		err = exec.Command("rundll32", "url.dll,FileProtocolHandler", url).Start()
	case "darwin":
		err = exec.Command("open", url).Start()
	default:
		err = fmt.Errorf("unsupported platform")
	}
	if err != nil {
		log.Fatal(err)
	}
}

func main() {
	var view bool
	var allowFile bool
	var daemon string
	var configFile string
	flag.BoolVar(&view, "view", false, "open web browser")
	flag.BoolVar(&allowFile, "allow-file", false, "allow local file access via file://")
	flag.StringVar(&daemon, "daemon", "", "start daemon on specified address")
	flag.StringVar(&configFile, "config", "", "path to configuration file")
	// TODO option to include response header in page
	flag.Parse()

	if configFile == "" {
		homedir, err := os.UserHomeDir()
		if err == nil && homedir != "" {
			defaultConfig := path.Join(homedir, ".config", "gmitohtml", "config.yaml")
			if _, err := os.Stat(defaultConfig); !os.IsNotExist(err) {
				configFile = defaultConfig
			}
		}
	}

	if configFile != "" {
		err := readconfig(configFile)
		if err != nil {
			log.Fatalf("failed to read configuration file at %s: %v\nSee CONFIGURATION.md for information on configuring gmitohtml", configFile, err)
		}
	}

	for domain, cc := range config.Certs {
		certData, err := ioutil.ReadFile(cc.Cert)
		if err != nil {
			log.Fatalf("failed to load client certificate for domain %s: %s", domain, err)
		}

		keyData, err := ioutil.ReadFile(cc.Key)
		if err != nil {
			log.Fatalf("failed to load client certificate for domain %s: %s", domain, err)
		}

		err = gmitohtml.SetClientCertificate(domain, certData, keyData)
		if err != nil {
			log.Fatalf("failed to load client certificate for domain %s", domain)
		}
	}

	if daemon != "" {
		err := gmitohtml.StartDaemon(daemon, allowFile)
		if err != nil {
			log.Fatal(err)
		}

		if view {
			openBrowser("http://" + daemon)
		}

		select {}
	}

	data, err := ioutil.ReadAll(os.Stdin)
	if err != nil {
		log.Fatal(err)
	}

	data = gmitohtml.Convert(data, "")

	if view {
		openBrowser(string(append([]byte("data:text/html,"), []byte(url.PathEscape(string(data)))...)))
		return
	}
	fmt.Print(gmitohtml.Convert(data, ""))
}