Database back-end 'read_db' action.

nthnn · Dec 31, 2023 · 2b38c1b · 2b38c1b
1 parent 093b1ab
commit 2b38c1b
Show file tree

Hide file tree

Showing 2 changed files with 50 additions and 1 deletion.
diff --git a/backend/database/callback.go b/backend/database/callback.go
@@ -87,7 +87,12 @@ func getByNameCallback(apiKey string, args []string) func(*sql.DB) {
 			query.Scan(&mode, &content)
 		}
 
-		proc.ShowResult("[\"" + mode + "\", \"" + content + "\"]")
+		if strings.Contains(mode, "r") {
+			proc.ShowResult("[\"" + mode + "\", \"" + content + "\"]")
+		} else {
+			proc.ShowResult("[\"" + mode + "\"]")
+		}
+
 		query.Close()
 	}
 }
@@ -179,6 +184,46 @@ func getDbModeCallback(apiKey string, args []string) func(*sql.DB) {
 	}
 }
 
+func readDbCallback(apiKey string, args []string) func(*sql.DB) {
+	return func(d *sql.DB) {
+		name := args[2]
+		query, err := d.Query("SELECT mode, content FROM " + apiKey + "_database WHERE name=\"" + name + "\"")
+
+		if err != nil {
+			proc.ShowFailedResponse("Internal error occured.")
+			query.Close()
+
+			return
+		}
+
+		count := 0
+		mode := ""
+		content := ""
+
+		for query.Next() {
+			query.Scan(&mode, &content)
+			count++
+		}
+
+		if count != 1 {
+			proc.ShowFailedResponse("Cannot resolve database name.")
+			query.Close()
+
+			return
+		}
+
+		if !strings.Contains(mode, "r") {
+			proc.ShowFailedResponse("Read operation denied.")
+			query.Close()
+
+			return
+		}
+
+		proc.ShowResult("\"" + content + "\"")
+		query.Close()
+	}
+}
+
 func writeDbCallback(apiKey string, args []string) func(*sql.DB) {
 	return func(d *sql.DB) {
 		name := args[2]

diff --git a/backend/database/main.go b/backend/database/main.go
@@ -45,6 +45,10 @@ func main() {
 		failOnUmatchedArgSize(4, args)
 		callback = writeDbCallback(apiKey, args)
 
+	case "read_db":
+		failOnUmatchedArgSize(3, args)
+		callback = readDbCallback(apiKey, args)
+
 	case "delete_db":
 		failOnUmatchedArgSize(3, args)
 		callback = deleteDbCallback(apiKey, args)