package main import ( "flag" "html/template" "net/http" "github.com/gin-gonic/gin" "github.com/drone/drone/datastore/bolt" "github.com/drone/drone/remote/github" "github.com/drone/drone/server" "github.com/drone/drone/server/session" "github.com/drone/drone/settings" "github.com/elazarl/go-bindata-assetfs" eventbus "github.com/drone/drone/eventbus/builtin" queue "github.com/drone/drone/queue/builtin" ) var conf = flag.String("config", "drone.toml", "") func main() { flag.Parse() settings, err := settings.Parse(*conf) if err != nil { panic(err) } remote := github.New(settings.Service) session := session.New(settings.Session) ds := bolt.Must(settings.Database.Path) defer ds.Close() r := gin.Default() api := r.Group("/api") api.Use(server.SetHeaders()) api.Use(server.SetBus(eventbus.New())) api.Use(server.SetDatastore(ds)) api.Use(server.SetRemote(remote)) api.Use(server.SetQueue(queue.New())) api.Use(server.SetSettings(settings)) api.Use(server.SetSession(session)) api.Use(server.SetUser(session)) user := api.Group("/user") { user.Use(server.MustUser()) user.GET("", server.GetUserCurr) user.PATCH("", server.PutUserCurr) user.GET("/repos", server.GetUserRepos) user.GET("/tokens", server.GetUserTokens) user.POST("/tokens", server.PostToken) user.DELETE("/tokens/:label", server.DelToken) } users := api.Group("/users") { users.Use(server.MustAdmin()) users.GET("", server.GetUsers) users.GET("/:name", server.GetUser) users.POST("/:name", server.PostUser) users.PATCH("/:name", server.PutUser) users.DELETE("/:name", server.DeleteUser) } agents := api.Group("/agents") { agents.Use(server.MustAdmin()) agents.GET("/token", server.GetAgentToken) } repos := api.Group("/repos/:owner/:name") { repos.POST("", server.PostRepo) repo := repos.Group("") { repo.Use(server.SetRepo()) repo.Use(server.SetPerm()) repo.Use(server.CheckPull()) repo.Use(server.CheckPush()) repo.GET("", server.GetRepo) repo.PATCH("", server.PutRepo) repo.DELETE("", server.DeleteRepo) repo.POST("/watch", server.Subscribe) repo.DELETE("/unwatch", server.Unsubscribe) repo.GET("/builds", server.GetBuilds) repo.GET("/builds/:number", server.GetBuild) repo.POST("/builds/:number", server.RunBuild) repo.DELETE("/builds/:number", server.KillBuild) repo.GET("/logs/:number/:task", server.GetBuildLogs) repo.POST("/status/:number", server.PostBuildStatus) } } badges := api.Group("/badges/:owner/:name") { badges.Use(server.SetRepo()) badges.GET("/status.svg", server.GetBadge) badges.GET("/cc.xml", server.GetCC) } hooks := api.Group("/hook") { hooks.POST("", server.PostHook) } queue := api.Group("/queue") { queue.GET("", server.GetQueue) queue.POST("/pull", server.PollBuild) push := queue.Group("/push/:owner/:name") { push.Use(server.SetRepo()) push.POST("", server.PushBuild) push.POST("/:build", server.PushTask) push.POST("/:build/:task/logs", server.PushLogs) } } events := api.Group("/stream") { events.GET("/user", server.GetEvents) stream := events.Group("/logs") { stream.Use(server.SetRepo()) stream.Use(server.SetPerm()) stream.GET("/:owner/:name/:build/:number", server.GetStream) } } auth := r.Group("/authorize") { auth.Use(server.SetHeaders()) auth.Use(server.SetDatastore(ds)) auth.Use(server.SetRemote(remote)) auth.Use(server.SetSettings(settings)) auth.Use(server.SetSession(session)) auth.GET("", server.GetLogin) auth.POST("", server.GetLogin) } r.SetHTMLTemplate(index()) r.NoRoute(func(c *gin.Context) { c.HTML(200, "index.html", nil) }) http.Handle("/static/", static()) http.Handle("/", r) http.ListenAndServe(settings.Server.Addr, nil) } // static is a helper function that will setup handlers // for serving static files. func static() http.Handler { return http.StripPrefix("/static/", http.FileServer(&assetfs.AssetFS{ Asset: Asset, AssetDir: AssetDir, Prefix: "server/static", })) } // index is a helper function that will setup a template // for rendering the main angular index.html file. func index() *template.Template { file := MustAsset("server/static/index.html") filestr := string(file) return template.Must(template.New("index.html").Parse(filestr)) }