import base64
import httplib2
import mimetypes
import os

from apiclient import discovery

from email.mime.audio import MIMEAudio
from email.mime.base import MIMEBase
from email.mime.image import MIMEImage
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText

from platypush.message.response import Response
from platypush.plugins.google import GooglePlugin


class GoogleMailPlugin(GooglePlugin):
    scopes = ['https://www.googleapis.com/auth/gmail.modify']

    def __init__(self, *args, **kwargs):
        super().__init__(scopes=self.scopes, *args, **kwargs)


    def compose(self, sender, to, subject, body, file=None):
        message = MIMEMultipart() if file else MIMEText(body)
        message['to'] = to
        message['from'] = sender
        message['subject'] = subject

        if file:
            msg = MIMEText(body)
            message.attach(msg)
            content_type, encoding = mimetypes.guess_type(file)

            if content_type is None or encoding is not None:
                content_type = 'application/octet-stream'
            main_type, sub_type = content_type.split('/', 1)

            with open(file, 'rb') as fp:
                if main_type == 'text':
                    msg = mimetypes.MIMEText(fp.read(), _subtype=sub_type)
                elif main_type == 'image':
                    msg = MIMEImage(fp.read(), _subtype=sub_type)
                elif main_type == 'audio':
                    msg = MIMEAudio(fp.read(), _subtype=sub_type)
                else:
                    msg = MIMEBase(main_type, sub_type)
                    msg.set_payload(fp.read())

            filename = os.path.basename(file)
            msg.add_header('Content-Disposition', 'attachment', filename=filename)
            message.attach(msg)

        service = self._get_service()
        body = { 'raw': base64.urlsafe_b64encode(message.as_bytes()).decode() }
        message = (service.users().messages().send(
            userId='me', body=body).execute())

        return Response(output=message)


    def get_labels(self):
        service = self._get_service()
        results = service.users().labels().list(userId='me').execute()
        labels = results.get('labels', [])
        return Response(output=labels)


    def _get_service(self):
        scope = self.scopes[0]
        credentials = self.credentials[scope]
        http = credentials.authorize(httplib2.Http())
        return discovery.build('gmail', 'v1', http=http, cache_discovery=False)


# vim:sw=4:ts=4:et: