#!/usr/bin/python

# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.

import lldb


class FBInputHandler:
    def __init__(self, debugger, callback):
        self.debugger = debugger
        self.callback = callback
        self.inputReader = lldb.SBInputReader()
        self.inputReader.Initialize(
            debugger,
            self.handleInput,
            lldb.eInputReaderGranularityLine,
            None,
            None,  # prompt
            True,  # echo
        )

    def isValid(self):
        return not self.inputReader.IsDone()

    def start(self):
        self.debugger.PushInputReader(self.inputReader)

    def stop(self):
        self.inputReader.SetIsDone(True)

    def handleInput(self, inputReader, notification, bytes):
        if notification == lldb.eInputReaderGotToken:
            self.callback(bytes)
        elif notification == lldb.eInputReaderInterrupt:
            self.stop()

        return len(bytes)