You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
127 lines
2.9 KiB
127 lines
2.9 KiB
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/url"
|
|
"os"
|
|
"os/exec"
|
|
"runtime"
|
|
|
|
"code.rocketnine.space/tslocum/ez"
|
|
"code.rocketnine.space/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
|
|
allowFile bool
|
|
daemon string
|
|
hostname string
|
|
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(&hostname, "hostname", "", "server hostname (e.g. rocketnine.space) (defaults to daemon address)")
|
|
flag.StringVar(&configFile, "config", "", "path to configuration file")
|
|
// TODO option to include response header in page
|
|
flag.Parse()
|
|
|
|
defaultConfig, err := ez.DefaultConfigPath("gmitohtml")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
} else if configFile == "" {
|
|
configFile = defaultConfig
|
|
}
|
|
|
|
if configFile != "" {
|
|
var configExists bool
|
|
if _, err := os.Stat(defaultConfig); !os.IsNotExist(err) {
|
|
configExists = true
|
|
}
|
|
|
|
if configExists || configFile != defaultConfig {
|
|
err := ez.Deserialize(gmitohtml.Config, 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 u, label := range gmitohtml.Config.Bookmarks {
|
|
gmitohtml.AddBookmark(u, label)
|
|
}
|
|
}
|
|
}
|
|
|
|
for domain, cc := range gmitohtml.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 != "" {
|
|
gmitohtml.SetOnBookmarksChanged(func() {
|
|
gmitohtml.Config.Bookmarks = gmitohtml.GetBookmarks()
|
|
|
|
err := ez.Serialize(gmitohtml.Config, configFile)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
})
|
|
|
|
err := gmitohtml.StartDaemon(daemon, hostname, 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, ""))
|
|
}
|