145 lines
4.6 KiB
Go
145 lines
4.6 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"flag"
|
|
"html/template"
|
|
"net/http"
|
|
"runtime"
|
|
"time"
|
|
|
|
"code.google.com/p/go.net/websocket"
|
|
"github.com/drone/drone/server/channel"
|
|
"github.com/drone/drone/server/database"
|
|
"github.com/drone/drone/server/handler"
|
|
"github.com/drone/drone/server/queue"
|
|
"github.com/drone/drone/server/render"
|
|
"github.com/drone/drone/server/resource/commit"
|
|
"github.com/drone/drone/server/resource/config"
|
|
"github.com/drone/drone/server/resource/perm"
|
|
"github.com/drone/drone/server/resource/repo"
|
|
"github.com/drone/drone/server/resource/user"
|
|
"github.com/drone/drone/server/session"
|
|
"github.com/drone/drone/shared/build/docker"
|
|
"github.com/drone/drone/shared/build/log"
|
|
|
|
"github.com/gorilla/pat"
|
|
//"github.com/justinas/nosurf"
|
|
"github.com/GeertJohan/go.rice"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
"github.com/russross/meddler"
|
|
)
|
|
|
|
var (
|
|
// port the server will run on
|
|
port string
|
|
|
|
// database driver used to connect to the database
|
|
driver string
|
|
|
|
// driver specific connection information. In this
|
|
// case, it should be the location of the SQLite file
|
|
datasource string
|
|
|
|
// optional flags for tls listener
|
|
sslcert string
|
|
sslkey string
|
|
|
|
// commit sha for the current build.
|
|
version string = "0.2-dev"
|
|
revision string
|
|
|
|
// build will timeout after N milliseconds.
|
|
// this will default to 500 minutes (6 hours)
|
|
timeout time.Duration
|
|
|
|
// Number of concurrent build workers to run
|
|
// default to number of CPUs on machine
|
|
workers int
|
|
)
|
|
|
|
// drone cofiguration data, loaded from the
|
|
// $HOME/.drone/config.toml file.
|
|
var conf config.Config
|
|
|
|
func main() {
|
|
|
|
log.SetPriority(log.LOG_NOTICE)
|
|
|
|
// parse command line flags
|
|
flag.StringVar(&port, "port", ":8080", "")
|
|
flag.StringVar(&driver, "driver", "sqlite3", "")
|
|
flag.StringVar(&datasource, "datasource", "drone.sqlite", "")
|
|
flag.StringVar(&sslcert, "sslcert", "", "")
|
|
flag.StringVar(&sslkey, "sslkey", "", "")
|
|
flag.DurationVar(&timeout, "timeout", 300*time.Minute, "")
|
|
flag.IntVar(&workers, "workers", runtime.NumCPU(), "")
|
|
flag.Parse()
|
|
|
|
// parse the template files
|
|
// TODO we need to retrieve these from go.rice
|
|
//templ := template.Must(
|
|
// template.New("_").Funcs(render.FuncMap).ParseGlob("template/html/*.html"),
|
|
//).ExecuteTemplate
|
|
|
|
templateBox := rice.MustFindBox("template/html")
|
|
templateFiles := []string{"login.html", "repo_branch.html", "repo_commit.html", "repo_conf.html", "repo_feed.html", "user_conf.html", "user_feed.html", "user_login.html", "user_repos.html", "404.html", "400.html"}
|
|
templ := template.New("_").Funcs(render.FuncMap)
|
|
for _, file := range templateFiles {
|
|
templateData, _ := templateBox.String(file)
|
|
templ, _ = templ.New(file).Parse(templateData)
|
|
}
|
|
|
|
// setup the database
|
|
meddler.Default = meddler.SQLite
|
|
db, _ := sql.Open(driver, datasource)
|
|
database.Load(db)
|
|
|
|
// setup the database managers
|
|
repos := repo.NewManager(db)
|
|
users := user.NewManager(db)
|
|
perms := perm.NewManager(db)
|
|
commits := commit.NewManager(db)
|
|
|
|
// cancel all previously running builds
|
|
go commits.CancelAll()
|
|
|
|
// setup the build queue
|
|
queueRunner := queue.NewBuildRunner(docker.New(), timeout)
|
|
queue := queue.Start(workers, commits, queueRunner)
|
|
|
|
// setup the session managers
|
|
sess := session.NewSession(users)
|
|
|
|
// setup the router and register routes
|
|
router := pat.New()
|
|
handler.NewUsersHandler(users, sess).Register(router)
|
|
handler.NewUserHandler(users, repos, commits, sess).Register(router)
|
|
handler.NewHookHandler(users, repos, commits, &conf, queue).Register(router)
|
|
handler.NewLoginHandler(users, repos, perms, sess, &conf).Register(router)
|
|
handler.NewCommitHandler(repos, commits, perms, sess, queue).Register(router)
|
|
handler.NewBranchHandler(repos, commits, perms, sess).Register(router)
|
|
handler.NewRepoHandler(repos, commits, perms, sess, &conf).Register(router)
|
|
handler.NewBadgeHandler(repos, commits).Register(router)
|
|
handler.NewConfigHandler(conf, sess).Register(router)
|
|
handler.NewSiteHandler(users, repos, commits, perms, sess, templ.ExecuteTemplate).Register(router)
|
|
|
|
// serve static assets
|
|
// TODO we need to replace this with go.rice
|
|
http.Handle("/static/", http.StripPrefix("/static/", http.FileServer(rice.MustFindBox("static/").HTTPBox())))
|
|
|
|
// server websocket data
|
|
http.Handle("/feed", websocket.Handler(channel.Read))
|
|
|
|
// register the router
|
|
// TODO we disabled nosurf because it was impacting API calls.
|
|
// we need to disable nosurf for api calls (ie not coming from website).
|
|
http.Handle("/", router)
|
|
|
|
// start webserver using HTTPS or HTTP
|
|
if len(sslcert) != 0 && len(sslkey) != 0 {
|
|
panic(http.ListenAndServeTLS(port, sslcert, sslkey, nil))
|
|
} else {
|
|
panic(http.ListenAndServe(port, nil))
|
|
}
|
|
}
|