main.go

package main

import(
  "fmt"
  "os"
  "github.com/codegangsta/cli"
  "github.com/ricallinson/forgery"
  "github.com/ricallinson/stackr"
  "github.com/spacedock-io/registry/db"
  "github.com/spacedock-io/registry/router"
  "github.com/spacedock-io/registry/config"
  "github.com/spacedock-io/registry/models"
  "github.com/spacedock-io/registry/session"
  "github.com/spacedock-io/registry/cloudfiles"
  "github.com/Southern/logger"
  "github.com/Southern/middleware"
)

const VERSION = "0.0.1"

func main() {
  app := cli.NewApp()

  app.Name = "Registry"
  app.Usage = "Run a standalone Docker registry"
  app.Version = "0.0.1"
  app.Flags = []cli.Flag {
    cli.StringFlag{"port, p", "", "Port number"},
    cli.StringFlag{"index, i", "", "Index URL"},
    cli.StringFlag{"env, e", "dev", "Environment"},
    cli.StringFlag{"config, c", "", "Configuration directory"},
  }

  app.Action = func(c *cli.Context) {
    env := c.String("env")
    dir := c.String("config")
    index := c.String("index")

    if len(env) == 0 {
      env = "dev"
    }
    if len(dir) > 0 {
      config.Dir = dir
    }

    config.Global = config.Load(env)
    config.Logger = logger.New()

    if len(index) > 0 {
      config.Global = config.Global.Set("index", index)
    }

    server := f.CreateServer()
    server.Use(func(req *stackr.Request, res *stackr.Response, next func()) {
      config.Logger.Log(fmt.Sprintf("%s %s", req.Method, req.Url))
      next()
    })
    server.Use(middleware.BodyParser)
    server.Use(func (req *stackr.Request, res *stackr.Response, next func()) {
      defer next()

      res.SetHeader("X-Docker-Registry-Version", VERSION)
      res.SetHeader("X-Docker-Registry-Config", "dev")
    })
    server.Use(sx.Middleware("SECRETVERYSECRET"))
    server.Use(f.ErrorHandler())

    port := c.Int("port")
    if port == 0 {
      // Bug(Colton): Not quite sure why port is being picked up as Float64 at
      // the moment. Still looking into this. It may be intended functionality.
      port = int(config.Global.Get("port").Float64())
    }

    db.New(config.Global)
    db.DB.AutoMigrate(&models.Image{})
    db.DB.AutoMigrate(&models.Tag{})
    // db.DB.AutoMigrate(&models.Ancestor{})

    cloudfiles.New(config.Global)

    router.Routes(server)
    config.Logger.Log("Registry listening on port " + fmt.Sprint(port))
    server.Listen(port)
  }

  app.Run(os.Args)
}