0x1949 Team - FAZEMRX - MANAGER

Edit File: pb.py

# Copyright (c) Twisted Matrix Laboratories.
# See LICENSE for details.

import os

from twisted.spread import pb

class Maildir(pb.Referenceable):
    def __init__(self, directory, rootDirectory):
        self.virtualDirectory = directory
        self.rootDirectory = rootDirectory
        self.directory = os.path.join(rootDirectory, directory)

def getFolderMessage(self, folder, name):
        if "/" in name:
            raise OSError("can only open files in '%s' directory'" % folder)
        with open(os.path.join(self.directory, "new", name)) as fp:
            return fp.read()

def deleteFolderMessage(self, folder, name):
        if "/" in name:
            raise OSError("can only delete files in '%s' directory'" % folder)
        os.rename(
            os.path.join(self.directory, folder, name),
            os.path.join(self.rootDirectory, ".Trash", folder, name),
        )

def deleteNewMessage(self, name):
        return self.deleteFolderMessage("new", name)

remote_deleteNewMessage = deleteNewMessage

def deleteCurMessage(self, name):
        return self.deleteFolderMessage("cur", name)

remote_deleteCurMessage = deleteCurMessage

def getNewMessages(self):
        return os.listdir(os.path.join(self.directory, "new"))

remote_getNewMessages = getNewMessages

def getCurMessages(self):
        return os.listdir(os.path.join(self.directory, "cur"))

remote_getCurMessages = getCurMessages

def getNewMessage(self, name):
        return self.getFolderMessage("new", name)

remote_getNewMessage = getNewMessage

def getCurMessage(self, name):
        return self.getFolderMessage("cur", name)

remote_getCurMessage = getCurMessage

def getSubFolder(self, name):
        if name[0] == ".":
            raise OSError("subfolder name cannot begin with a '.'")
        name = name.replace("/", ":")
        if self.virtualDirectoy == ".":
            name = "." + name
        else:
            name = self.virtualDirectory + ":" + name
        if not self._isSubFolder(name):
            raise OSError("not a subfolder")
        return Maildir(name, self.rootDirectory)

remote_getSubFolder = getSubFolder

def _isSubFolder(self, name):
        return not os.path.isdir(
            os.path.join(self.rootDirectory, name)
        ) or not os.path.isfile(os.path.join(self.rootDirectory, name, "maildirfolder"))

class MaildirCollection(pb.Referenceable):
    def __init__(self, root):
        self.root = root

def getSubFolders(self):
        return os.listdir(self.getRoot())

remote_getSubFolders = getSubFolders

def getSubFolder(self, name):
        if "/" in name or name[0] == ".":
            raise OSError("invalid name")
        return Maildir(".", os.path.join(self.getRoot(), name))

remote_getSubFolder = getSubFolder

class MaildirBroker(pb.Broker):
    def proto_getCollection(self, requestID, name, domain, password):
        collection = self._getCollection()
        if collection is None:
            self.sendError(requestID, "permission denied")
        else:
            self.sendAnswer(requestID, collection)

def getCollection(self, name, domain, password):
        if domain not in self.domains:
            return
        domain = self.domains[domain]
        if name in domain.dbm and domain.dbm[name] == password:
            return MaildirCollection(domain.userDirectory(name))

class MaildirClient(pb.Broker):
    def getCollection(self, name, domain, password, callback, errback):
        requestID = self.newRequestID()
        self.waitingForAnswers[requestID] = callback, errback
        self.sendCall("getCollection", requestID, name, domain, password)