query: utils_private

Author: ThomasTJdev

src/sqlbuilderpkg/utils_private.nim

@@ -1,5 +1,31 @@
 # Copyright Thomas T. Jarløv (TTJ) - ttj@ttj.dk
 
+when NimMajor >= 2:
+  import db_connector/db_common
+else:
+  import std/db_common
+
+
+import
+  std/strutils
+
+
+proc querycompare*(a, b: SqlQuery): bool =
+  var
+    a1: seq[string]
+    b1: seq[string]
+  for c in splitWhitespace(string(a)):
+    a1.add($c)
+  for c in splitWhitespace(string(b)):
+    b1.add($c)
+
+  if a1 != b1:
+    echo ""
+    echo "a1: ", string(a)
+    echo "b1: ", string(b).replace("\n", " ").splitWhitespace().join(" ")
+    echo ""
+
+  return a1 == b1
 
 
 proc dbQuotePrivate*(s: string): string =
@@ -10,4 +36,106 @@ proc dbQuotePrivate*(s: string): string =
     of '\'': add(result, "''")
     of '\0': add(result, "\\0")
     else: add(result, c)
-  add(result, '\'')
+  add(result, '\'')
+
+
+
+proc formatWhereParams*(v: string): string =
+  ## Format the WHERE part of the query.
+  let
+    field = v.strip()
+
+  var fieldSplit: seq[string]
+  if field.contains(" "):
+    if field.contains("="):
+      fieldSplit = field.split("=")
+    elif field.contains("IS NOT"):
+      fieldSplit = field.split("IS NOT")
+    elif field.contains("IS"):
+      fieldSplit = field.split("IS")
+    elif field.contains("NOT IN"):
+      fieldSplit = field.split("NOT IN")
+    elif field.contains("IN"):
+      fieldSplit = field.split("IN")
+    elif field.contains("!="):
+      fieldSplit = field.split("!=")
+    elif field.contains("<="):
+      fieldSplit = field.split("<=")
+    elif field.contains(">="):
+      fieldSplit = field.split(">=")
+    elif field.contains("<"):
+      fieldSplit = field.split("<")
+    elif field.contains(">"):
+      fieldSplit = field.split(">")
+  else:
+    fieldSplit = field.split("=")
+
+  #
+  # Does the data have a `=` sign?
+  #
+  if fieldSplit.len() == 2:
+    #
+    # If the data is only having equal but no value, insert a `?` sign
+    #
+    if fieldSplit[1] == "":
+      return (field & " ?")
+    #
+    # Otherwise just add the data as is, eg. `field = value`
+    #
+    else:
+      return (field)
+
+  #
+  # Otherwise revert to default
+  #
+  else:
+    return (field & " = ?")
+
+
+
+proc hasIllegalFormats*(query: string): string =
+  const illegalFormats = [
+    "WHERE AND",
+    "WHERE OR",
+    "AND AND",
+    "OR OR",
+    "AND OR",
+    "OR AND",
+    "WHERE IN",
+    "WHERE =",
+    "WHERE >",
+    "WHERE <",
+    "WHERE !",
+    "WHERE LIKE",
+    "WHERE NOT",
+    "WHERE IS",
+    "WHERE NULL",
+    "WHERE ANY"
+  ]
+
+  for illegalFormat in illegalFormats:
+    if illegalFormat in query:
+      return illegalFormat
+
+
+  #
+  # Parentheses check
+  #
+  let
+    parentheseOpen = count(query, "(")
+    parentheseClose = count(query, ")")
+
+  if parentheseOpen > parentheseClose:
+    return "parentheses does not match. Missing closing parentheses. (" & $parentheseOpen & " open, " & $parentheseClose & " close)"
+  elif parentheseOpen < parentheseClose:
+    return "parentheses does not match. Missing opening parentheses. (" & $parentheseOpen & " open, " & $parentheseClose & " close)"
+
+
+  #
+  # Check for double insert
+  #
+  let noSpaces = query.strip().replace(" ", "")
+
+  if "??" in noSpaces:
+    return "double insert detected. (??)"
+