From b8ca7286f2e2767a56e0b21f24bd3f25d2abdbbf Mon Sep 17 00:00:00 2001
From: William Colmenares <colmenares.william@gmail.com>
Date: Thu, 9 May 2019 01:34:18 -0400
Subject: [PATCH] Add view to check if the vm belongs to a user

---
 hosting/urls.py  |  3 ++-
 hosting/views.py | 20 ++++++++++++++++++++
 2 files changed, 22 insertions(+), 1 deletion(-)

diff --git a/hosting/urls.py b/hosting/urls.py
index a3579f06..1d0033ef 100644
--- a/hosting/urls.py
+++ b/hosting/urls.py
@@ -9,13 +9,14 @@ from .views import (
     HostingPricingView, CreateVirtualMachinesView, HostingBillListView,
     HostingBillDetailView, SSHKeyDeleteView, SSHKeyCreateView, SSHKeyListView,
     SSHKeyChoiceView, DashboardView, SettingsView, ResendActivationEmailView,
-    InvoiceListView, InvoiceDetailView
+    InvoiceListView, InvoiceDetailView, CheckUserVM
 )
 
 
 urlpatterns = [
     url(r'index/?$', IndexView.as_view(), name='index'),
     url(r'django/?$', DjangoHostingView.as_view(), name='djangohosting'),
+    url(r'checkvm/?$', CheckUserVM.as_view(), name='chech_vm'),
     url(r'dashboard/?$', DashboardView.as_view(), name='dashboard'),
     url(r'nodejs/?$', NodeJSHostingView.as_view(), name='nodejshosting'),
     url(r'rails/?$', RailsHostingView.as_view(), name='railshosting'),
diff --git a/hosting/views.py b/hosting/views.py
index 92dd5aa8..4c0d8b41 100644
--- a/hosting/views.py
+++ b/hosting/views.py
@@ -26,6 +26,8 @@ from django.views.generic import (
     View, CreateView, FormView, ListView, DetailView, DeleteView,
     TemplateView, UpdateView
 )
+from rest_framework.views import APIView
+from rest_framework.response import Response
 from guardian.mixins import PermissionRequiredMixin
 from oca.pool import WrongIdError
 from stored_messages.api import mark_read
@@ -1755,3 +1757,21 @@ def forbidden_view(request, exception=None, reason=''):
                 'again.')
     messages.add_message(request, messages.ERROR, err_msg)
     return HttpResponseRedirect(request.get_full_path())
+
+
+class CheckUserVM(APIView):
+
+    def get(self, request):
+        try:
+            email = request.data['email']
+            ip = request.data['ip']
+            uservms = VMDetail.objects.filter(user__email=email)
+            if len(uservms) > 0:
+                for i in range(len(uservms)):
+                    if uservms[i].ipv4 == ip or uservms[i].ipv6 == ip:
+                        return Response('success', 200)
+                return Response('No VM found matching the ip address provided', 403)
+            else:
+                return Response('No VM found with the given email address', 403)
+        except KeyError:
+            return Response('Not enough data provided', 400)