import requests
import os
import msal
from office365.sharepoint.client_context import ClientContext
from office365.runtime.auth.user_credential import UserCredential
from office365.runtime.auth.client_credential import ClientCredential



class SharepointManager:
    def __init__(self, site, userSharepoint, passwordSharepoint):#,client_id,tenant_id,client_secret):
        self.site = site
        self.userSharepoint = userSharepoint
        self.passwordSharepoint = passwordSharepoint
        #self.client_id = client_id
        #self.tenant_id = tenant_id
        #self.client_secret = client_secret

        #print(self.client_secret)


    def uploadFile2(self, task, sharepoint_path):
        self._connect2()
        file_name = os.path.basename(task.docsend_Path)

        with open(task.docsend_Path, "rb") as file_content:
            folder = self.ctx.web.get_folder_by_server_relative_url(sharepoint_path)
            upload = folder.upload_file(file_name, file_content.read())  # solo prepara la query
            self.ctx.execute_query()  # ejecuta en el contexto
            print(f"✅ Archivo {file_name} subido a {upload.serverRelativeUrl}")

        self._disconnect()
        
    def _connect2(self):
        #"""Conexión a SharePoint con access_token"""
        #token = self._get_token()
        #self.ctx = ClientContext(self.site).with_access_token(token)
        #self.ctx.load(self.ctx.web)
        #self.ctx.execute_query()

        self.ctx = ClientContext(self.site).with_credentials(ClientCredential(self.client_id, self.client_secret))
        self.ctx.load(self.ctx.web)
        self.ctx.execute_query()


    def _connect(self):
        # 👇 Nada de AuthenticationContext aquí
        self.ctx = ClientContext(self.site).with_credentials(
            UserCredential(self.userSharepoint, self.passwordSharepoint)
        )
        self.ctx.load(self.ctx.web)
        self.ctx.execute_query()

    def _disconnect(self):
        self.ctx = None

    def uploadFile(self, task, sharepoint_path):
        self._connect()
        file_name = os.path.basename(task.docsend_Path)
        with open(task.docsend_Path, "rb") as file_content:
            response = (
                self.ctx.web.get_folder_by_server_relative_url(sharepoint_path)
                .upload_file(file_name, file_content.read())
                .execute_query()
            )
            print(f"Archivo {file_name} subido a {response.serverRelativeUrl}")
        self._disconnect()


    def uploadFile2(self, task, sharepoint_path):
        """Subida con MSAL (App Registration)"""
        self._connect2()
        file_name = os.path.basename(task.docsend_Path)

        with open(task.docsend_Path, "rb") as file_content:
            folder = self.ctx.web.get_folder_by_server_relative_url(sharepoint_path)
            upload = folder.upload_file(file_name, file_content.read())
            self.ctx.execute_query()
            print(f"✅ Archivo {file_name} subido a {upload.serverRelativeUrl}")

        self._disconnect()

    def uploadFileTemp(self, task, data):
        self._connect()
        file_name = os.path.basename(task.docsend_Path)
        with open(task.docsend_Path, "rb") as file_content:
            response = (
                self.ctx.web.get_folder_by_server_relative_url(data[6])
                .upload_file(file_name, file_content.read())
                .execute_query()
            )
            print(f"Archivo {file_name} subido a {response.serverRelativeUrl}")
        self._disconnect()

    def dropFilesFolder(self, data):
        self._connect()
        folder = (
            self.ctx.web.get_folder_by_server_relative_url(data[6])
            .expand(["Files"])
            .get()
            .execute_query()
        )
        for file in folder.files:
            print(f"Eliminando archivo: {file.name}")
            file.delete_object()
        self.ctx.execute_query()
        self._disconnect()