func main() {
version := flag.Bool("version", false, "current version")
flag.Parse()
if *version {
fmt.Println(VERSION)
os.Exit(0)
}
// All the parallelism are belong to us!
runtime.GOMAXPROCS(runtime.NumCPU())
// Load configuration file.
webhog.LoadConfig()
// Load DB instance.
webhog.LoadDB()
// Load route handlers
webhog.LoadRoutes()
}
func main() {
// All the parallelism are belong to us!
runtime.GOMAXPROCS(runtime.NumCPU())
// Load configuration file.
webhog.LoadConfig()
// Make a database connection.
webhog.ConnectDB()
// Load configuration options.
// webhog.LaunchConfig()
// Start the server.
m := martini.Classic()
m.Use(martini.Logger())
m.Use(render.Renderer())
// Middleware to make sure each request has a specified API key
m.Use(func(res http.ResponseWriter, req *http.Request, r render.Render) {
if req.Header.Get("X-API-KEY") != webhog.Config.ApiKey {
r.JSON(401, map[string]interface{}{"error": "Invalid API key."})
}
})
m.Post("/scrape", binding.Json(Url{}), func(url Url, r render.Render) {
entity, err := webhog.NewScraper(url.Url)
// dat, err := json.Marshal(entity)
if err != nil {
r.JSON(400, map[string]interface{}{"errors": err.Error()})
} else {
r.JSON(200, entity)
}
})
m.Run()
}
