import vim
import abc


class Enviroment(abc.ABC):
    @property
    @abc.abstractmethod
    def plugin_root_dir(self):
        """Return absolute path to directory one level above of directory
        containing python scripts.
        """

    @property
    @abc.abstractmethod
    def python_style(self):
        """Returns string containing docstring style: google, numpy, etc.
        The style has to have corresponding template of name
        <style>-method.txt end <style>-class.txt in folder 'styles'.
        """

    @property
    @abc.abstractmethod
    def python_indent(self):
        """Returns indentation of python source (tab, 4 spaces, ...)"""

    @property
    @abc.abstractmethod
    def current_line_nr(self):
        """Returns number of the current line, indexed from 0."""

    @property
    @abc.abstractmethod
    def current_line(self):
        """Returns current line."""

    @abc.abstractmethod
    def append_after_line(self, line_nr, text):
        """Print to buffer

        Append `text` to buffer on line below `line_nr`.

        Args:
            line_nr: (int)
            text: (str)

        """

    @abc.abstractmethod
    def lines_following_cursor(self):
        """Generator that iterates lines in buffer, starting from current line."""

    # #TODO will this work???
    # @abc.abstractmethod
    # def lines_till_end(self):
    # """ Returns all lines from the current one until the end. """


class VimEnviroment(Enviroment):
    def __init__(self):
        pass

    def _get_var(self, name):
        return vim.eval(name)

    @property
    def plugin_root_dir(self):
        return self._get_var("s:plugin_root_dir")

    @property
    def python_style(self):
        if not int(vim.eval('exists("g:python_style")')):
            return "google"
        else:
            return self._get_var("g:python_style")

    @property
    def python_indent(self):
        if not int(vim.eval('exists("g:vpd_indent")')):
            return "    "
        else:
            return self._get_var("g:vpd_indent")

    @property
    def current_line_nr(self):
        return vim.current.window.cursor[0] - 1

    @property
    def current_line(self):
        return vim.current.line

    def append_after_line(self, line_nr, text):
        line_nr += 1
        for line in reversed(text.split("\n")):
            vim.current.buffer.append(line, line_nr)

    def lines_following_cursor(self):
        import vim

        buffer = vim.current.buffer
        cursor_row = vim.current.window.cursor[0] - 1
        current_row = cursor_row
        while True:
            if current_row > len(vim.current.buffer) - 1:
                raise StopIteration("Buffer is out of lines.")
            yield current_row, buffer[current_row]
            current_row += 1