From eb1d1e30d1cdcaceb64e6bbba56c64a41269147e Mon Sep 17 00:00:00 2001
From: Omar Roth <omarroth@hotmail.com>
Date: Fri, 9 Mar 2018 12:42:23 -0600
Subject: [PATCH] Add config file

---
 config/config.yml |  8 ++++++++
 src/helpers.cr    | 14 ++++++++++++++
 src/invidious.cr  | 18 +++++++++++++++---
 3 files changed, 37 insertions(+), 3 deletions(-)
 create mode 100644 config/config.yml

diff --git a/config/config.yml b/config/config.yml
new file mode 100644
index 00000000..f8a06faa
--- /dev/null
+++ b/config/config.yml
@@ -0,0 +1,8 @@
+pool_size: 10
+threads: 5
+db:
+  user: kemal
+  password: kemal
+  host: localhost
+  port: 5432
+  dbname: invidious
\ No newline at end of file
diff --git a/src/helpers.cr b/src/helpers.cr
index 5b3d6625..7ca5bd61 100644
--- a/src/helpers.cr
+++ b/src/helpers.cr
@@ -13,6 +13,20 @@ macro templated(filename)
   render "src/views/#{{{filename}}}.ecr", "src/views/layout.ecr"
 end
 
+class Config
+  YAML.mapping({
+    pool_size: Int32,
+    threads:   Int32,
+    db:        NamedTuple(
+      user: String,
+      password: String,
+      host: String,
+      port: Int32,
+      dbname: String,
+    ),
+  })
+end
+
 class Video
   module HTTPParamConverter
     def self.from_rs(rs)
diff --git a/src/invidious.cr b/src/invidious.cr
index cdd0b50a..1474fa40 100644
--- a/src/invidious.cr
+++ b/src/invidious.cr
@@ -18,10 +18,13 @@ require "kemal"
 require "option_parser"
 require "pg"
 require "xml"
+require "yaml"
 require "./helpers"
 
-pool_size = 10
-threads = 5
+CONFIG = Config.from_yaml(File.read("config/config.yml"))
+
+pool_size = CONFIG.pool_size
+threads = CONFIG.threads
 
 Kemal.config.extra_options do |parser|
   parser.banner = "Usage: invidious [arguments]"
@@ -45,7 +48,16 @@ end
 
 Kemal::CLI.new
 
-PG_DB      = DB.open "postgres://kemal:kemal@localhost:5432/invidious"
+PG_URL = URI.new(
+  scheme: "postgres",
+  user: CONFIG.db[:user],
+  password: CONFIG.db[:password],
+  host: CONFIG.db[:host],
+  port: CONFIG.db[:port],
+  path: CONFIG.db[:dbname],
+)
+
+PG_DB      = DB.open PG_URL
 YT_URL     = URI.parse("https://www.youtube.com")
 REDDIT_URL = URI.parse("https://api.reddit.com")