brightdigit/OrchardNest

import Fluent
import FluentPostgresDriver
import Ink
import OrchardNestKit
import Plot
import QueuesFluentDriver
import Vapor
extension Date {
  func get(_ type: Calendar.Component) -> Int {
    let calendar = Calendar.current
    return calendar.component(type, from: self)
  }
}

//
public final class Configurator: ConfiguratorProtocol {
  public static let shared: ConfiguratorProtocol = Configurator()

  //
  ///// Called before your application initializes.
  public func configure(_ app: Application) throws {
    let html = HTMLController(markdownDirectory: app.directory.viewsDirectory)
    app.middleware = .init()
    app.middleware.use(ErrorPageMiddleware(htmlController: html))
    app.middleware.use(FileMiddleware(publicDirectory: app.directory.publicDirectory))

    // Configure a SQLite database
    let postgreSQLConfig: PostgresConfiguration

    if let url = Environment.get("DATABASE_URL") {
      postgreSQLConfig = PostgresConfiguration(url: url)!
    } else {
      postgreSQLConfig = PostgresConfiguration(hostname: "localhost", username: "orchardnest")
    }

    app.databases.use(.postgres(configuration: postgreSQLConfig, maxConnectionsPerEventLoop: 8, connectionPoolTimeout: .seconds(60)), as: .psql)
    app.migrations.add([
      CategoryMigration(),
      LanguageMigration(),
      CategoryTitleMigration(),
      ChannelMigration(),
      EntryMigration(),
      PodcastEpisodeMigration(),
      YouTubeChannelMigration(),
      YouTubeVideoMigration(),
      PodcastChannelMigration(),
      ChannelStatusMigration(),
      LatestEntriesMigration(),
      JobModelMigrate(schema: "queue_jobs")
    ])

    app.queues.configuration.refreshInterval = .seconds(25)
    app.queues.use(.fluent())

    app.queues.add(RefreshJob())
    app.queues.schedule(RefreshJob()).daily().at(.midnight)
    app.queues.schedule(RefreshJob()).daily().at(7, 30)
    app.queues.schedule(RefreshJob()).daily().at(19, 30)
    #if DEBUG
      if !app.environment.isRelease {
        let minute = Date().get(.minute)
        [0, 30].map { ($0 + minute + 5).remainderReportingOverflow(dividingBy: 60).partialValue }.forEach { minute in
          app.queues.schedule(RefreshJob()).hourly().at(.init(integerLiteral: minute))
        }
      }
    #endif
    try app.queues.startInProcessJobs(on: .default)
    app.commands.use(RefreshCommand(help: "Imports data into the database"), as: "refresh")

    try app.autoMigrate().wait()
    //   services.register(wss, as: WebSocketServer.self)

    let api = app.grouped("api", "v1")

    try app.register(collection: html)
    try api.grouped("entires").register(collection: EntryController())
    try api.grouped("channels").register(collection: ChannelController())
    try api.grouped("categories").register(collection: CategoryController())

    app.post("jobs") { req in
      req.queue.dispatch(
        RefreshJob.self,
        RefreshConfiguration()
      ).map { HTTPStatus.created }
    }
  }
}