Quentin McGaw
4 years ago
parent
66ae17527f
commit
424b3bb932
@ -1,9 +1,8 @@
|
||||
module github.com/qdm12/stackedit-docker
|
||||
|
||||
require (
|
||||
github.com/julienschmidt/httprouter v1.3.0
|
||||
github.com/kyokomi/emoji v1.5.1
|
||||
github.com/qdm12/golibs v0.0.0-20200108212314-82caecb70123
|
||||
github.com/kyokomi/emoji v2.2.4+incompatible
|
||||
github.com/qdm12/golibs v0.0.0-20200528010515-765b7cd4f0db
|
||||
)
|
||||
|
||||
go 1.13
|
||||
go 1.14
|
||||
|
||||
@ -1,121 +1,157 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"os"
|
||||
"strings"
|
||||
|
||||
"github.com/kyokomi/emoji"
|
||||
"github.com/qdm12/golibs/healthcheck"
|
||||
"github.com/qdm12/golibs/logging"
|
||||
"github.com/qdm12/golibs/params"
|
||||
"github.com/qdm12/golibs/server"
|
||||
)
|
||||
|
||||
const (
|
||||
defaultConf = `{"dropboxAppKey":"","dropboxAppKeyFull":"","githubClientId":"","googleClientId":"","googleApiKey":"","wordpressClientId":"","allowSponsorship":true}`
|
||||
)
|
||||
|
||||
func main() {
|
||||
if healthcheck.Mode(os.Args) {
|
||||
if err := healthcheck.Query(); err != nil {
|
||||
logging.Err(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
os.Exit(0)
|
||||
}
|
||||
fmt.Println("#####################################")
|
||||
fmt.Println("########## StackEdit Server #########")
|
||||
fmt.Println("########## by Quentin McGaw #########")
|
||||
fmt.Println("########## Give some " + emoji.Sprint(":heart:") + " at ##########")
|
||||
fmt.Println("# github.com/qdm12/stackedit-docker #")
|
||||
fmt.Print("#####################################\n\n")
|
||||
envParams := params.NewEnvParams()
|
||||
logging.InitLogger("console", logging.InfoLevel, 0)
|
||||
listeningPort, err := envParams.GetListeningPort(params.Default("8000"))
|
||||
if err != nil {
|
||||
logging.Err(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
logging.Infof("Using internal listening port %s", listeningPort)
|
||||
rootURL, err := envParams.GetRootURL(params.Default("/"))
|
||||
if err != nil {
|
||||
logging.Err(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
logging.Infof("Using root URL %q", rootURL)
|
||||
|
||||
productionRouter := http.NewServeMux()
|
||||
productionRouter.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
||||
urlStackeditPath := strings.TrimPrefix(r.URL.Path, rootURL)
|
||||
filepath := "/dist/" + urlStackeditPath
|
||||
switch urlStackeditPath {
|
||||
case "/conf", "/app/conf":
|
||||
w.Write(getAllStackeditEnv())
|
||||
return
|
||||
case "/":
|
||||
filepath = "/static/landing/"
|
||||
case "/sitemap.xml":
|
||||
filepath = "/static/sitemap.xml"
|
||||
case "/oauth2/callback":
|
||||
filepath = "/static/oauth2/callback.html"
|
||||
case "/app", "/app/":
|
||||
filepath = "/dist/index.html"
|
||||
default:
|
||||
switch {
|
||||
case strings.HasPrefix(urlStackeditPath, "/static/css/static/fonts/"):
|
||||
filepath = "/dist/" + strings.TrimPrefix(urlStackeditPath, "/static/css/")
|
||||
case strings.HasPrefix(urlStackeditPath, "/app/static/css/static/fonts/"):
|
||||
filepath = "/dist/" + strings.TrimPrefix(urlStackeditPath, "/app/static/css/")
|
||||
case strings.HasPrefix(urlStackeditPath, "/app/"):
|
||||
filepath = "/dist/" + strings.TrimPrefix(urlStackeditPath, "/app/")
|
||||
}
|
||||
}
|
||||
http.ServeFile(w, r, "/html"+filepath)
|
||||
})
|
||||
healthcheckRouter := healthcheck.CreateRouter(func() error { return nil })
|
||||
serverErrs := server.RunServers(
|
||||
server.Settings{Name: "production", Addr: "0.0.0.0:" + listeningPort, Handler: productionRouter},
|
||||
server.Settings{Name: "healthcheck", Addr: "127.0.0.1:9999", Handler: healthcheckRouter},
|
||||
)
|
||||
for _, err := range serverErrs {
|
||||
if err != nil {
|
||||
logging.Err(err)
|
||||
}
|
||||
}
|
||||
if len(serverErrs) > 0 {
|
||||
logging.Errorf("%v", serverErrs)
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
|
||||
// Returns all stackedit env values as JSON
|
||||
func getAllStackeditEnv() []byte {
|
||||
env := struct {
|
||||
DropboxAppKey string `json:"dropboxAppKey"`
|
||||
DropboxAppKeyFull string `json:"dropboxAppKeyFull"`
|
||||
GithubClientID string `json:"githubClientId"`
|
||||
GithubClientSecret string `json:"githubClientSecret"`
|
||||
GoogleClientID string `json:"googleClientId"`
|
||||
GoogleAPIKey string `json:"googleApiKey"`
|
||||
WordpressClientID string `json:"wordpressClientId"`
|
||||
PaypalReceiverEmail string `json:"paypalReceiverEmail"`
|
||||
AllowSponsorship bool `json:"allowSponsorship"`
|
||||
}{}
|
||||
envParams := params.NewEnvParams()
|
||||
env.DropboxAppKey, _ = envParams.GetEnv("DROPBOX_APP_KEY")
|
||||
env.DropboxAppKeyFull, _ = envParams.GetEnv("DROPBOX_APP_KEY_FULL")
|
||||
env.GithubClientID, _ = envParams.GetEnv("GITHUB_CLIENT_ID")
|
||||
env.GithubClientSecret, _ = envParams.GetEnv("GITHUB_CLIENT_SECRET")
|
||||
env.GoogleClientID, _ = envParams.GetEnv("GOOGLE_CLIENT_ID")
|
||||
env.GoogleAPIKey, _ = envParams.GetEnv("GOOGLE_API_KEY")
|
||||
env.WordpressClientID, _ = envParams.GetEnv("WORDPRESS_CLIENT_ID")
|
||||
env.PaypalReceiverEmail, _ = envParams.GetEnv("PAYPAL_RECEIVER_EMAIL")
|
||||
if len(env.PaypalReceiverEmail) > 0 {
|
||||
env.AllowSponsorship = true
|
||||
}
|
||||
b, _ := json.Marshal(env)
|
||||
return b
|
||||
}
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/signal"
|
||||
"strings"
|
||||
"syscall"
|
||||
|
||||
"github.com/kyokomi/emoji"
|
||||
"github.com/qdm12/golibs/healthcheck"
|
||||
"github.com/qdm12/golibs/logging"
|
||||
"github.com/qdm12/golibs/params"
|
||||
"github.com/qdm12/golibs/server"
|
||||
)
|
||||
|
||||
func main() {
|
||||
ctx := context.Background()
|
||||
os.Exit(_main(ctx, os.Args))
|
||||
}
|
||||
|
||||
func _main(ctx context.Context, args []string) int {
|
||||
ctx, cancel := context.WithCancel(ctx)
|
||||
defer cancel()
|
||||
|
||||
if healthcheck.Mode(args) {
|
||||
if err := healthcheck.Query(); err != nil {
|
||||
fmt.Println(err)
|
||||
return 1
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
fmt.Println("#####################################")
|
||||
fmt.Println("########## StackEdit Server #########")
|
||||
fmt.Println("########## by Quentin McGaw #########")
|
||||
fmt.Println("########## Give some " + emoji.Sprint(":heart:") + " at ##########")
|
||||
fmt.Println("# github.com/qdm12/stackedit-docker #")
|
||||
fmt.Print("#####################################\n\n")
|
||||
envParams := params.NewEnvParams()
|
||||
logger, err := logging.NewLogger(logging.ConsoleEncoding, logging.InfoLevel, -1)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
return 1
|
||||
}
|
||||
|
||||
listeningPort, warning, err := envParams.GetListeningPort(params.Default("8000"))
|
||||
if len(warning) > 0 {
|
||||
logger.Warn(warning)
|
||||
}
|
||||
if err != nil {
|
||||
logger.Error(err)
|
||||
return 1
|
||||
}
|
||||
logger.Info("Using internal listening port %s", listeningPort)
|
||||
|
||||
rootURL, err := envParams.GetRootURL(params.Default("/"))
|
||||
if err != nil {
|
||||
logger.Error(err)
|
||||
return 1
|
||||
}
|
||||
logger.Info("Using root URL %q", rootURL)
|
||||
|
||||
productionRouter := http.NewServeMux()
|
||||
productionRouter.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
||||
urlStackeditPath := strings.TrimPrefix(r.URL.Path, rootURL)
|
||||
filepath := "/dist/" + urlStackeditPath
|
||||
switch urlStackeditPath {
|
||||
case "/conf", "/app/conf":
|
||||
bytes := getAllStackeditEnv()
|
||||
if _, err := w.Write(bytes); err != nil {
|
||||
logger.Error(err)
|
||||
}
|
||||
return
|
||||
case "/":
|
||||
filepath = "/static/landing/"
|
||||
case "/sitemap.xml":
|
||||
filepath = "/static/sitemap.xml"
|
||||
case "/oauth2/callback":
|
||||
filepath = "/static/oauth2/callback.html"
|
||||
case "/app", "/app/":
|
||||
filepath = "/dist/index.html"
|
||||
default:
|
||||
switch {
|
||||
case strings.HasPrefix(urlStackeditPath, "/static/css/static/fonts/"):
|
||||
filepath = "/dist/" + strings.TrimPrefix(urlStackeditPath, "/static/css/")
|
||||
case strings.HasPrefix(urlStackeditPath, "/app/static/css/static/fonts/"):
|
||||
filepath = "/dist/" + strings.TrimPrefix(urlStackeditPath, "/app/static/css/")
|
||||
case strings.HasPrefix(urlStackeditPath, "/app/"):
|
||||
filepath = "/dist/" + strings.TrimPrefix(urlStackeditPath, "/app/")
|
||||
}
|
||||
}
|
||||
http.ServeFile(w, r, "/html"+filepath)
|
||||
})
|
||||
healthcheckHandlerFunc := healthcheck.GetHandler(func() error { return nil })
|
||||
serverErrors := make(chan []error)
|
||||
go func() {
|
||||
serverErrors <- server.RunServers(ctx,
|
||||
server.Settings{Name: "production", Addr: "0.0.0.0:" + listeningPort, Handler: productionRouter},
|
||||
server.Settings{Name: "healthcheck", Addr: "127.0.0.1:9999", Handler: healthcheckHandlerFunc},
|
||||
)
|
||||
}()
|
||||
|
||||
osSignals := make(chan os.Signal, 1)
|
||||
signal.Notify(osSignals,
|
||||
syscall.SIGINT,
|
||||
syscall.SIGTERM,
|
||||
os.Interrupt,
|
||||
)
|
||||
select {
|
||||
case errors := <-serverErrors:
|
||||
for _, err := range errors {
|
||||
logger.Error(err)
|
||||
}
|
||||
return 1
|
||||
case signal := <-osSignals:
|
||||
message := fmt.Sprintf("Stopping program: caught OS signal %q", signal)
|
||||
logger.Warn(message)
|
||||
return 2
|
||||
case <-ctx.Done():
|
||||
message := fmt.Sprintf("Stopping program: %s", ctx.Err())
|
||||
logger.Warn(message)
|
||||
return 1
|
||||
}
|
||||
}
|
||||
|
||||
// Returns all stackedit env values as JSON
|
||||
func getAllStackeditEnv() []byte {
|
||||
env := struct {
|
||||
DropboxAppKey string `json:"dropboxAppKey"`
|
||||
DropboxAppKeyFull string `json:"dropboxAppKeyFull"`
|
||||
GithubClientID string `json:"githubClientId"`
|
||||
GithubClientSecret string `json:"githubClientSecret"`
|
||||
GoogleClientID string `json:"googleClientId"`
|
||||
GoogleAPIKey string `json:"googleApiKey"`
|
||||
WordpressClientID string `json:"wordpressClientId"`
|
||||
PaypalReceiverEmail string `json:"paypalReceiverEmail"`
|
||||
AllowSponsorship bool `json:"allowSponsorship"`
|
||||
}{}
|
||||
envParams := params.NewEnvParams()
|
||||
env.DropboxAppKey, _ = envParams.GetEnv("DROPBOX_APP_KEY")
|
||||
env.DropboxAppKeyFull, _ = envParams.GetEnv("DROPBOX_APP_KEY_FULL")
|
||||
env.GithubClientID, _ = envParams.GetEnv("GITHUB_CLIENT_ID")
|
||||
env.GithubClientSecret, _ = envParams.GetEnv("GITHUB_CLIENT_SECRET")
|
||||
env.GoogleClientID, _ = envParams.GetEnv("GOOGLE_CLIENT_ID")
|
||||
env.GoogleAPIKey, _ = envParams.GetEnv("GOOGLE_API_KEY")
|
||||
env.WordpressClientID, _ = envParams.GetEnv("WORDPRESS_CLIENT_ID")
|
||||
env.PaypalReceiverEmail, _ = envParams.GetEnv("PAYPAL_RECEIVER_EMAIL")
|
||||
if len(env.PaypalReceiverEmail) > 0 {
|
||||
env.AllowSponsorship = true
|
||||
}
|
||||
b, _ := json.Marshal(env)
|
||||
return b
|
||||
}
|
||||
|
||||
Loading…
Reference in new issue