From e564fbac4d0916525aab5e8fa102dc2d33203e5c Mon Sep 17 00:00:00 2001
From: Deividas Raila <deividas.raila@hostinger.com>
Date: Thu, 3 Oct 2024 10:23:21 +0300
Subject: [PATCH] src: Option to select loglevel

---
 shard.yml        |  2 +-
 src/razor.cr     | 33 ++++++++++++++++++++++++++-------
 tools/razor.json |  1 +
 3 files changed, 28 insertions(+), 8 deletions(-)

diff --git a/shard.yml b/shard.yml
index 0e7f9a9..9725016 100644
--- a/shard.yml
+++ b/shard.yml
@@ -1,5 +1,5 @@
 name: pdns-razor
-version: 0.5.5
+version: 0.5.6
 
 dependencies:
   redis:
diff --git a/src/razor.cr b/src/razor.cr
index 65abf9f..09c0b5a 100644
--- a/src/razor.cr
+++ b/src/razor.cr
@@ -25,12 +25,30 @@ class Razor
     @redis_unixsocket = "/tmp/redis.sock"
     @debug = false
     @log = Logger.new(STDERR)
+    @loglevel = 4
     @hash_method = "edns"
     @zone = nil
     @geoip_db_path = "/usr/share/GeoIP/GeoIP2-Country.mmdb"
 
     parse_arguments
 
+    case @loglevel
+    when 0
+      @log.level = Logger::DEBUG
+    when 1
+      @log.level = Logger::INFO
+    when 2
+      @log.level = Logger::WARN
+    when 3
+      @log.level = Logger::ERROR
+    when 4
+      @log.level = Logger::FATAL
+    when 5
+      @log.level = Logger::UNKNOWN
+    else
+      @log.level = Logger::FATAL
+    end
+
     @geoip = GeoIP2.open(@geoip_db_path)
     @geoip_db_checksum = geoip_db_checksum
     @redis = Redis.new(host: @redis_host,
@@ -56,8 +74,8 @@ class Razor
         exit
       end
       parser.invalid_option do |flag|
-        @log.error("#{flag} is not a valid option")
-        @log.error(parser)
+        @log.fatal("#{flag} is not a valid option")
+        @log.fatal(parser)
         exit(1)
       end
     end
@@ -66,7 +84,7 @@ class Razor
       config = JSON.parse(file)
 
       unless config["contexts"]
-        @log.error("can't parse JSON #{@config} configuration file")
+        @log.fatal("can't parse JSON #{@config} configuration file")
         exit(1)
       end
 
@@ -75,6 +93,7 @@ class Razor
 
         @banner = context["banner"].as_s if context["banner"]?
         @debug = context["debug"].as_bool if context["debug"]?
+        @loglevel = context["loglevel"].as_i if context["loglevel"]?
         @hash_method = context["hash_method"].as_s if context["hash_method"]?
         @redis_host = context["redis_host"].as_s if context["redis_host"]?
         @redis_port = context["redis_port"].as_i if context["redis_port"]?
@@ -84,7 +103,7 @@ class Razor
           @geoip_db_path = context["geoip_db_path"].as_s
         end
       rescue
-        @log.error("no such context found #{@context} or missing configuration in JSON")
+        @log.fatal("no such context found #{@context} or missing configuration in JSON")
         exit(1)
       end
 
@@ -99,6 +118,7 @@ class Razor
 
   def mainLoop
     Schedule.every(1.hour) do
+      @log.debug("Hourly schedule starts")
       geoip_db_check
     end
 
@@ -152,6 +172,7 @@ class Razor
     # reopen to handle this internally, without PowerDNS restart.
     new_geoip_db_checksum = geoip_db_checksum
     if @geoip_db_checksum != new_geoip_db_checksum
+      @log.debug("Reloading GeoIP database")
       @geoip = GeoIP2.open(@geoip_db_path)
       @geoip_db_checksum = new_geoip_db_checksum
     end
@@ -181,9 +202,7 @@ class Razor
       country = rec.country.iso_code
       state = rec.subdivisions.size > 0 ? rec.subdivisions[0].iso_code : nil
       if !continent && !country && !state
-        if @debug
-          @log.warn("No continent/country/state found for #{ip}")
-        end
+        @log.warn("No continent/country/state found for #{ip}")
         return nil, nil, nil
       end
       return continent, country, state
diff --git a/tools/razor.json b/tools/razor.json
index 761f8fb..c960068 100644
--- a/tools/razor.json
+++ b/tools/razor.json
@@ -5,6 +5,7 @@
       "zone": "example.org",
       "hash_method": "edns",
       "debug": true,
+      "loglevel": 0,
       "redis_host": "127.0.0.1",
       "redis_port": 6379,
       "redis_unixsocket": "/tmp/redis.sock",