add redis ping

chirps/target/providers/redis.py

@@ -2,6 +2,7 @@
 from logging import getLogger
 
 from django.db import models
+from redis import Redis
 from target.models import BaseTarget
 
 logger = getLogger(__name__)
@@ -28,5 +29,11 @@ def search(self, query: str, max_results: int) -> str:
 
     def test_connection(self) -> bool:
         """Ensure that the Redis target can be connected to."""
-        logger.error('RedisTarget search not implemented')
-        raise NotImplementedError
+        client = Redis(
+            host=self.host,
+            port=self.port,
+            db=self.database_name,
+            password=self.password,
+            username=self.username,
+        )
+        return client.ping()

chirps/target/templates/target/dashboard.html

@@ -1,58 +1,81 @@
-{% extends 'base.html' %}  
-{% block title %}Targets{% endblock %}  
-  
-{% block content %}  
+{% extends 'base.html' %}
+{% block title %}Targets{% endblock %}
+
+{% block content %}
 <div>
     <div class="d-flex">
         <div class="ml-0 mr-auto my-auto">
             <h1 class="text-success">Targets</h1>
         </div>
         <div class="ml-auto mr-0 my-auto">
             <div class="dropdown"></div>
-                <button class="btn btn-primary dropdown-toggle" data-bs-toggle="dropdown">Create</button>
-                <div class="dropdown-menu">
-                    {% for target in available_targets %}
-                        <a class="dropdown-item" href="{% url 'target_create' target.model.html_name %}">{{target.model.html_name}}</a>
-                    {% endfor %}
-                </div>
+            <button class="btn btn-primary dropdown-toggle" data-bs-toggle="dropdown">Create</button>
+            <div class="dropdown-menu">
+                {% for target in available_targets %}
+                <a class="dropdown-item"
+                    href="{% url 'target_create' target.model.html_name %}">{{target.model.html_name}}</a>
+                {% endfor %}
             </div>
         </div>
     </div>
-    <hr>
-
-    {% for target in user_targets %}  
-
-        <div class="card my-4">
-            <div class="d-flex card-header">
-                <img width=48 class="rounded float-end my-auto ml-0 mr-3" src="{{target.logo_url}}">
-                <h5 class="my-auto ml-0 mr-auto">{{ target.name }}</h5>
-            </div>  
-            <!-- <h5 class="card-header"><img width=48 class="rounded float-end" src="{{target.logo_url}}">{{ target.name }}</h5>   -->
-            <div class="card-body">  
-                <p class="card-text">  
-
-                {% if target.html_name == 'Mantium' %}  
-                    Application ID: {{ target.app_id }}  
-                {% elif target.html_name == 'Redis' %}  
-                    Host: {{ target.host }}<br/>  
-                    Port: {{ target.port }}<br/>  
-                    Database Name: {{ target.database_name }}<br/>  
-                    Username: {{ target.username }}  
-                {% endif %}  
-                {% if target.html_name == 'Pinecone' %}  
-                    API Key: {{ target.decrypted_api_key }}<br/>  
-                    Environment: {{ target.environment }}<br/>  
-                    Index Name: {{ target.index_name }}<br/>  
-                    Project Name: {{ target.project_name }}  
-                {% endif %}  
-                </p>
-                <div class="d-flex">
-                    <a href="{% url 'target_delete' target.id %}" class="btn btn-danger ml-auto mr-0">Delete</a>
-                    <a href="#" class="btn btn-primary ml-2 mr-0 disabled">Edit</a> 
-                </div> 
-            </div>  
-        </div>  
-
-    {% endfor %}  
-</div> 
-{% endblock %}  
+</div>
+<hr>
+
+{% for target in user_targets %}
+
+<div class="card my-4">
+    <div class="d-flex card-header">
+        <img width=48 class="rounded float-end my-auto ml-0 mr-3" src="{{target.logo_url}}">
+        <h5 class="my-auto ml-0 mr-auto">{{ target.name }}</h5>
+    </div>
+    <!-- <h5 class="card-header"><img width=48 class="rounded float-end" src="{{target.logo_url}}">{{ target.name }}</h5>   -->
+    <div class="card-body">
+        <p class="card-text">
+
+            {% if target.html_name == 'Mantium' %}
+            Application ID: {{ target.app_id }}
+            {% elif target.html_name == 'Redis' %}
+            Host: {{ target.host }}<br />
+            Port: {{ target.port }}<br />
+            Database Name: {{ target.database_name }}<br />
+            Username: {{ target.username }}
+            {% endif %}
+            {% if target.html_name == 'Pinecone' %}
+            API Key: {{ target.decrypted_api_key }}<br />
+            Environment: {{ target.environment }}<br />
+            Index Name: {{ target.index_name }}<br />
+            Project Name: {{ target.project_name }}
+            {% endif %}
+        </p>
+        <div class="d-flex">
+            <a href="{% url 'target_delete' target.id %}" class="btn btn-danger ml-auto mr-0">Delete</a>
+            <a href="#" class="btn btn-primary ml-2 mr-0 disabled">Edit</a>
+            {% if target.html_name == 'Redis' %}
+            <button data-target-id="{{ target.id }}" class="btn btn-info ml-2 mr-0 ping-btn">Ping</button>
+            {% endif %}
+        </div>
+    </div>
+</div>
+
+{% endfor %}
+</div>
+
+<script>
+    document.addEventListener('DOMContentLoaded', function () {
+        const pingButtons = document.querySelectorAll('.ping-btn');
+        pingButtons.forEach(btn => {
+            btn.addEventListener('click', async function () {
+                const targetId = btn.getAttribute('data-target-id');
+                const response = await fetch(`/target/ping/${targetId}/`, { headers: { 'X-Requested-With': 'XMLHttpRequest' } });
+                const result = await response.json();
+                if (result.success) {
+                    alert('Ping successful');
+                } else {
+                    alert('Ping failed: ' + (result.error || 'Unknown error'));
+                }
+            });
+        });
+    });  
+</script>
+
+{% endblock %}

chirps/target/urls.py

@@ -7,6 +7,7 @@
 urlpatterns = [
     path('', views.dashboard, name='target_dashboard'),
     path('create/<str:html_name>', views.create, name='target_create'),
+    path('ping/<int:target_id>/', views.ping, name='target_ping'),
     path('delete/<int:target_id>', views.delete, name='target_delete'),
     path('decrypted_keys/', views.decrypted_keys, name='decrypted_keys'),
 ]

chirps/target/views.py

@@ -3,9 +3,12 @@
 from django.http import JsonResponse
 from django.shortcuts import get_object_or_404, redirect, render
 
+from redis import exceptions
+
 from .forms import target_from_html_name, targets
 from .models import BaseTarget
 from .providers.pinecone import PineconeTarget
+from .providers.redis import RedisTarget
 
 
 def decrypted_keys(request):
@@ -64,6 +67,20 @@ def create(request, html_name):
     return render(request, 'target/create.html', {'form': form, 'target': target})
 
 
+@login_required  
+def ping(request, target_id):  
+    """Ping a RedisTarget database using the test_connection() function."""  
+    target = get_object_or_404(BaseTarget, pk=target_id)  
+    if isinstance(target, RedisTarget):  
+        try:
+            result = target.test_connection()
+            return JsonResponse({'success': result})
+        except exceptions.ConnectionError:
+            return JsonResponse({'success': False, 'error': 'Unable to connect to Redis'})
+    return JsonResponse({'success': False, 'error': 'Not a RedisTarget'})
+
+
+
 @login_required
 def delete(request, target_id):   # pylint: disable=unused-argument
     """Delete a target from the database."""