1 files changed, 21 insertions, 5 deletions

diff --git a/ui-tree.c b/ui-tree.c
index 553dbaa..c608754 100644
--- a/ui-tree.c
+++ b/ui-tree.c
@@ -17,3 +17,3 @@ int header = 0;
 
-static void print_text_buffer(char *buf, unsigned long size)
+static void print_text_buffer(const char *name, char *buf, unsigned long size)
 {
@@ -24,2 +24,12 @@ static void print_text_buffer(char *buf, unsigned long size)
         html("<table summary='blob content' class='blob'>\n");
+        if (ctx.repo->source_filter) {
+                html("<tr><td class='lines'><pre><code>");
+                ctx.repo->source_filter->argv[1] = xstrdup(name);
+                cgit_open_filter(ctx.repo->source_filter);
+                write(STDOUT_FILENO, buf, size);
+                cgit_close_filter(ctx.repo->source_filter);
+                html("</code></pre></td></tr></table>\n");
+                return;
+        }
+
         html("<tr><td class='linenumbers'><pre>");
@@ -67,3 +77,3 @@ static void print_binary_buffer(char *buf, unsigned long size)
 
-static void print_object(const unsigned char *sha1, char *path)
+static void print_object(const unsigned char *sha1, char *path, const char *basename)
 {
@@ -95,3 +105,3 @@ static void print_object(const unsigned char *sha1, char *path)
         else
-                print_text_buffer(buf, size);
+                print_text_buffer(basename, buf, size);
 }
@@ -105,2 +115,3 @@ static int ls_item(const unsigned char *sha1, const char *base, int baselen,
         char *fullpath;
+        char *class;
         enum object_type type;
@@ -137,3 +148,8 @@ static int ls_item(const unsigned char *sha1, const char *base, int baselen,
         } else {
-                cgit_tree_link(name, NULL, "ls-blob", ctx.qry.head,
+                class = strrchr(name, '.');
+                if (class != NULL) {
+                        class = fmt("ls-blob %s", class + 1);
+                } else
+                        class = "ls-blob";
+                cgit_tree_link(name, NULL, class, ctx.qry.head,
                                curr_rev, fullpath);
@@ -215,3 +231,3 @@ static int walk_tree(const unsigned char *sha1, const char *base, int baselen,
                 } else {
-                        print_object(sha1, buffer);
+                        print_object(sha1, buffer, pathname);
                         return 0;