package db

import (
	"encoding/json"
	"fmt"
	"io"
	"os"
)

func ExecDB(db_name string) map[string]interface{} {
	var result map[string]interface{}

	if db_name == "users" {
		users_db, err := os.Open("./test_data/users.json")
		if err != nil {
			fmt.Println(err)
			return nil
		}
		fmt.Println("Successfully opened Users DB")
		defer users_db.Close()

		byteValue, _ := io.ReadAll(users_db)
		var users []interface{}
		json.Unmarshal(byteValue, &users)
		result = map[string]interface{}{"users": users}

	} else if db_name == "messages" {
		messages_db, err := os.Open("./test_data/messages.json")
		if err != nil {
			fmt.Println(err)
			return nil
		}
		fmt.Println("Successfully opened Messages DB")
		defer messages_db.Close()

		byteValue, _ := io.ReadAll(messages_db)
		var messages []interface{}
		json.Unmarshal(byteValue, &messages)
		result = map[string]interface{}{"messages": messages}

	} else {
		fmt.Println("Invalid DB name")
		return nil
	}

	return result
}