diff --git a/.gitignore b/.gitignore index f0523d7..88f5edc 100644 --- a/.gitignore +++ b/.gitignore @@ -149,7 +149,7 @@ environment.json shell-profile -hooks.db +*.db requirements-pwrhook.txt diff --git a/powerindex/app.py b/powerindex/app.py index ad38939..2c76561 100644 --- a/powerindex/app.py +++ b/powerindex/app.py @@ -3,6 +3,7 @@ import string import re import subprocess +import sqlite3 from flask import Flask, render_template, request, jsonify from pygments import highlight from pygments.lexers import BashLexer, JsonLexer @@ -35,6 +36,44 @@ def update_curl_flags(): # Run the update immediately update_curl_flags() +# Initialize the database +def init_db(): + db_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'the-power.db') + conn = sqlite3.connect(db_path) + cursor = conn.cursor() + + # Create FTS5 table + cursor.execute(''' + CREATE VIRTUAL TABLE IF NOT EXISTS scripts_fts USING fts5( + script, + body + ) + ''') + + # Clear existing entries + cursor.execute('DELETE FROM scripts_fts') + + # Load all scripts into the database + parent_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) + shell_scripts = glob.glob(os.path.join(parent_dir, '*.sh')) + + for script_path in shell_scripts: + script_name = os.path.basename(script_path) + try: + with open(script_path, 'r') as file: + content = file.read() + cursor.execute('INSERT INTO scripts_fts (script, body) VALUES (?, ?)', + (script_name, content)) + except Exception as e: + print(f"Error loading script {script_name}: {str(e)}") + + conn.commit() + conn.close() + print(f"Database initialized at {db_path}") + +# Add database initialization before Flask app creation +init_db() + app = Flask(__name__) # Add this at the top level for use in the template @@ -232,5 +271,29 @@ def highlight_search(text, search): pattern = re.compile(f'({re.escape(search)})', re.IGNORECASE) return pattern.sub(r'\1', text) +@app.route('/search', methods=['GET']) +def search(): + query = request.args.get('q', '') + results = [] + + if query: + # Use the same database path as init_db() + db_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'the-power.db') + conn = sqlite3.connect(db_path) + cursor = conn.cursor() + + # Use FTS5 match to find results and highlight matching text + cursor.execute(''' + SELECT script, snippet(scripts_fts, 1, '', '', '...', 50) + FROM scripts_fts + WHERE body MATCH ? + ORDER BY rank + ''', (query,)) + + results = cursor.fetchall() + conn.close() + + return render_template('search.html', query=query, results=results) + if __name__ == '__main__': app.run(debug=False , host='localhost', port=8001) diff --git a/powerindex/templates/base.html b/powerindex/templates/base.html new file mode 100644 index 0000000..6e5b7da --- /dev/null +++ b/powerindex/templates/base.html @@ -0,0 +1,57 @@ + + + + + + {% block title %}Power Index{% endblock %} + + + + +
+
+
+ +
+ {% include 'navbar.html' %} +
+ +
+ {% block content %} + {% endblock %} +
+
+ + {% block scripts %} + + {% endblock %} + + \ No newline at end of file diff --git a/powerindex/templates/index.html b/powerindex/templates/index.html index e4c8ed6..bba1aa5 100644 --- a/powerindex/templates/index.html +++ b/powerindex/templates/index.html @@ -1,135 +1,125 @@ - - - - - - Power Index - - - - -
-
- -

The Power Viewer

-
- -
- - -
-
- - {% if script_content %} -
-

- - - {{ selected_script }} -

-
-
-
-

Original Script

-
-
- {{ highlighted_content | safe }} -
+{% extends "base.html" %} + +{% block title %}The Power Viewer{% endblock %} + +{% block header %}The Power Viewer{% endblock %} + +{% block content %} +
+ + +
+
+ + {% if script_content %} +
+

+ + + {{ selected_script }} +

+
+
+
+

Original Script

+
+
+ {{ highlighted_content | safe }}
-
-
-
-

Rendered Script

- -
-
- {{ rendered_content | safe }} -
+
+
+
+
+

Rendered Script

+ +
+
+ {{ rendered_content | safe }}
-
-
-
-
-

Standard Output

- -
- +
+
+
+
+
+

Standard Output

+
- 

+                        
                     

-                    

-                        
Standard Error

-                        
-                        

-                            

-                                
{{ stderr }}

-                            

+                    

+                

+                

+                    
Standard Error

+                    
+                    

+                        

+                            
{{ stderr }}

                         

                     

-                    

-                        
Headers

-                        
-                        
-                        

-                            

-                                
{{ headers }}

-                            

+                

+                

+                    
Headers

+                    
+                    
+                    

+                        

+                            
{{ headers }}

                         

                     

                 

-                
-                
Configuration

-                
-                
-                    
-                        
-                            
-                            
-                        
-                    
-                    
-                        {% for key, value in config.items() %}
-                        
-                            
-                            
-                        
-                        {% endfor %}
-                    
-                
KeyValue
{{ key }}{{ value }}

             

-            {% endif %}
+            
+            
Configuration

+            
+            
+                
+                    
+                        
+                        
+                    
+                
+                
+                    {% for key, value in config.items() %}
+                    
+                        
+                        
+                    
+                    {% endfor %}
+                
+            
KeyValue
{{ key }}{{ value }}

         

+        {% endif %}
     

+{% endblock %}
 
+{% block scripts %}
+    {{ super() }}
     
-
- 
+{% endblock %} 
diff --git a/powerindex/templates/navbar.html b/powerindex/templates/navbar.html
new file mode 100644
index 0000000..484239d
--- /dev/null
+++ b/powerindex/templates/navbar.html
@@ -0,0 +1,6 @@
+
diff --git a/powerindex/templates/search.html b/powerindex/templates/search.html
new file mode 100644
index 0000000..f28ced4
--- /dev/null
+++ b/powerindex/templates/search.html
@@ -0,0 +1,33 @@
+{% extends "base.html" %}
+
+{% block title %}Power Index - Full Text Search{% endblock %}
+
+{% block header %}Full Text Search{% endblock %}
+
+{% block content %}
+

+    

+        
+        
+    

+

+
+{% if query %}
+    

+        
Results for "{{ query }}"

+        {% if results %}
+            {% for script, snippet in results %}
+            

+                
{{ script }}

+                
{{ snippet | safe }}

+                

+                    View Full Script
+                

+            

+            {% endfor %}
+        {% else %}
+            
No results found.

+        {% endif %}
+    

+{% endif %}
+{% endblock %} 
\ No newline at end of file