refactor to have all DB executive functions be in api/
This commit is contained in:
@@ -71,58 +71,3 @@ func WriteDB(db_name string, data interface{}) error {
|
||||
fmt.Printf("Successfully wrote to %s DB\n", db_name)
|
||||
return nil
|
||||
}
|
||||
|
||||
func AddUser(user map[string]interface{}) error {
|
||||
currentData := ExecDB("users")
|
||||
if currentData == nil {
|
||||
return fmt.Errorf("error reading users database")
|
||||
}
|
||||
|
||||
users := currentData["users"].([]interface{})
|
||||
users = append(users, user)
|
||||
return WriteDB("users", users)
|
||||
}
|
||||
|
||||
func AddMessage(message map[string]interface{}) error {
|
||||
currentData := ExecDB("messages")
|
||||
if currentData == nil {
|
||||
return fmt.Errorf("error reading messages database")
|
||||
}
|
||||
|
||||
messages := currentData["messages"].([]interface{})
|
||||
messages = append(messages, message)
|
||||
return WriteDB("messages", messages)
|
||||
}
|
||||
|
||||
func DeleteMessage(messageID string) error {
|
||||
currentData := ExecDB("messages")
|
||||
if currentData == nil {
|
||||
return fmt.Errorf("error reading messages database")
|
||||
}
|
||||
|
||||
messages, ok := currentData["messages"].([]interface{})
|
||||
if !ok {
|
||||
return fmt.Errorf("messages data is in an unexpected format")
|
||||
}
|
||||
|
||||
var updatedMessages []interface{}
|
||||
found := false
|
||||
|
||||
for _, value := range messages {
|
||||
if item, ok := value.(map[string]interface{}); ok {
|
||||
if id, exists := item["ID"]; exists {
|
||||
if idStr, ok := id.(string); ok && idStr == messageID {
|
||||
found = true
|
||||
continue
|
||||
}
|
||||
}
|
||||
}
|
||||
updatedMessages = append(updatedMessages, value)
|
||||
}
|
||||
|
||||
if !found {
|
||||
return fmt.Errorf("message with ID %s not found", messageID)
|
||||
}
|
||||
|
||||
return WriteDB("messages", updatedMessages)
|
||||
}
|
||||
|
Reference in New Issue
Block a user