package main import ( "bufio" "flag" "context" "net/http" "os" "time" "github.com/NyaaPantsu/nyaa/cache" "github.com/NyaaPantsu/nyaa/config" "github.com/NyaaPantsu/nyaa/db" "github.com/NyaaPantsu/nyaa/network" "github.com/NyaaPantsu/nyaa/router" "github.com/NyaaPantsu/nyaa/service/scraper" "github.com/NyaaPantsu/nyaa/service/torrent/metainfoFetcher" "github.com/NyaaPantsu/nyaa/service/user" "github.com/NyaaPantsu/nyaa/util/log" "github.com/NyaaPantsu/nyaa/util/publicSettings" "github.com/NyaaPantsu/nyaa/util/search" "github.com/NyaaPantsu/nyaa/util/signals" ) var buildversion string // RunServer runs webapp mainloop func RunServer(conf *config.Config) { // TODO Use config from cli os.Mkdir(router.DatabaseDumpPath, 700) // TODO Use config from cli os.Mkdir(router.GPGPublicKeyPath, 700) http.Handle("/", router.Router) // Set up server, srv := &http.Server{ WriteTimeout: 30 * time.Second, ReadTimeout: 10 * time.Second, } l, err := network.CreateHTTPListener(conf) log.CheckError(err) if err != nil { return } log.Infof("listening on %s", l.Addr()) // http.Server.Shutdown closes associated listeners/clients. // context.Background causes srv to indefinitely try to // gracefully shutdown. add a timeout if this becomes a problem. signals.OnInterrupt(func() { srv.Shutdown(context.Background()) }) err = srv.Serve(l) if err == nil { log.Panic("http.Server.Serve never returns nil") } if err == http.ErrServerClosed { return } log.CheckError(err) } // RunScraper runs tracker scraper mainloop func RunScraper(conf *config.Config) { // bind to network pc, err := network.CreateScraperSocket(conf) if err != nil { log.Fatalf("failed to bind udp socket for scraper: %s", err) } // configure tracker scraperv var scraper *scraperService.Scraper scraper, err = scraperService.New(&conf.Scrape) if err != nil { pc.Close() log.Fatalf("failed to configure scraper: %s", err) } workers := conf.Scrape.NumWorkers if workers < 1 { workers = 1 } signals.OnInterrupt(func() { pc.Close() scraper.Close() }) // run udp scraper worker for workers > 0 { log.Infof("starting up worker %d", workers) go scraper.RunWorker(pc) workers-- } // run scraper go scraper.Run() scraper.Wait() } // RunMetainfoFetcher runs the database filesize fetcher main loop func RunMetainfoFetcher(conf *config.Config) { fetcher, err := metainfoFetcher.New(&conf.MetainfoFetcher) if err != nil { log.Fatalf("failed to start fetcher, %s", err) return } signals.OnInterrupt(func() { fetcher.Close() }) fetcher.RunAsync() fetcher.Wait() } func main() { conf := config.Conf if buildversion != "" { conf.Build = buildversion } else { conf.Build = "unknown" } processFlags := conf.BindFlags() defaults := flag.Bool("print-defaults", false, "print the default configuration file on stdout") mode := flag.String("mode", "webapp", "which mode to run daemon in, either webapp, scraper or metainfo_fetcher") flag.Float64Var(&conf.Cache.Size, "c", config.Conf.Cache.Size, "size of the search cache in MB") flag.Parse() if *defaults { stdout := bufio.NewWriter(os.Stdout) err := conf.Pretty(stdout) if err != nil { log.Fatal(err.Error()) } err = stdout.Flush() if err != nil { log.Fatal(err.Error()) } os.Exit(0) } else { err := processFlags() if err != nil { log.CheckError(err) } db.ORM, err = db.GormInit(conf, db.DefaultLogger) if err != nil { log.Fatal(err.Error()) } err = publicSettings.InitI18n(conf.I18n, userService.NewCurrentUserRetriever()) if err != nil { log.Fatal(err.Error()) } err = cache.Configure(&conf.Cache) if err != nil { log.Fatal(err.Error()) } err = search.Configure(&conf.Search) if err != nil { log.Fatal(err.Error()) } signals.Handle() if len(config.Conf.Torrents.FileStorage) > 0 { err := os.MkdirAll(config.Conf.Torrents.FileStorage, 0700) if err != nil { log.Fatal(err.Error()) } } if *mode == "scraper" { RunScraper(conf) } else if *mode == "webapp" { RunServer(conf) } else if *mode == "metainfo_fetcher" { RunMetainfoFetcher(conf) } else { log.Fatalf("invalid runtime mode: %s", *mode) } } }