import json

from pynput.mouse import Controller, Button


class ClickMessage:
    def __init__(self, x, y, button):
        self.function = 0
        self.x = x
        self.y = y
        if button == Button.left:
            self.button = "left"
        elif button == Button.right:
            self.button = "right"

    def getMessage(self):
        return json.dumps(self.__dict__)


class ScrollMessage:
    def __init__(self, x, y, dx, dy):
        self.function = 1
        self.x = x
        self.y = y
        self.dx = dx
        self.dy = dy

    def getMessage(self):
        return json.dumps(self.__dict__)