From 8c4a9f272d1d98fdebc55d4d33857884f19dbd3d Mon Sep 17 00:00:00 2001
From: Arjun Lall <lallarjun@gmail.com>
Date: Thu, 28 Nov 2024 15:44:00 -0800
Subject: [PATCH] Add WARN log level

---
 Makefile               | 2 +-
 scripts/install.sh     | 2 +-
 src/logger.go          | 8 ++++++++
 src/main.go            | 2 +-
 src/select_remapper.go | 2 +-
 5 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/Makefile b/Makefile
index 646e21e..a41ca79 100644
--- a/Makefile
+++ b/Makefile
@@ -1,5 +1,5 @@
 sh:
-	devbox shell
+	devbox --env-file .env shell
 
 install:
 	devbox run "cd src && go mod tidy"
diff --git a/scripts/install.sh b/scripts/install.sh
index dc55785..bdb7dbb 100755
--- a/scripts/install.sh
+++ b/scripts/install.sh
@@ -1,6 +1,6 @@
 #!/bin/bash
 
-VERSION="0.14.3"
+VERSION="0.14.4"
 
 # Detect OS and architecture
 OS=$(uname -s | tr '[:upper:]' '[:lower:]')
diff --git a/src/logger.go b/src/logger.go
index 676e472..09ac4fa 100644
--- a/src/logger.go
+++ b/src/logger.go
@@ -8,12 +8,14 @@ type LogLevel string
 
 const (
 	LOG_LEVEL_DEBUG = "DEBUG"
+	LOG_LEVEL_WARN  = "WARN"
 	LOG_LEVEL_INFO  = "INFO"
 	LOG_LEVEL_ERROR = "ERROR"
 )
 
 var LOG_LEVELS = []string{
 	LOG_LEVEL_DEBUG,
+	LOG_LEVEL_WARN,
 	LOG_LEVEL_INFO,
 	LOG_LEVEL_ERROR,
 }
@@ -22,6 +24,12 @@ func LogError(config *Config, message ...interface{}) {
 	log.Println(append([]interface{}{"[ERROR]"}, message...)...)
 }
 
+func LogWarn(config *Config, message ...interface{}) {
+	if config.LogLevel == LOG_LEVEL_WARN || config.LogLevel == LOG_LEVEL_INFO || config.LogLevel == LOG_LEVEL_DEBUG {
+		log.Println(append([]interface{}{"[WARN]"}, message...)...)
+	}
+}
+
 func LogInfo(config *Config, message ...interface{}) {
 	if config.LogLevel == LOG_LEVEL_INFO || config.LogLevel == LOG_LEVEL_DEBUG {
 		log.Println(append([]interface{}{"[INFO]"}, message...)...)
diff --git a/src/main.go b/src/main.go
index 00387ea..c5aa283 100644
--- a/src/main.go
+++ b/src/main.go
@@ -6,7 +6,7 @@ import (
 	"time"
 )
 
-const VERSION = "0.14.3"
+const VERSION = "0.14.4"
 
 func main() {
 	config := LoadConfig()
diff --git a/src/select_remapper.go b/src/select_remapper.go
index 466d177..abe5a41 100644
--- a/src/select_remapper.go
+++ b/src/select_remapper.go
@@ -53,7 +53,7 @@ func (selectRemapper *SelectRemapper) RemapQueryTreeWithSet(queryTree *pgQuery.P
 	setStatement := queryTree.Stmts[0].Stmt.GetVariableSetStmt()
 
 	if !KNOWN_SET_STATEMENTS.Contains(setStatement.Name) {
-		LogError(selectRemapper.config, "Unsupported SET ", setStatement.Name, ":", setStatement)
+		LogWarn(selectRemapper.config, "Unsupported SET ", setStatement.Name, ":", setStatement)
 	}
 
 	queryTree.Stmts[0].Stmt.GetVariableSetStmt().Name = "schema"