# Copyright (c) 2012 Roberto Alsina y otros.

# Permission is hereby granted, free of charge, to any
# person obtaining a copy of this software and associated
# documentation files (the "Software"), to deal in the
# Software without restriction, including without limitation
# the rights to use, copy, modify, merge, publish,
# distribute, sublicense, and/or sell copies of the
# Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice
# shall be included in all copies or substantial portions of
# the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY
# KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
# WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
# PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS
# OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
# OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
# OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
# SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

from __future__ import unicode_literals, print_function

import os

from pygments import highlight
from pygments.lexers import get_lexer_for_filename, TextLexer
from pygments.formatters import HtmlFormatter

from nikola.plugin_categories import Task
from nikola import utils


class Listings(Task):
    """Render pretty listings."""

    name = "render_listings"

    def gen_tasks(self):
        """Render pretty code listings."""
        kw = {
            "default_lang": self.site.config["DEFAULT_LANG"],
            "listings_folder": self.site.config["LISTINGS_FOLDER"],
            "output_folder": self.site.config["OUTPUT_FOLDER"],
            "index_file": self.site.config["INDEX_FILE"],
        }

        # Things to ignore in listings
        ignored_extensions = (".pyc",)

        def render_listing(in_name, out_name, folders=[], files=[]):
            if in_name:
                with open(in_name, 'r') as fd:
                    try:
                        lexer = get_lexer_for_filename(in_name)
                    except:
                        lexer = TextLexer()
                    code = highlight(fd.read(), lexer,
                                     HtmlFormatter(cssclass='code',
                                                   linenos="table", nowrap=False,
                                                   lineanchors=utils.slugify(f),
                                                   anchorlinenos=True))
                title = os.path.basename(in_name)
            else:
                code = ''
                title = ''
            crumbs = utils.get_crumbs(os.path.relpath(out_name,
                                                      kw['output_folder']),
                                      is_file=True)
            context = {
                'code': code,
                'title': title,
                'crumbs': crumbs,
                'lang': kw['default_lang'],
                'folders': folders,
                'files': files,
                'description': title,
            }
            self.site.render_template('listing.tmpl', out_name,
                                      context)
        flag = True
        template_deps = self.site.template_system.template_deps('listing.tmpl')
        for root, dirs, files in os.walk(kw['listings_folder']):
            flag = False
            # Render all files
            out_name = os.path.join(
                kw['output_folder'],
                root, kw['index_file']
            )
            yield {
                'basename': self.name,
                'name': out_name,
                'file_dep': template_deps,
                'targets': [out_name],
                'actions': [(render_listing, [None, out_name, dirs, files])],
                # This is necessary to reflect changes in blog title,
                # sidebar links, etc.
                'uptodate': [utils.config_changed(
                    self.site.config['GLOBAL_CONTEXT'])],
                'clean': True,
            }
            for f in files:
                ext = os.path.splitext(f)[-1]
                if ext in ignored_extensions:
                    continue
                in_name = os.path.join(root, f)
                out_name = os.path.join(
                    kw['output_folder'],
                    root,
                    f) + '.html'
                yield {
                    'basename': self.name,
                    'name': out_name,
                    'file_dep': template_deps + [in_name],
                    'targets': [out_name],
                    'actions': [(render_listing, [in_name, out_name])],
                    # This is necessary to reflect changes in blog title,
                    # sidebar links, etc.
                    'uptodate': [utils.config_changed(
                        self.site.config['GLOBAL_CONTEXT'])],
                    'clean': True,
                }
        if flag:
            yield {
                'basename': self.name,
                'actions': [],
            }

[((86, 33, 86, 63), 'os.walk', 'os.walk', ({(86, 41, 86, 62): "kw['listings_folder']"}, {}), "(kw['listings_folder'])", False, 'import os\n'), ((89, 23, 92, 13), 'os.path.join', 'os.path.join', ({(90, 16, 90, 35): "kw['output_folder']", (91, 16, 91, 20): 'root', (91, 22, 91, 38): "kw['index_file']"}, {}), "(kw['output_folder'], root, kw['index_file'])", False, 'import os\n'), ((66, 24, 66, 49), 'os.path.basename', 'os.path.basename', ({(66, 41, 66, 48): 'in_name'}, {}), '(in_name)', False, 'import os\n'), ((70, 38, 71, 74), 'os.path.relpath', 'os.path.relpath', ({(70, 54, 70, 62): 'out_name', (71, 54, 71, 73): "kw['output_folder']"}, {}), "(out_name, kw['output_folder'])", False, 'import os\n'), ((109, 26, 109, 47), 'os.path.join', 'os.path.join', ({(109, 39, 109, 43): 'root', (109, 45, 109, 46): 'f'}, {}), '(root, f)', False, 'import os\n'), ((106, 22, 106, 41), 'os.path.splitext', 'os.path.splitext', ({(106, 39, 106, 40): 'f'}, {}), '(f)', False, 'import os\n'), ((110, 27, 113, 22), 'os.path.join', 'os.path.join', ({(111, 20, 111, 39): "kw['output_folder']", (112, 20, 112, 24): 'root', (113, 20, 113, 21): 'f'}, {}), "(kw['output_folder'], root, f)", False, 'import os\n'), ((58, 32, 58, 63), 'pygments.lexers.get_lexer_for_filename', 'get_lexer_for_filename', ({(58, 55, 58, 62): 'in_name'}, {}), '(in_name)', False, 'from pygments.lexers import get_lexer_for_filename, TextLexer\n'), ((101, 29, 102, 55), 'nikola.utils.config_changed', 'utils.config_changed', ({(102, 20, 102, 54): "self.site.config['GLOBAL_CONTEXT']"}, {}), "(self.site.config['GLOBAL_CONTEXT'])", False, 'from nikola import utils\n'), ((60, 32, 60, 43), 'pygments.lexers.TextLexer', 'TextLexer', ({}, {}), '()', False, 'from pygments.lexers import get_lexer_for_filename, TextLexer\n'), ((122, 33, 123, 59), 'nikola.utils.config_changed', 'utils.config_changed', ({(123, 24, 123, 58): "self.site.config['GLOBAL_CONTEXT']"}, {}), "(self.site.config['GLOBAL_CONTEXT'])", False, 'from nikola import utils\n'), ((64, 63, 64, 79), 'nikola.utils.slugify', 'utils.slugify', ({(64, 77, 64, 78): 'f'}, {}), '(f)', False, 'from nikola import utils\n')]