80 lines
1.5 KiB
Go
80 lines
1.5 KiB
Go
package server
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"path/filepath"
|
|
|
|
"github.com/drone/drone/model"
|
|
"github.com/drone/drone/server/template"
|
|
"github.com/drone/drone/shared/token"
|
|
)
|
|
|
|
type local struct {
|
|
dir string
|
|
fs http.Handler
|
|
}
|
|
|
|
// NewLocalWebsite returns a new website loader.
|
|
func NewLocalWebsite(path string) Website {
|
|
return &local{
|
|
dir: path,
|
|
fs: http.FileServer(
|
|
http.Dir(path),
|
|
),
|
|
}
|
|
}
|
|
|
|
// Page serves a page in the user interface.
|
|
func (w *local) Page(rw http.ResponseWriter, r *http.Request, u *model.User) {
|
|
rw.WriteHeader(200)
|
|
|
|
path := r.URL.Path
|
|
switch path {
|
|
case "/login":
|
|
if err := r.FormValue("error"); err != "" {
|
|
// TODO login error
|
|
} else {
|
|
http.Redirect(rw, r, "/authorize", 303)
|
|
}
|
|
|
|
default:
|
|
var csrf string
|
|
if u != nil {
|
|
csrf, _ = token.New(
|
|
token.CsrfToken,
|
|
u.Login,
|
|
).Sign(u.Hash)
|
|
}
|
|
params := map[string]interface{}{
|
|
"user": u,
|
|
"csrf": csrf,
|
|
}
|
|
|
|
index, err := ioutil.ReadFile(filepath.Join(w.dir, "index.html"))
|
|
if err != nil {
|
|
rw.WriteHeader(404)
|
|
return
|
|
}
|
|
var buf bytes.Buffer
|
|
template.T.ExecuteTemplate(&buf, "script.html", params)
|
|
index = bytes.Replace(index, []byte("<!-- inject:js -->"), buf.Bytes(), 1)
|
|
rw.Write(index)
|
|
}
|
|
}
|
|
|
|
// File serves a static file for the user interface.
|
|
func (w *local) File(rw http.ResponseWriter, r *http.Request) {
|
|
w.fs.ServeHTTP(rw, r)
|
|
}
|
|
|
|
func (w *local) Routes() []string {
|
|
return []string{
|
|
"/favicon-32x32.png",
|
|
"/favicon-16x16.png",
|
|
"/src/*filepath",
|
|
"/bower_components/*filepath",
|
|
"/static/*filepath",
|
|
}
|
|
}
|