import time
import os
import config
from celery import shared_task
from app import db
from misc_utils import *
from models import *
from smtplib import SMTP


@shared_task(ignore_result=False)
def send_notification(notification_id, users, level):
    notification = db.session.get(Notification, notification_id)

    for user in users:
        db.session.add(UserNotification(db.session.get(User, user), notification, level))
    db.session.commit()

    return 0  # notification sent successfully


@shared_task(ignore_result=False)
def merge_heads(head_route, head_branch, base_route, base_branch):
    server_repo_location = os.path.join(config.REPOS_PATH, base_route.lstrip("/"))
    if not os.path.isdir(server_repo_location):
        raise FileNotFoundError(f"Repo {server_repo_location} not found, cannot merge.")

    git_command(server_repo_location, b"", "checkout", "-f", f"heads/{base_branch}")
    out, err = git_command(server_repo_location, b"", "merge", f"heads/{head_branch}", return_err=True)

    return out, err