forked from hswaw/hscloud
83 lines
1.8 KiB
Go
83 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"flag"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"code.hackerspace.pl/hscloud/go/mirko"
|
|
"github.com/golang/glog"
|
|
)
|
|
|
|
var (
|
|
flagListenPublic string
|
|
flagDomain string
|
|
)
|
|
|
|
type wellKnownHomeserver struct {
|
|
BaseURL string `json:"base_url"`
|
|
}
|
|
|
|
type wellKnown struct {
|
|
Homeserver wellKnownHomeserver `json:"m.homeserver"`
|
|
Server string `json:"m.server"`
|
|
}
|
|
|
|
type server struct {
|
|
domain string
|
|
}
|
|
|
|
func (s *server) register(mux *http.ServeMux) {
|
|
mux.HandleFunc("/.well-known/matrix/server", s.makeHandler(true))
|
|
mux.HandleFunc("/.well-known/matrix/client", s.makeHandler(false))
|
|
}
|
|
|
|
func (s *server) makeHandler(server bool) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
wk := wellKnown{}
|
|
if server {
|
|
wk.Server = fmt.Sprintf("%s:443", s.domain)
|
|
} else {
|
|
wk.Homeserver = wellKnownHomeserver{
|
|
BaseURL: fmt.Sprintf("https://%s", s.domain),
|
|
}
|
|
}
|
|
err := json.NewEncoder(w).Encode(wk)
|
|
if err != nil {
|
|
glog.Errorf("When handling request: %v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
flag.StringVar(&flagListenPublic, "listen_public", "0.0.0.0:8080", "Address to listen at for well-known traffic")
|
|
flag.StringVar(&flagDomain, "domain", "matrix.hackerspace.pl", "Address to which clients and servers should connect")
|
|
flag.Parse()
|
|
|
|
m := mirko.New()
|
|
if err := m.Listen(); err != nil {
|
|
glog.Exitf("Listen(): %v", err)
|
|
}
|
|
|
|
mux := http.NewServeMux()
|
|
s := &server{
|
|
domain: flagDomain,
|
|
}
|
|
s.register(mux)
|
|
|
|
go func() {
|
|
glog.Infof("Listening on %s...", flagListenPublic)
|
|
if err := http.ListenAndServe(flagListenPublic, mux); err != nil {
|
|
glog.Exitf("listen failed: %v", err)
|
|
}
|
|
}()
|
|
|
|
if err := m.Serve(); err != nil {
|
|
glog.Exitf("Serve(): %v", err)
|
|
}
|
|
|
|
<-m.Done()
|
|
}
|