153 lines
3.5 KiB
Go
153 lines
3.5 KiB
Go
package appservice
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"github.com/gorilla/mux"
|
|
"context"
|
|
"time"
|
|
)
|
|
|
|
// Listen starts the HTTP server that listens for calls from the Matrix homeserver.
|
|
func (as *AppService) Start() {
|
|
r := mux.NewRouter()
|
|
r.HandleFunc("/transactions/{txnID}", as.PutTransaction).Methods(http.MethodPut)
|
|
r.HandleFunc("/rooms/{roomAlias}", as.GetRoom).Methods(http.MethodGet)
|
|
r.HandleFunc("/users/{userID}", as.GetUser).Methods(http.MethodGet)
|
|
|
|
var err error
|
|
as.server = &http.Server{
|
|
Addr: as.Host.Address(),
|
|
Handler: r,
|
|
}
|
|
as.Log.Infoln("Listening on", as.Host.Address())
|
|
if len(as.Host.TLSCert) == 0 || len(as.Host.TLSKey) == 0 {
|
|
err = as.server.ListenAndServe()
|
|
} else {
|
|
err = as.server.ListenAndServeTLS(as.Host.TLSCert, as.Host.TLSKey)
|
|
}
|
|
if err != nil && err.Error() != "http: Server closed" {
|
|
as.Log.Fatalln("Error while listening:", err)
|
|
} else {
|
|
as.Log.Debugln("Listener stopped.")
|
|
}
|
|
}
|
|
|
|
func (as *AppService) Stop() {
|
|
if as.server == nil {
|
|
return
|
|
}
|
|
|
|
ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
|
|
as.server.Shutdown(ctx)
|
|
as.server = nil
|
|
}
|
|
|
|
// CheckServerToken checks if the given request originated from the Matrix homeserver.
|
|
func (as *AppService) CheckServerToken(w http.ResponseWriter, r *http.Request) bool {
|
|
query := r.URL.Query()
|
|
val, ok := query["access_token"]
|
|
if !ok {
|
|
Error{
|
|
ErrorCode: ErrForbidden,
|
|
HTTPStatus: http.StatusForbidden,
|
|
Message: "Bad token supplied.",
|
|
}.Write(w)
|
|
return false
|
|
}
|
|
for _, str := range val {
|
|
return str == as.Registration.ServerToken
|
|
}
|
|
return false
|
|
}
|
|
|
|
// PutTransaction handles a /transactions PUT call from the homeserver.
|
|
func (as *AppService) PutTransaction(w http.ResponseWriter, r *http.Request) {
|
|
if !as.CheckServerToken(w, r) {
|
|
return
|
|
}
|
|
|
|
vars := mux.Vars(r)
|
|
txnID := vars["txnID"]
|
|
if len(txnID) == 0 {
|
|
Error{
|
|
ErrorCode: ErrNoTransactionID,
|
|
HTTPStatus: http.StatusBadRequest,
|
|
Message: "Missing transaction ID.",
|
|
}.Write(w)
|
|
return
|
|
}
|
|
defer r.Body.Close()
|
|
body, err := ioutil.ReadAll(r.Body)
|
|
if err != nil || len(body) == 0 {
|
|
Error{
|
|
ErrorCode: ErrNoBody,
|
|
HTTPStatus: http.StatusBadRequest,
|
|
Message: "Missing request body.",
|
|
}.Write(w)
|
|
return
|
|
}
|
|
if as.lastProcessedTransaction == txnID {
|
|
// Duplicate transaction ID: no-op
|
|
WriteBlankOK(w)
|
|
return
|
|
}
|
|
|
|
eventList := EventList{}
|
|
err = json.Unmarshal(body, &eventList)
|
|
if err != nil {
|
|
Error{
|
|
ErrorCode: ErrInvalidJSON,
|
|
HTTPStatus: http.StatusBadRequest,
|
|
Message: "Failed to parse body JSON.",
|
|
}.Write(w)
|
|
return
|
|
}
|
|
|
|
for _, event := range eventList.Events {
|
|
as.Log.Debugln("Received event", event.ID)
|
|
as.UpdateState(event)
|
|
as.Events <- event
|
|
}
|
|
as.lastProcessedTransaction = txnID
|
|
WriteBlankOK(w)
|
|
}
|
|
|
|
// GetRoom handles a /rooms GET call from the homeserver.
|
|
func (as *AppService) GetRoom(w http.ResponseWriter, r *http.Request) {
|
|
if !as.CheckServerToken(w, r) {
|
|
return
|
|
}
|
|
|
|
vars := mux.Vars(r)
|
|
roomAlias := vars["roomAlias"]
|
|
ok := as.QueryHandler.QueryAlias(roomAlias)
|
|
if ok {
|
|
WriteBlankOK(w)
|
|
} else {
|
|
Error{
|
|
ErrorCode: ErrUnknown,
|
|
HTTPStatus: http.StatusNotFound,
|
|
}.Write(w)
|
|
}
|
|
}
|
|
|
|
// GetUser handles a /users GET call from the homeserver.
|
|
func (as *AppService) GetUser(w http.ResponseWriter, r *http.Request) {
|
|
if !as.CheckServerToken(w, r) {
|
|
return
|
|
}
|
|
|
|
vars := mux.Vars(r)
|
|
userID := vars["userID"]
|
|
ok := as.QueryHandler.QueryUser(userID)
|
|
if ok {
|
|
WriteBlankOK(w)
|
|
} else {
|
|
Error{
|
|
ErrorCode: ErrUnknown,
|
|
HTTPStatus: http.StatusNotFound,
|
|
}.Write(w)
|
|
}
|
|
}
|