From 7adb17452d8c845f46858b4cebe8198988edfdbc Mon Sep 17 00:00:00 2001
From: Thomas Pointhuber <thomas.pointhuber@gmx.at>
Date: Sat, 20 Dec 2014 23:33:03 +0100
Subject: [enh] add result_templates/code.html

---
 searx/engines/searchcode_code.py | 32 ++++++++++++++++++++------------
 1 file changed, 20 insertions(+), 12 deletions(-)

(limited to 'searx/engines/searchcode_code.py')

diff --git a/searx/engines/searchcode_code.py b/searx/engines/searchcode_code.py
index 2ba0e52f1..ac9647fcd 100644
--- a/searx/engines/searchcode_code.py
+++ b/searx/engines/searchcode_code.py
@@ -21,6 +21,18 @@ paging = True
 url = 'https://searchcode.com/'
 search_url = url+'api/codesearch_I/?{query}&p={pageno}'
 
+code_endings = {'c': 'c',
+                'css': 'css',
+                'cpp': 'cpp',
+                'c++': 'cpp',
+                'h': 'c',
+                'html': 'html',
+                'hpp': 'cpp',
+                'js': 'js',
+                'lua': 'lua',
+                'php': 'php',
+                'py': 'python'}
+
 
 # do search-request
 def request(query, params):
@@ -40,26 +52,22 @@ def response(resp):
     for result in search_results['results']:
         href = result['url']
         title = "" + result['name'] + " - " + result['filename']
-        content = result['repo'] + "<br />"
+        repo = result['repo']
         
         lines = dict()
         for line, code in result['lines'].items():
             lines[int(line)] = code
 
-        content = content + '<pre class="code-formatter"><table class="code">'
-        for line, code in sorted(lines.items()):
-            content = content + '<tr><td class="line-number" style="padding-right:5px;">' 
-            content = content + str(line) + '</td><td class="code-snippet">' 
-            # Replace every two spaces with ' &nbps;' to keep formatting while allowing the browser to break the line if necessary
-            content = content + cgi.escape(code).replace('\t', '    ').replace('  ', '&nbsp; ').replace('  ', ' &nbsp;') 
-            content = content + "</td></tr>"
-            
-        content = content + "</table></pre>"
-        
+        code_language = code_endings.get(result['filename'].split('.')[-1].lower(), None)
+
         # append result
         results.append({'url': href,
                         'title': title,
-                        'content': content})
+                        'content': '',
+                        'repository': repo,
+                        'codelines': sorted(lines.items()),
+                        'code_language': code_language,
+                        'template': 'code.html'})
 
     # return results
     return results
-- 
cgit v1.2.3