comparison mercurial/commandserver.py @ 22989:dc8803ce3dfe

cmdserver: switch service objects by mode

server class will be changed to accept fin/fout pair instead of mode string
so that it can interact with socket files.

22988:32b77aba2772

class server(object):
    """
    Listens for commands on stdin, runs them and writes the output on a channel
    based stream to stdout.
    """
    def __init__(self, ui, repo, mode):
        self.cwd = os.getcwd()

        logpath = ui.config("cmdserver", "log", None)
        if logpath:
            global logfile

            self.repoui = repo.ui
        else:
            self.ui = ui
            self.repo = self.repoui = None

        if mode == 'pipe':
            self.cerr = channeledoutput(sys.stdout, 'e')
            self.cout = channeledoutput(sys.stdout, 'o')
            self.cin = channeledinput(sys.stdin, sys.stdout, 'I')
            self.cresult = channeledoutput(sys.stdout, 'r')

            self.client = sys.stdin
        else:
            raise util.Abort(_('unknown mode %s') % mode)

    def _read(self, size):
        if not size:
            return ''

        return 0

class pipeservice(object):
    def __init__(self, ui, repo, opts):
        self.server = server(ui, repo, opts['cmdserver'])

    def init(self):
        pass

    def run(self):
        return self.server.serve()

22989:dc8803ce3dfe

class server(object):
    """
    Listens for commands on stdin, runs them and writes the output on a channel
    based stream to stdout.
    """
    def __init__(self, ui, repo):
        self.cwd = os.getcwd()

        logpath = ui.config("cmdserver", "log", None)
        if logpath:
            global logfile

            self.repoui = repo.ui
        else:
            self.ui = ui
            self.repo = self.repoui = None

        self.cerr = channeledoutput(sys.stdout, 'e')
        self.cout = channeledoutput(sys.stdout, 'o')
        self.cin = channeledinput(sys.stdin, sys.stdout, 'I')
        self.cresult = channeledoutput(sys.stdout, 'r')

        self.client = sys.stdin

    def _read(self, size):
        if not size:
            return ''

        return 0

class pipeservice(object):
    def __init__(self, ui, repo, opts):
        self.server = server(ui, repo)

    def init(self):
        pass

    def run(self):
        return self.server.serve()

_servicemap = {
    'pipe': pipeservice,
    }

def createservice(ui, repo, opts):
    mode = opts['cmdserver']
    try:
        return _servicemap[mode](ui, repo, opts)
    except KeyError:
        raise util.Abort(_('unknown mode %s') % mode)