contented/Server.go

132 lines
3.0 KiB
Go

package contented
import (
"bytes"
"encoding/json"
"log"
"net/http"
"strings"
"time"
"github.com/boltdb/bolt"
"github.com/mxk/go-flowrate/flowrate"
)
var SERVER_HEADER string = `contented/0.0.0-dev`
type ServerPublicProperties struct {
AppTitle string
MaxUploadBytes int64
}
type ServerOptions struct {
DataDirectory string
DBPath string
BandwidthLimit int64
TrustXForwardedFor bool
ServerPublicProperties
}
type Server struct {
opts ServerOptions
db *bolt.DB
startTime time.Time
metadataBucket []byte
}
func NewServer(opts *ServerOptions) (*Server, error) {
s := &Server{
opts: *opts,
metadataBucket: []byte(`METADATA`),
startTime: time.Now(),
}
b, err := bolt.Open(opts.DBPath, 0644, bolt.DefaultOptions)
if err != nil {
return nil, err
}
err = b.Update(func(tx *bolt.Tx) error {
_, err := tx.CreateBucketIfNotExists(s.metadataBucket)
return err
})
if err != nil {
return nil, err
}
s.db = b
return s, nil
}
func (this *Server) serveJsonObject(w http.ResponseWriter, o interface{}) {
jb, err := json.Marshal(o)
if err != nil {
log.Println(err.Error())
http.Error(w, "Internal error", 500)
return
}
w.Header().Set(`Content-Type`, `application/json`)
w.WriteHeader(200)
w.Write(jb)
}
func (this *Server) handleAbout(w http.ResponseWriter) {
this.serveJsonObject(w, this.opts.ServerPublicProperties)
}
func (this *Server) remoteIP(r *http.Request) string {
if this.opts.TrustXForwardedFor {
if xff := r.Header.Get("X-Forwarded-For"); len(xff) > 0 {
return xff
}
}
return strings.TrimRight(strings.TrimRight(r.RemoteAddr, "0123456789"), ":")
}
const (
downloadUrlPrefix = `/get/`
metadataUrlPrefix = `/info/`
)
func (this *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
w.Header().Set(`Server`, SERVER_HEADER)
w.Header().Set(`Access-Control-Allow-Origin`, `*`) // Blanket allow CORS
if this.opts.MaxUploadBytes > 0 {
r.Body = http.MaxBytesReader(w, r.Body, this.opts.MaxUploadBytes)
}
if this.opts.BandwidthLimit > 0 {
r.Body = flowrate.NewReader(r.Body, this.opts.BandwidthLimit)
}
//
if r.Method == "GET" && strings.HasPrefix(r.URL.Path, downloadUrlPrefix) {
this.handleView(w, r, r.URL.Path[len(downloadUrlPrefix):])
} else if r.Method == "GET" && strings.HasPrefix(r.URL.Path, metadataUrlPrefix) {
this.handleInformation(w, r.URL.Path[len(metadataUrlPrefix):])
} else if r.Method == "GET" && r.URL.Path == `/about` {
this.handleAbout(w)
} else if r.Method == "POST" && r.URL.Path == `/upload` {
this.handleUpload(w, r)
} else if r.Method == "OPTIONS" {
// Blanket allow (headers already set)
w.WriteHeader(200)
} else if r.Method == "GET" && r.URL.Path == `/` {
http.Redirect(w, r, `/index.html`, http.StatusFound)
} else if static, err := Asset(r.URL.Path[1:]); err == nil && r.Method == "GET" {
http.ServeContent(w, r, r.URL.Path[1:], this.startTime, bytes.NewReader(static))
} else {
http.Error(w, "Not found", 404)
}
}