# coding=utf-8
# Copyright 2015 Pants project contributors (see CONTRIBUTORS.md).
# Licensed under the Apache License, Version 2.0 (see LICENSE).

from __future__ import absolute_import, division, print_function, unicode_literals

import pycodestyle

from pants.contrib.python.checks.checker.common import CheckstylePlugin, Nit


class PyCodeStyleError(Nit):
    def __init__(self, python_file, code, line_number, text):
        line_range = python_file.line_range(line_number)
        lines = python_file.lines[line_range]
        super(PyCodeStyleError, self).__init__(
            code, Nit.ERROR, python_file, text, line_range, lines
        )


def reporter(python_file):
    class PantsReporter(pycodestyle.BaseReport):
        errors = []

        def error(self, line_number, offset, text, check):
            code = super(PantsReporter, self).error(line_number, offset, text, check)
            if code:
                self.errors.append(PyCodeStyleError(python_file, code, line_number, text[5:]))
            return code

    return PantsReporter


class PyCodeStyleChecker(CheckstylePlugin):
    """Enforce PEP8 checks from the pycodestyle tool."""

    @classmethod
    def name(cls):
        return "pycodestyle"

    def __init__(self, *args, **kwargs):
        super(PyCodeStyleChecker, self).__init__(*args, **kwargs)
        self.STYLE_GUIDE = pycodestyle.StyleGuide(
            max_line_length=self.options.max_length,
            verbose=False,
            reporter=reporter(self.python_file),
            ignore=self.options.ignore,
        )

    def nits(self):
        report = self.STYLE_GUIDE.check_files([self.python_file.path])
        return report.errors