import pymongo
from datetime import datetime
from config import MONGODB_URL, MONGODB_USERNAME, MONGODB_PASSWORD, MONGODB_DATABASE
from utils import get_password_hash

client = pymongo.MongoClient(MONGODB_URL, username=MONGODB_USERNAME, password=MONGODB_PASSWORD)
db = client[MONGODB_DATABASE]

def populate_data():
    users_collection = db["users"]
    pins_collection = db["pins"]

    user1_id = users_collection.insert_one({
        "username": "string",
        "password": get_password_hash("string")
    }).inserted_id

    user2_id = users_collection.insert_one({
        "username": "test",
        "password": get_password_hash("test")
    }).inserted_id

    pins_collection.insert_many([
        {
            "title": f"Tour Eiffel",
            "description": "Description A",
            "location": [48.858296, 2.294526],
            "files": ["file_a"],
            "user_id": str(user1_id)
        },
        {
            "title": f"Mont St Michel",
            "description": "Description B",
            "location": [48.636111, -1.511389],
            "files": ["file_b"],
            "user_id": str(user1_id)
        },
        {
            "title": f"Eiffel Tower",
            "description": "Description X",
            "location": [48.858296, 2.294526],
            "files": ["file_x"],
            "user_id": str(user2_id)
        },
        {
            "title": f"Mont Saint Michel",
            "description": "Description Y",
            "location": [48.636111, -1.511389],
            "files": ["file_y"],
            "user_id": str(user2_id)
        }
    ])
    
    friends_collection = db["friends"]
    friends_collection.insert_one({
        "user_id": str(user1_id),
        "friend_user_id": str(user2_id),
        "status": "accepted"
    })

if __name__ == "__main__":
    populate_data()
    print("Data inserted.")