dynamicweb/hosting/views.py

import logging
import uuid
from datetime import datetime
from time import sleep

import stripe
from django import forms
from django.conf import settings
from django.contrib import messages
from django.contrib.auth.mixins import LoginRequiredMixin
from django.contrib.auth.tokens import default_token_generator
from django.core.exceptions import ValidationError
from django.core.files.base import ContentFile
from django.core.paginator import Paginator, EmptyPage, PageNotAnInteger
from django.core.urlresolvers import reverse_lazy, reverse
from django.db.models import Q
from django.http import (
    Http404, HttpResponseRedirect, HttpResponse, JsonResponse
)
from django.shortcuts import redirect, render
from django.utils.decorators import method_decorator
from django.utils.html import escape
from django.utils.http import urlsafe_base64_decode
from django.utils.safestring import mark_safe
from django.utils.translation import get_language, ugettext_lazy as _
from django.utils.translation import ugettext
from django.views.decorators.cache import never_cache
from django.views.generic import (
    View, CreateView, FormView, ListView, DetailView, DeleteView,
    TemplateView, UpdateView
)
from guardian.mixins import PermissionRequiredMixin
from oca.pool import WrongIdError
from rest_framework.renderers import JSONRenderer
from rest_framework.response import Response
from rest_framework.views import APIView
from stored_messages.api import mark_read
from stored_messages.models import Message
from stored_messages.settings import stored_messages_settings

from datacenterlight.cms_models import DCLCalculatorPluginModel
from datacenterlight.models import VMTemplate, VMPricing
from datacenterlight.utils import (
    create_vm, get_cms_integration, check_otp, validate_vat_number
)
from hosting.models import UserCardDetail, StripeTaxRate
from membership.models import CustomUser, StripeCustomer
from opennebula_api.models import OpenNebulaManager
from opennebula_api.serializers import (
    VirtualMachineSerializer, VirtualMachineTemplateSerializer,
    VMTemplateSerializer
)
from utils.forms import (
    BillingAddressForm, PasswordResetRequestForm, UserBillingAddressForm,
    ResendActivationEmailForm
)
from utils.hosting_utils import get_all_public_keys
from utils.hosting_utils import (
    get_vm_price_with_vat, get_vm_price_for_given_vat, HostingUtils,
    get_vat_rate_for_country
)
from utils.ldap_manager import LdapManager
from utils.mailer import BaseEmail
from utils.stripe_utils import StripeUtils
from utils.tasks import send_plain_email_task
from utils.views import (
    PasswordResetViewMixin, PasswordResetConfirmViewMixin, LoginViewMixin,
    ResendActivationLinkViewMixin
)
from .forms import (
    HostingUserSignupForm, HostingUserLoginForm, UserHostingKeyForm,
    generate_ssh_key_name
)
from .mixins import ProcessVMSelectionMixin, HostingContextMixin
from .models import (
    HostingOrder, HostingBill, HostingPlan, UserHostingKey, VMDetail,
    GenericProduct, MonthlyHostingBill
)

logger = logging.getLogger(__name__)


CONNECTION_ERROR = "Your VMs cannot be displayed at the moment due to a \
                    backend connection error. please try again in a few \
                    minutes."


decorators = [never_cache]


class DashboardView(LoginRequiredMixin, View):
    template_name = "hosting/dashboard.html"
    login_url = reverse_lazy('hosting:login')

    def get_context_data(self, **kwargs):
        context = {}
        return context

    @method_decorator(decorators)
    def get(self, request, *args, **kwargs):
        context = self.get_context_data()
        context['has_invoices'] = False
        try:
            bills = []
            if hasattr(self.request.user, 'stripecustomer'):
                bills = MonthlyHostingBill.objects.filter(
                    customer=self.request.user.stripecustomer
                )
            if len(bills) > 0:
                context['has_invoices'] = True
        except MonthlyHostingBill.DoesNotExist as dne:
            logger.error("{}'s monthly hosting bill not imported ?".format(
                self.request.user.email
            ))
        return render(request, self.template_name, context)


class DjangoHostingView(ProcessVMSelectionMixin, View):
    template_name = "hosting/django.html"

    def get_context_data(self, **kwargs):
        HOSTING = 'django'
        templates = OpenNebulaManager().get_templates()
        data = VirtualMachineTemplateSerializer(templates, many=True).data
        configuration_options = HostingPlan.get_serialized_configs()
        context = {
            'hosting': HOSTING,
            'hosting_long': "Django",
            # 'configuration_detail': configuration_detail,
            'domain': "django-hosting.ch",
            'google_analytics': "UA-62285904-6",
            'vm_types': data,
            'email': "info@django-hosting.ch",
            'configuration_options': configuration_options,
            'templates': templates,
        }

        return context

    def get(self, request, *args, **kwargs):
        request.session['hosting_url'] = reverse('hosting:djangohosting')
        context = self.get_context_data()

        return render(request, self.template_name, context)


class RailsHostingView(ProcessVMSelectionMixin, View):
    template_name = "hosting/rails.html"

    def get_context_data(self, **kwargs):
        HOSTING = 'rails'

        templates = OpenNebulaManager().get_templates()
        configuration_options = HostingPlan.get_serialized_configs()

        context = {
            'hosting': HOSTING,
            'hosting_long': "Ruby On Rails",
            'domain': "rails-hosting.ch",
            'google_analytics': "UA-62285904-5",
            'email': "info@rails-hosting.ch",
            'configuration_options': configuration_options,
            'templates': templates,
        }
        return context

    def get(self, request, *args, **kwargs):
        request.session['hosting_url'] = reverse('hosting:railshosting')
        context = self.get_context_data()
        return render(request, self.template_name, context)


class NodeJSHostingView(ProcessVMSelectionMixin, View):
    template_name = "hosting/nodejs.html"

    def get_context_data(self, **kwargs):
        HOSTING = 'nodejs'
        templates = OpenNebulaManager().get_templates()
        configuration_options = HostingPlan.get_serialized_configs()

        context = {
            'hosting': HOSTING,
            'hosting_long': "NodeJS",
            # 'configuration_detail': configuration_detail,
            'domain': "node-hosting.ch",
            'google_analytics': "UA-62285904-7",
            'email': "info@node-hosting.ch",
            'templates': templates,
            'configuration_options': configuration_options,

        }
        return context

    def get(self, request, *args, **kwargs):
        request.session['hosting_url'] = reverse('hosting:nodejshosting')
        context = self.get_context_data()

        return render(request, self.template_name, context)


class HostingPricingView(ProcessVMSelectionMixin, View):
    template_name = "hosting/hosting_pricing.html"

    def get_context_data(self, **kwargs):
        # configuration_options = dict(VirtualMachinePlan.VM_CONFIGURATION)
        templates = OpenNebulaManager().get_templates()
        configuration_options = HostingPlan.get_serialized_configs()

        context = {
            # 'configuration_options': configuration_options,
            'email': "info@django-hosting.ch",
            'templates': templates,
            'configuration_options': configuration_options,

        }

        return context

    def get(self, request, *args, **kwargs):
        request.session['hosting_url'] = reverse('hosting:djangohosting')
        context = self.get_context_data()

        return render(request, self.template_name, context)


class IndexView(View):
    template_name = "hosting/index.html"

    def get_context_data(self, **kwargs):
        templates = OpenNebulaManager().get_templates()
        data = VirtualMachineTemplateSerializer(templates, many=True).data

        context = {
            'hosting': "nodejs",
            'hosting_long': "NodeJS",
            'domain': "node-hosting.ch",
            'google_analytics': "UA-62285904-7",
            'email': "info@node-hosting.ch",
            'vm_types': data
            # 'vm_types': VirtualMachineType.get_serialized_vm_types(),
        }
        return context

    @method_decorator(decorators)
    def get(self, request, *args, **kwargs):
        context = self.get_context_data()
        return render(request, self.template_name, context)


class LoginView(HostingContextMixin, LoginViewMixin):
    template_name = "hosting/login.html"
    form_class = HostingUserLoginForm
    success_url = reverse_lazy('hosting:dashboard')


class SignupView(HostingContextMixin, CreateView):
    template_name = 'hosting/signup.html'
    form_class = HostingUserSignupForm
    model = CustomUser
    success_url = reverse_lazy('hosting:dashboard')

    def get_success_url(self):
        next_url = self.request.session.get(
            'next', self.success_url)
        return next_url

    def form_valid(self, form):
        name = form.cleaned_data.get('name')
        email = form.cleaned_data.get('email')
        password = form.cleaned_data.get('password')
        this_base_url = "{0}://{1}".format(self.request.scheme,
                                           self.request.get_host())
        CustomUser.register(name, password, email,
                            app='dcl', base_url=this_base_url)

        return HttpResponseRedirect(reverse_lazy('hosting:signup-validate'))

    @method_decorator(decorators)
    def get(self, request, *args, **kwargs):
        if self.request.user.is_authenticated():
            return HttpResponseRedirect(self.get_success_url())
        return super(SignupView, self).get(request, *args, **kwargs)


class SignupValidateView(HostingContextMixin, TemplateView):
    template_name = "hosting/signup_validate.html"

    def get_context_data(self, **kwargs):
        context = super(SignupValidateView, self).get_context_data(**kwargs)
        login_url = '<a href="' + \
                    reverse('hosting:login') + '">' + str(_('login')) + '</a>'
        home_url = '<a href="' + \
                   reverse('datacenterlight:index') + '">Data Center Light</a>'
        message = '{signup_success_message} {lurl}</a> \
                 <br />{go_back} {hurl}.'.format(
            signup_success_message=_(
                'Thank you for signing up. We have sent an email to you. '
                'Please follow the instructions in it to activate your '
                'account. Once activated, you can login using'),
            go_back=_('Go back to'),
            lurl=login_url,
            hurl=home_url
        )
        context['message'] = mark_safe(message)
        context['section_title'] = _('Sign up')
        return context


class SignupValidatedView(SignupValidateView, HostingContextMixin):
    template_name = "hosting/signup_validate.html"

    def get_context_data(self, **kwargs):
        context = super(SignupValidateView, self).get_context_data(**kwargs)
        validated = CustomUser.validate_url(self.kwargs['validate_slug'])
        login_url = '<a href="' + \
                    reverse('hosting:login') + '">' + str(_('login')) + '</a>'
        section_title = _('Account activation')
        user = CustomUser.objects.filter(
            validation_slug=self.kwargs['validate_slug']).first()
        if validated:
            message = ('{account_activation_string} <br />'
                       ' {login_string} {lurl}.').format(
                account_activation_string=_(
                    "Your account has been activated."),
                login_string=_("You can now"),
                lurl=login_url)
            email_data = {
                'subject': _('Welcome to Data Center Light!'),
                'to': user.email,
                'context': {
                    'base_url': "{0}://{1}".format(
                        self.request.scheme,
                        self.request.get_host()
                    )
                },
                'template_name': 'welcome_user',
                'template_path': 'datacenterlight/emails/',
                'from_address': settings.DCL_SUPPORT_FROM_ADDRESS,
            }
            email = BaseEmail(**email_data)
            email.send()
        else:
            home_url = '<a href="' + \
                       reverse('datacenterlight:cms_index') + \
                       '">Data Center Light</a>'
            message = '{sorry_message} <br />{go_back_to} {hurl}'.format(
                sorry_message=_("Sorry. Your request is invalid."),
                go_back_to=_('Go back to'),
                hurl=home_url
            )
        context['message'] = mark_safe(message)
        context['section_title'] = section_title
        return context

    @method_decorator(decorators)
    def get(self, request, *args, **kwargs):
        if self.request.user.is_authenticated():
            return HttpResponseRedirect(reverse_lazy('hosting:dashboard'))
        return super(SignupValidatedView, self).get(request, *args, **kwargs)


class ResendActivationEmailView(HostingContextMixin,
                                ResendActivationLinkViewMixin):
    template_name = 'hosting/resend_activation_link.html'
    form_class = ResendActivationEmailForm
    success_url = reverse_lazy('hosting:login')
    email_template_path = 'datacenterlight/emails/'
    email_template_name = 'user_activation'


class PasswordResetView(HostingContextMixin, PasswordResetViewMixin):
    site = 'dcl'
    template_name = 'hosting/reset_password.html'
    form_class = PasswordResetRequestForm
    success_url = reverse_lazy('hosting:login')
    template_email_path = 'hosting/emails/'


class PasswordResetConfirmView(HostingContextMixin,
                               PasswordResetConfirmViewMixin):
    template_name = 'hosting/confirm_reset_password.html'
    success_url = reverse_lazy('hosting:login')

    def post(self, request, uidb64=None, token=None, *arg, **kwargs):
        try:
            uid = urlsafe_base64_decode(uidb64)
            user = CustomUser.objects.get(pk=uid)

            opennebula_client = OpenNebulaManager(
                email=user.username,
                password=user.password,
            )

        except (TypeError, ValueError, OverflowError, CustomUser.DoesNotExist):
            user = None
            opennebula_client = None

        form = self.form_class(request.POST)

        if user is not None and default_token_generator.check_token(user,
                                                                    token):
            if form.is_valid():
                ldap_manager = LdapManager()
                new_password = form.cleaned_data['new_password2']

                # Make sure the user have an ldap account already
                user.create_ldap_account(new_password)

                # We are changing password in ldap before changing in database because
                # ldap have more chances of failure than local database
                if ldap_manager.change_password(user.username, new_password):
                    user.set_password(new_password)
                    user.save()

                    messages.success(request, _('Password has been reset.'))

                    # Change opennebula password
                    opennebula_client.change_user_password(user.password)

                    return self.form_valid(form)

            messages.error(
                request, _('Password reset has not been successful.'))
            form.add_error(None,
                           _('Password reset has not been successful.'))
            return self.form_invalid(form)

        else:
            error_msg = _('The reset password link is no longer valid.')
            messages.error(request, _(error_msg))
            form.add_error(None, error_msg)
            return self.form_invalid(form)


class NotificationsView(LoginRequiredMixin, TemplateView):
    template_name = 'hosting/notifications.html'
    login_url = reverse_lazy('hosting:login')

    def get_context_data(self, **kwargs):
        context = super(NotificationsView, self).get_context_data(**kwargs)
        backend = stored_messages_settings.STORAGE_BACKEND()
        unread_notifications = backend.inbox_list(self.request.user)
        read_notifications = backend.archive_list(self.request.user)
        context.update({
            'unread_notifications': unread_notifications,
            'all_notifications': read_notifications + unread_notifications
        })
        return context


class MarkAsReadNotificationView(LoginRequiredMixin, UpdateView):
    model = Message
    success_url = reverse_lazy('hosting:notifications')
    login_url = reverse_lazy('hosting:login')
    fields = '__all__'

    def post(self, *args, **kwargs):
        message = self.get_object()
        backend = stored_messages_settings.STORAGE_BACKEND()
        backend.archive_store([self.request.user], message)
        mark_read(self.request.user, message)
        return HttpResponseRedirect(reverse('hosting:notifications'))


class SSHKeyDeleteView(LoginRequiredMixin, DeleteView):
    login_url = reverse_lazy('hosting:login')
    success_url = reverse_lazy('hosting:ssh_keys')
    model = UserHostingKey

    def get_object(self, queryset=None):
        """ Hook to ensure UserHostingKey object is owned by request.user.
            We reply with a Http404 if the user is not the owner of the key.
        """
        obj = super(SSHKeyDeleteView, self).get_object()
        if not obj.user == self.request.user:
            raise Http404
        return obj

    def delete(self, request, *args, **kwargs):
        owner = self.request.user
        manager = OpenNebulaManager(
            email=owner.username,
            password=owner.password
        )
        pk = self.kwargs.get('pk')
        # Get user ssh key
        public_key = UserHostingKey.objects.get(pk=pk).public_key
        keys = UserHostingKey.objects.filter(user=self.request.user)
        keys_to_save = [k.public_key for k in keys if k.public_key != public_key]
        manager.save_key_in_opennebula_user('\n'.join(keys_to_save), update_type=0)

        return super(SSHKeyDeleteView, self).delete(request, *args, **kwargs)


class SSHKeyListView(LoginRequiredMixin, ListView):
    template_name = "hosting/user_keys.html"
    login_url = reverse_lazy('hosting:login')
    context_object_name = "keys"
    model = UserHostingKey
    paginate_by = 10
    ordering = '-id'

    def get_queryset(self):
        user = self.request.user
        self.queryset = UserHostingKey.objects.filter(user=user)
        return super(SSHKeyListView, self).get_queryset()

    @method_decorator(decorators)
    def render_to_response(self, context, **response_kwargs):
        if not self.queryset:
            return HttpResponseRedirect(reverse('hosting:choice_ssh_keys'))
        return super(SSHKeyListView, self).render_to_response(context,
                                                              **response_kwargs)


class SSHKeyChoiceView(LoginRequiredMixin, View):
    template_name = "hosting/choice_ssh_keys.html"
    login_url = reverse_lazy('hosting:login')

    @method_decorator(decorators)
    def get(self, request, *args, **kwargs):
        context = {}
        return render(request, self.template_name, context)

    @method_decorator(decorators)
    def post(self, request, *args, **kwargs):
        name = generate_ssh_key_name()
        private_key, public_key = UserHostingKey.generate_keys()
        content = ContentFile(private_key)
        ssh_key = UserHostingKey.objects.create(
            user=request.user, public_key=public_key, name=name)
        filename = name + '_' + str(uuid.uuid4())[:8] + '_private.pem'
        ssh_key.private_key.save(filename, content)
        owner = self.request.user
        manager = OpenNebulaManager(
            email=owner.username,
            password=owner.password
        )
        keys = get_all_public_keys(request.user)
        manager.save_key_in_opennebula_user('\n'.join(keys))
        return redirect(reverse_lazy('hosting:ssh_keys'), foo='bar')


@method_decorator(decorators, name='dispatch')
class SettingsView(LoginRequiredMixin, FormView):
    template_name = "hosting/settings.html"
    login_url = reverse_lazy('hosting:login')
    form_class = BillingAddressForm
    permission_required = ['view_usercarddetail']

    def get_form(self, form_class):
        """
        Check if the user already saved contact details. If so, then show
        the form populated with those details, to let user change them.
        """
        return form_class(
            instance=self.request.user.billing_addresses.first(),
            **self.get_form_kwargs())

    def get_context_data(self, **kwargs):
        context = super(SettingsView, self).get_context_data(**kwargs)
        # Get user
        user = self.request.user
        stripe_customer = None
        if hasattr(user, 'stripecustomer'):
            stripe_customer = user.stripecustomer
        cards_list = UserCardDetail.get_all_cards_list(
            stripe_customer=stripe_customer
        )
        context.update({
            'cards_list': cards_list,
            'stripe_key': settings.STRIPE_API_PUBLIC_KEY
        })

        return context

    def post(self, request, *args, **kwargs):
        if 'card' in request.POST and request.POST['card'] is not '':
            card_id = escape(request.POST['card'])
            user_card_detail = UserCardDetail.objects.get(id=card_id)
            UserCardDetail.set_default_card(
                stripe_api_cus_id=request.user.stripecustomer.stripe_id,
                stripe_source_id=user_card_detail.card_id
            )
            msg = _(
                ("Your {brand} card ending in {last4} set as "
                 "default card").format(
                    brand=user_card_detail.brand,
                    last4=user_card_detail.last4
                )
            )
            messages.add_message(request, messages.SUCCESS, msg)
            return HttpResponseRedirect(reverse_lazy('hosting:settings'))
        if 'delete_card' in request.POST:
            try:
                card = UserCardDetail.objects.get(pk=self.kwargs.get('pk'))
                if (request.user.has_perm(self.permission_required[0], card)
                        and
                        request.user
                                .stripecustomer
                                .usercarddetail_set
                                .count() > 1):
                    if card.card_id is not None:
                        stripe_utils = StripeUtils()
                        stripe_utils.dissociate_customer_card(
                            request.user.stripecustomer.stripe_id,
                            card.card_id
                        )
                        if card.preferred:
                            UserCardDetail.set_default_card_from_stripe(
                                request.user.stripecustomer.stripe_id
                            )
                        card.delete()
                        msg = _("Card deassociation successful")
                        messages.add_message(request, messages.SUCCESS, msg)
                else:
                    msg = _("You are not permitted to do this operation")
                    messages.add_message(request, messages.ERROR, msg)
            except UserCardDetail.DoesNotExist:
                msg = _("The selected card does not exist")
                messages.add_message(request, messages.ERROR, msg)
            return HttpResponseRedirect(reverse_lazy('hosting:settings'))
        form = self.get_form()
        if form.is_valid():
            if 'billing-form' in request.POST:
                current_billing_address = self.request.user.billing_addresses.last()
                billing_address_data = form.cleaned_data
                billing_address_data.update({
                    'user': self.request.user.id
                })
                billing_address_user_form = UserBillingAddressForm(
                    instance=self.request.user.billing_addresses.order_by('-id').first(),
                    data=billing_address_data)
                billing_address = billing_address_user_form.save()
                billing_address.stripe_tax_id = ''
                billing_address.vat_number_validated_on = None
                billing_address.vat_validation_status = ''
                billing_address.save()
                vat_number = billing_address_user_form.cleaned_data.get(
                    'vat_number').strip()
                logger.debug("Vat number = %s" % vat_number)
                if vat_number:
                    try:
                        stripe_customer = request.user.stripecustomer
                    except StripeCustomer.DoesNotExist as dne:
                        logger.debug(
                            "User %s does not have a stripecustomer. "
                            "Creating one." % request.user.email)
                        stripe_customer = StripeCustomer.get_or_create(
                            email=request.user.email,
                            token=None)
                        request.user.stripecustomer = stripe_customer
                        request.user.save()
                    validate_result = validate_vat_number(
                        stripe_customer_id=request.user.stripecustomer.stripe_id,
                        billing_address_id=billing_address.id,
                        is_user_ba=True
                    )
                    logger.debug("validate_result = %s" % str(validate_result))
                    if 'error' in validate_result and validate_result['error']:
                        messages.add_message(
                            request, messages.ERROR,
                            "VAT Number validation error: %s" % validate_result["error"],
                            extra_tags='error'
                        )
                        billing_address = current_billing_address
                        if billing_address:
                            billing_address.save()
                        email_data = {
                            'subject': "%s updated VAT number to %s but failed" %
                                       (request.user.email, vat_number),
                            'from_email': settings.DCL_SUPPORT_FROM_ADDRESS,
                            'to': settings.DCL_ERROR_EMAILS_TO_LIST,
                            'body': "\n".join(
                                ["%s=%s" % (k, v) for (k, v) in
                                 validate_result.items()]),
                        }
                    else:
                        email_data = {
                            'subject': "%s updated VAT number to %s" % (
                                request.user.email, vat_number
                            ),
                            'from_email': settings.DCL_SUPPORT_FROM_ADDRESS,
                            'to': settings.DCL_ERROR_EMAILS_TO_LIST,
                            'body': "\n".join(
                                ["%s=%s" % (k, v) for (k, v) in
                                 validate_result.items()]),
                        }
                        msg = _("Billing address updated successfully")
                        messages.add_message(request, messages.SUCCESS, msg)
                    send_plain_email_task.delay(email_data)
                else:
                    msg = _("Billing address updated successfully")
                    messages.add_message(request, messages.SUCCESS, msg)
            else:
                token = form.cleaned_data.get('token')
                stripe_utils = StripeUtils()
                card_details = stripe_utils.get_cards_details_from_token(
                    token
                )
                if not card_details.get('response_object'):
                    form.add_error("__all__", card_details.get('error'))
                    return self.render_to_response(self.get_context_data())
                stripe_customer = StripeCustomer.get_or_create(
                    email=request.user.email, token=token
                )
                card = card_details['response_object']
                if UserCardDetail.get_user_card_details(stripe_customer, card):
                    msg = _('You seem to have already added this card')
                    messages.add_message(request, messages.ERROR, msg)
                else:
                    acc_result = stripe_utils.associate_customer_card(
                        request.user.stripecustomer.stripe_id, token
                    )
                    if acc_result['response_object'] is None:
                        msg = _(
                            'An error occurred while associating the card.'
                            ' Details: {details}'.format(
                                details=acc_result['error']
                            )
                        )
                        messages.add_message(request, messages.ERROR, msg)
                        return self.render_to_response(self.get_context_data())
                    preferred = False
                    if stripe_customer.usercarddetail_set.count() == 0:
                        preferred = True
                    UserCardDetail.create(
                        stripe_customer=stripe_customer,
                        last4=card['last4'],
                        brand=card['brand'],
                        fingerprint=card['fingerprint'],
                        exp_month=card['exp_month'],
                        exp_year=card['exp_year'],
                        card_id=card['card_id'],
                        preferred=preferred
                    )
                    msg = _(
                        "Successfully associated the card with your account"
                    )
                    messages.add_message(request, messages.SUCCESS, msg)
            return self.render_to_response(self.get_context_data())
        else:
            billing_address_data = form.cleaned_data
            return self.form_invalid(form)


class PaymentVMView(LoginRequiredMixin, FormView):
    template_name = 'hosting/payment.html'
    login_url = reverse_lazy('hosting:login')
    form_class = BillingAddressForm

    def get_form_kwargs(self):
        current_billing_address = self.request.user.billing_addresses.last()
        form_kwargs = super(PaymentVMView, self).get_form_kwargs()
        if not current_billing_address:
            return form_kwargs

        form_kwargs.update({
            'initial': {
                'cardholder_name': current_billing_address.cardholder_name,
                'street_address': current_billing_address.street_address,
                'city': current_billing_address.city,
                'postal_code': current_billing_address.postal_code,
                'country': current_billing_address.country,
                'vat_number': current_billing_address.vat_number
            }
        })
        return form_kwargs

    def get_context_data(self, **kwargs):
        context = super(PaymentVMView, self).get_context_data(**kwargs)
        # Get user
        user = self.request.user
        if hasattr(user, 'stripecustomer'):
            stripe_customer = user.stripecustomer
        else:
            stripe_customer = None
        cards_list = UserCardDetail.get_all_cards_list(
            stripe_customer=stripe_customer
        )
        context.update({
            'stripe_key': settings.STRIPE_API_PUBLIC_KEY,
            'vm_pricing': VMPricing.get_vm_pricing_by_name(
                self.request.session.get('specs', {}).get('pricing_name')
            ),
            'cards_list': cards_list,
        })

        return context

    @method_decorator(decorators)
    def get(self, request, *args, **kwargs):
        if 'next' in request.session:
            del request.session['next']
        HostingUtils.clear_items_from_list(
            request.session,
            ['token', 'card_id', 'customer', 'user']
        )
        return self.render_to_response(self.get_context_data())

    @method_decorator(decorators)
    def post(self, request, *args, **kwargs):
        form = self.get_form()
        if form.is_valid():
            # Get billing address data
            billing_address_data = form.cleaned_data
            token = form.cleaned_data.get('token')
            owner = self.request.user
            if token is '':
                card_id = form.cleaned_data.get('card')
                customer = owner.stripecustomer
                try:
                    user_card_detail = UserCardDetail.objects.get(id=card_id)
                    if not request.user.has_perm(
                            'view_usercarddetail', user_card_detail
                    ):
                        raise UserCardDetail.DoesNotExist(
                            _("{user} does not have permission to access the "
                              "card").format(user=request.user.email)
                        )
                except UserCardDetail.DoesNotExist as e:
                    ex = str(e)
                    logger.error("Card Id: {card_id}, Exception: {ex}".format(
                            card_id=card_id, ex=ex
                        )
                    )
                    msg = _("An error occurred. Details: {}".format(ex))
                    messages.add_message(
                        self.request, messages.ERROR, msg,
                        extra_tags='make_charge_error'
                    )
                    return HttpResponseRedirect(
                        reverse('hosting:payment') + '#payment_error'
                    )
                request.session['card_id'] = user_card_detail.id
            else:
                # Get or create stripe customer
                customer = StripeCustomer.get_or_create(
                    email=owner.email, token=token
                )
                if not customer:
                    msg = _("Invalid credit card")
                    messages.add_message(
                        self.request, messages.ERROR, msg,
                        extra_tags='make_charge_error')
                    return HttpResponseRedirect(
                        reverse('hosting:payment') + '#payment_error')
                request.session['token'] = token
            request.session['billing_address_data'] = billing_address_data
            self.request.session['order_confirm_url'] = "{url}?{query_params}".format(
                    url=reverse('hosting:order-confirmation'),
                    query_params='page=payment')
            return HttpResponseRedirect(reverse('hosting:add_ssh_key'))
        else:
            return self.form_invalid(form)


class OrdersHostingDetailView(LoginRequiredMixin, DetailView, FormView):
    form_class = UserHostingKeyForm
    template_name = "hosting/order_detail.html"
    context_object_name = "order"
    login_url = reverse_lazy('hosting:login')
    permission_required = ['view_hostingorder']
    model = HostingOrder

    def get_form_kwargs(self):
        kwargs = super(OrdersHostingDetailView, self).get_form_kwargs()
        kwargs.update({'request': self.request})
        return kwargs

    def get_object(self, queryset=None):
        order_id = self.kwargs.get('pk')
        try:
            hosting_order_obj = HostingOrder.objects.get(pk=order_id)
            logger.debug("Found HostingOrder for id {order_id}".format(
                order_id=order_id
            ))
        except HostingOrder.DoesNotExist:
            logger.debug("HostingOrder not found for id {order_id}".format(
                order_id=order_id
            ))
            hosting_order_obj = None
        return hosting_order_obj

    def get_context_data(self, **kwargs):
        # Get context
        context = super(
            OrdersHostingDetailView, self
        ).get_context_data(**kwargs)
        obj = self.get_object()
        owner = self.request.user

        if self.request.GET.get('page') == 'payment':
            context['page_header_text'] = _('Confirm Order')
            context['form'] = UserHostingKeyForm(request=self.request)
            context['keys'] = get_all_public_keys(self.request.user)
        else:
            context['page_header_text'] = _('Invoice')
            if not self.request.user.has_perm(
                    self.permission_required[0], obj
            ):
                logger.debug(
                    "User {user} does not have permission on HostingOrder "
                    "{order_id}. Raising 404 error now.".format(
                        user=self.request.user.email,
                        order_id=obj.id if obj else 'None'
                    )
                )
                raise Http404

        if obj is not None:
            if obj.generic_product_id is not None:
                # generic payment case
                logger.debug("Generic payment case")
                context['product_name'] = GenericProduct.objects.get(
                    id=obj.generic_product_id
                ).product_name
            else:
                # invoice for previous order
                logger.debug("Invoice of VM order")
                try:
                    vm_detail = VMDetail.objects.get(vm_id=obj.vm_id)
                    context['vm'] = vm_detail.__dict__
                    context['vm']['name'] = '{}-{}'.format(
                        context['vm']['configuration'], context['vm']['vm_id'])
                    user_vat_country = obj.billing_address.country
                    user_country_vat_rate = get_vat_rate_for_country(
                        user_vat_country)
                    price, vat, vat_percent, discount = get_vm_price_for_given_vat(
                        cpu=context['vm']['cores'],
                        ssd_size=context['vm']['disk_size'],
                        memory=context['vm']['memory'],
                        pricing_name=(obj.vm_pricing.name
                                      if obj.vm_pricing else 'default'),
                        vat_rate= (
                            user_country_vat_rate * 100
                            if obj.vm_id >= settings.FIRST_VM_ID_AFTER_EU_VAT
                            else settings.PRE_EU_VAT_RATE
                        )
                    )
                    context['vm']["after_eu_vat_intro"] = (
                        True if obj.vm_id >= settings.FIRST_VM_ID_AFTER_EU_VAT
                        else False
                    )
                    context['vm']["price"] = price
                    context['vm']["vat"] = vat
                    context['vm']["vat_percent"] = vat_percent
                    context['vm']["vat_country"] = user_vat_country
                    context['vm']["discount"] = discount
                    context['vm']["total_price"] = round(
                        price + vat - discount['amount'], 2)
                    context['subscription_end_date'] = vm_detail.end_date()
                except VMDetail.DoesNotExist:
                    try:
                        manager = OpenNebulaManager(
                            email=owner.username, password=owner.password
                        )
                        vm = manager.get_vm(obj.vm_id)
                        context['vm'] = VirtualMachineSerializer(vm).data
                        user_vat_country = obj.billing_address.country
                        user_country_vat_rate = get_vat_rate_for_country(
                            user_vat_country)
                        price, vat, vat_percent, discount = get_vm_price_for_given_vat(
                            cpu=context['vm']['cores'],
                            ssd_size=context['vm']['disk_size'],
                            memory=context['vm']['memory'],
                            pricing_name=(obj.vm_pricing.name
                                          if obj.vm_pricing else 'default'),
                            vat_rate=(
                                user_country_vat_rate * 100
                                if obj.vm_id >= settings.FIRST_VM_ID_AFTER_EU_VAT
                                else settings.PRE_EU_VAT_RATE
                            )
                        )
                        context['vm']["after_eu_vat_intro"] = (
                            True if obj.vm_id >= settings.FIRST_VM_ID_AFTER_EU_VAT
                            else False
                        )
                        context['vm']["price"] = price
                        context['vm']["vat"] = vat
                        context['vm']["vat_percent"] = vat_percent
                        context['vm']["vat_country"] = user_vat_country
                        context['vm']["discount"] = discount
                        context['vm']["total_price"] = round(
                            price + vat - discount['amount'], 2)
                    except WrongIdError:
                        messages.error(
                            self.request,
                            _('The VM you are looking for is unavailable at the '
                              'moment. Please contact Data Center Light support.')
                        )
                        self.kwargs['error'] = 'WrongIdError'
                        context['error'] = 'WrongIdError'
                    except ConnectionRefusedError:
                        messages.error(
                            self.request,
                            _('In order to create a VM, you need to create/upload '
                              'your SSH KEY first.')
                        )
        else:
            # new order, confirm payment
            if 'token' in self.request.session:
                token = self.request.session['token']
                stripe_utils = StripeUtils()
                card_details = stripe_utils.get_cards_details_from_token(
                    token
                )
                if not card_details.get('response_object'):
                    return HttpResponseRedirect(reverse('hosting:payment'))
                card_details_response = card_details['response_object']
                context['cc_last4'] = card_details_response['last4']
                context['cc_brand'] = card_details_response['brand']
                context['cc_exp_year'] = card_details_response['exp_year']
                context['cc_exp_month'] = card_details_response['exp_month']
            else:
                card_id = self.request.session.get('card_id')
                card_detail = UserCardDetail.objects.get(id=card_id)
                context['cc_last4'] = card_detail.last4
                context['cc_brand'] = card_detail.brand
                context['cc_exp_year'] = card_detail.exp_year
                context['cc_exp_month'] = '{:02d}'.format(card_detail.exp_month)
            context['site_url'] = reverse('hosting:create_virtual_machine')
            vm_specs = self.request.session.get('specs')
            user_vat_country = (
                self.request.session.get('billing_address_data').get("country")
            )
            user_country_vat_rate = get_vat_rate_for_country(user_vat_country)
            price, vat, vat_percent, discount = get_vm_price_for_given_vat(
                cpu=vm_specs['cpu'],
                memory=vm_specs['memory'],
                ssd_size=vm_specs['disk_size'],
                pricing_name=vm_specs['pricing_name'],
                vat_rate=user_country_vat_rate * 100
            )
            vm_specs["price"] = price
            vm_specs["vat"] = vat
            vm_specs["vat_percent"] = vat_percent
            vm_specs["vat_country"] = user_vat_country
            vm_specs["discount"] = discount
            vm_specs["total_price"] = round(price + vat - discount['amount'],
                                            2)
            vm_specs["after_eu_vat_intro"] = True
            context['vm'] = vm_specs
        return context

    @method_decorator(decorators)
    def get(self, request, *args, **kwargs):
        if not self.kwargs.get('pk'):
            if 'specs' not in self.request.session:
                return HttpResponseRedirect(
                    reverse('hosting:create_virtual_machine')
                )

            if ('token' not in self.request.session and
                    'card_id' not in self.request.session):
                return HttpResponseRedirect(reverse('hosting:payment'))
        self.object = self.get_object()
        context = self.get_context_data(object=self.object)
        if 'failed_payment' in context:
            msg = context['card_details'].get('error')
            messages.add_message(
                self.request, messages.ERROR, msg,
                extra_tags='failed_payment'
            )
            return HttpResponseRedirect(
                reverse('hosting:payment') + '#payment_error'
            )
        return self.render_to_response(context)

    @method_decorator(decorators)
    def post(self, request):
        template = request.session.get('template')
        specs = request.session.get('specs')
        stripe_utils = StripeUtils()
        # We assume that if the user is here, his/her StripeCustomer
        # object already exists
        stripe_customer_id = request.user.stripecustomer.id
        billing_address_data = request.session.get('billing_address_data')
        vm_template_id = template.get('id', 1)
        stripe_api_cus_id = request.user.stripecustomer.stripe_id
        if 'token' in self.request.session:
            card_details = stripe_utils.get_cards_details_from_token(
                request.session['token']
            )
            if not card_details.get('response_object'):
                return HttpResponseRedirect(reverse('hosting:payment'))
            card_details_response = card_details['response_object']
            card_details_dict = {
                'last4': card_details_response['last4'],
                'brand': card_details_response['brand'],
                'card_id': card_details_response['card_id']
            }
            ucd = UserCardDetail.get_user_card_details(
                request.user.stripecustomer, card_details_response
            )
            if not ucd:
                acc_result = stripe_utils.associate_customer_card(
                    stripe_api_cus_id, request.session['token'],
                    set_as_default=True
                )
                if acc_result['response_object'] is None:
                    msg = _(
                        'An error occurred while associating the card.'
                        ' Details: {details}'.format(
                            details=acc_result['error']
                        )
                    )
                    messages.add_message(self.request, messages.ERROR, msg,
                                         extra_tags='failed_payment')
                    response = {
                        'status': False,
                        'redirect': "{url}#{section}".format(
                            url=reverse('hosting:payment'),
                            section='payment_error'),
                        'msg_title': str(_('Error.')),
                        'msg_body': str(
                            _('There was a payment related error.'
                              ' On close of this popup, you will be redirected'
                              ' back to the payment page.')
                        )
                    }
                    return JsonResponse(response)
        else:
            card_id = request.session.get('card_id')
            user_card_detail = UserCardDetail.objects.get(id=card_id)
            card_details_dict = {
                'last4': user_card_detail.last4,
                'brand': user_card_detail.brand,
                'card_id': user_card_detail.card_id
            }
            if not user_card_detail.preferred:
                UserCardDetail.set_default_card(
                    stripe_api_cus_id=stripe_api_cus_id,
                    stripe_source_id=user_card_detail.card_id
                )
        cpu = specs.get('cpu')
        memory = specs.get('memory')
        disk_size = specs.get('disk_size')
        amount_to_be_charged = specs.get('price')
        discount = specs.get('discount')
        plan_name = StripeUtils.get_stripe_plan_name(
            cpu=cpu,
            memory=memory,
            disk_size=disk_size,
            price=amount_to_be_charged
        )
        stripe_plan_id = StripeUtils.get_stripe_plan_id(
            cpu=cpu,
            ram=memory,
            ssd=disk_size,
            version=1,
            app='dcl',
            price=amount_to_be_charged
        )
        stripe_plan = stripe_utils.get_or_create_stripe_plan(
            amount=amount_to_be_charged,
            name=plan_name,
            stripe_plan_id=stripe_plan_id)
        # Create StripeTaxRate if applicable to the user
        stripe_tax_rate = None
        if specs["vat_percent"] > 0:
            try:
                stripe_tax_rate = StripeTaxRate.objects.get(
                    description="VAT for %s" % specs["vat_country"]
                )
                print("Stripe Tax Rate exists")
            except StripeTaxRate.DoesNotExist as dne:
                print("StripeTaxRate does not exist")
                tax_rate_obj = stripe.TaxRate.create(
                    display_name="VAT",
                    description="VAT for %s" % specs["vat_country"],
                    jurisdiction=specs["vat_country"],
                    percentage=specs["vat_percent"] * 100,
                    inclusive=False,
                )
                stripe_tax_rate = StripeTaxRate.objects.create(
                    display_name=tax_rate_obj.display_name,
                    description=tax_rate_obj.description,
                    jurisdiction=tax_rate_obj.jurisdiction,
                    percentage=tax_rate_obj.percentage,
                    inclusive=False,
                    tax_rate_id=tax_rate_obj.id
                )
                logger.debug("Created StripeTaxRate %s" %
                             stripe_tax_rate.tax_rate_id)
        subscription_result = stripe_utils.subscribe_customer_to_plan(
            stripe_api_cus_id,
            [{"plan": stripe_plan.get('response_object').stripe_plan_id}],
            coupon=(discount['stripe_coupon_id']
                    if 'name' in discount and
                       discount['name'] is not None and
                       'ipv6' in discount['name'].lower() and
                       discount['stripe_coupon_id']
                    else ""),
            tax_rates=[stripe_tax_rate.tax_rate_id] if stripe_tax_rate else [],
        )
        stripe_subscription_obj = subscription_result.get('response_object')
        # Check if the subscription was approved and is active
        if (stripe_subscription_obj is None or
                stripe_subscription_obj.status != 'active'):
            # At this point, we have created a Stripe API card and
            # associated it with the customer; but the transaction failed
            # due to some reason. So, we would want to dissociate this card
            # here.
            # ...

            msg = subscription_result.get('error')
            messages.add_message(self.request, messages.ERROR, msg,
                                 extra_tags='failed_payment')
            response = {
                'status': False,
                'redirect': "{url}#{section}".format(
                    url=reverse('hosting:payment'),
                    section='payment_error'),
                'msg_title': str(_('Error.')),
                'msg_body': str(
                    _('There was a payment related error.'
                      ' On close of this popup, you will be redirected back to'
                      ' the payment page.')
                )
            }
            return JsonResponse(response)

        if 'token' in request.session:
            ucd = UserCardDetail.get_or_create_user_card_detail(
                stripe_customer=self.request.user.stripecustomer,
                card_details=card_details_response
            )
            UserCardDetail.save_default_card_local(
                self.request.user.stripecustomer.stripe_id,
                ucd.card_id
            )
        user = {
            'name': self.request.user.name,
            'email': self.request.user.email,
            'username': self.request.user.username,
            'pass': self.request.user.password,
            'request_scheme': request.scheme,
            'request_host': request.get_host(),
            'language': get_language(),
        }

        create_vm(
            billing_address_data, stripe_customer_id, specs,
            stripe_subscription_obj, card_details_dict, request,
            vm_template_id, template, user
        )

        response = {
            'status': True,
            'redirect': reverse('hosting:virtual_machines'),
            'msg_title': str(_('Thank you for the order.')),
            'msg_body': str(
                _('Your VM will be up and running in a few moments.'
                  ' We will send you a confirmation email as soon as'
                  ' it is ready.'))
        }

        return JsonResponse(response)


class OrdersHostingListView(LoginRequiredMixin, ListView):
    template_name = "hosting/orders.html"
    login_url = reverse_lazy('hosting:login')
    context_object_name = "orders"
    model = HostingOrder
    paginate_by = 10
    ordering = '-id'

    def get_queryset(self):
        user = self.request.user
        self.queryset = HostingOrder.objects.filter(customer__user=user)
        return super(OrdersHostingListView, self).get_queryset()

    @method_decorator(decorators)
    def get(self, request, *args, **kwargs):
        return super(OrdersHostingListView, self).get(request, *args, **kwargs)


class InvoiceListView(LoginRequiredMixin, TemplateView):
    template_name = "hosting/invoices.html"
    login_url = reverse_lazy('hosting:login')
    context_object_name = "invoices"
    paginate_by = 10
    ordering = '-created'

    def get_context_data(self, **kwargs):
        page = self.request.GET.get('page', 1)
        context = super(InvoiceListView, self).get_context_data(**kwargs)
        invs_page = None
        invs_page_charges = None
        if ('user_email' in self.request.GET
            and self.request.user.email == settings.ADMIN_EMAIL):
            user_email = self.request.GET['user_email']
            context['user_email'] = user_email
            logger.debug(
                "user_email = {}".format(user_email)
            )
            try:
                cu = CustomUser.objects.get(email=user_email)
            except CustomUser.DoesNotExist as dne:
                logger.debug("User does not exist")
                cu = self.request.user
            invs = stripe.Invoice.list(customer=cu.stripecustomer.stripe_id,
                                       count=100)
            paginator = Paginator(invs.data, 10)
            try:
                invs_page = paginator.page(page)
            except PageNotAnInteger:
                invs_page = paginator.page(1)
            except EmptyPage:
                invs_page = paginator.page(paginator.num_pages)
            hosting_orders = HostingOrder.objects.filter(
                customer=cu.stripecustomer).filter(
                Q(subscription_id=None) | Q(subscription_id='')
            )
            stripe_chgs = []
            for ho in hosting_orders:
                stripe_chgs.append({ho.id: stripe.Charge.retrieve(ho.stripe_charge_id)})

            paginator_charges = Paginator(stripe_chgs, 10)
            try:
                invs_page_charges = paginator_charges.page(page)
            except PageNotAnInteger:
                invs_page_charges = paginator_charges.page(1)
            except EmptyPage:
                invs_page_charges = paginator_charges.page(paginator_charges.num_pages)
        else:
            try:
                invs = stripe.Invoice.list(
                    customer=self.request.user.stripecustomer.stripe_id,
                    count=100
                )
                paginator = Paginator(invs.data, 10)
                try:
                    invs_page = paginator.page(page)
                except PageNotAnInteger:
                    invs_page = paginator.page(1)
                except EmptyPage:
                    invs_page = paginator.page(paginator.num_pages)
                hosting_orders = HostingOrder.objects.filter(
                    customer=self.request.user.stripecustomer).filter(
                    Q(subscription_id=None) | Q(subscription_id='')
                )
                stripe_chgs = []
                for ho in hosting_orders:
                    stripe_chgs.append(
                        {ho: stripe.Charge.retrieve(ho.stripe_charge_id)})
                paginator_charges = Paginator(stripe_chgs, 10)
                try:
                    invs_page_charges = paginator_charges.page(page)
                except PageNotAnInteger:
                    invs_page_charges = paginator_charges.page(1)
                except EmptyPage:
                    invs_page_charges = paginator_charges.page(
                        paginator_charges.num_pages)
            except Exception as ex:
                logger.error(str(ex))
                invs_page = None
        context["invs"] = invs_page
        context["invs_charge"] = invs_page_charges
        return context

    @method_decorator(decorators)
    def get(self, request, *args, **kwargs):
        return super(InvoiceListView, self).get(request, *args, **kwargs)


class InvoiceDetailView(LoginRequiredMixin, DetailView):
    template_name = "hosting/invoice_detail.html"
    context_object_name = "invoice"
    login_url = reverse_lazy('hosting:login')
    permission_required = ['view_monthlyhostingbill']
    # model = MonthlyHostingBill

    def get_object(self, queryset=None):
        invoice_id = self.kwargs.get('invoice_id')
        try:
            invoice_obj = MonthlyHostingBill.objects.get(
                invoice_number=invoice_id
            )
            logger.debug("Found MHB for id {invoice_id}".format(
                invoice_id=invoice_id
            ))
            if self.request.user.has_perm(
                    self.permission_required[0], invoice_obj
            ) or self.request.user.email == settings.ADMIN_EMAIL:
                logger.debug("User has permission to invoice_obj")
            else:
                logger.error("User does not have permission to access")
                invoice_obj = None
        except MonthlyHostingBill.DoesNotExist as dne:
            logger.debug("MHB not found for id {invoice_id}".format(
                invoice_id=invoice_id
            ))
            invoice_obj = None
        return invoice_obj

    def get_context_data(self, **kwargs):
        # Get context
        context = super(InvoiceDetailView, self).get_context_data(**kwargs)
        obj = self.get_object()

        if obj is not None:
            vm_id = obj.get_vm_id()
            try:
                # Try to get vm details from database
                vm_detail = VMDetail.objects.get(vm_id=vm_id)
                context['vm'] = vm_detail.__dict__
                context['vm']['name'] = '{}-{}'.format(
                    context['vm']['configuration'], context['vm']['vm_id'])
                user_vat_country = obj.order.billing_address.country
                user_country_vat_rate = get_vat_rate_for_country(
                    user_vat_country)
                price, vat, vat_percent, discount = get_vm_price_for_given_vat(
                    cpu=context['vm']['cores'],
                    ssd_size=context['vm']['disk_size'],
                    memory=context['vm']['memory'],
                    pricing_name=(obj.order.vm_pricing.name
                                  if obj.order.vm_pricing else 'default'),
                    vat_rate=(
                        user_country_vat_rate * 100
                        if obj.order.vm_id >= settings.FIRST_VM_ID_AFTER_EU_VAT
                        else settings.PRE_EU_VAT_RATE
                    )
                )
                context['vm']["after_eu_vat_intro"] = (
                    True if obj.order.vm_id >= settings.FIRST_VM_ID_AFTER_EU_VAT
                    else False
                )
                context['vm']["price"] = price
                context['vm']["vat"] = vat
                context['vm']["vat_percent"] = vat_percent
                context['vm']["vat_country"] = user_vat_country
                context['vm']["discount"] = discount
                context['vm']["total_price"] = round(
                    price + vat - discount['amount'], 2)
            except VMDetail.DoesNotExist:
                # fallback to get it from the infrastructure
                try:
                    manager = OpenNebulaManager(
                        email=self.request.user.username,
                        password=self.request.user.password
                    )
                    vm = manager.get_vm(vm_id)
                    context['vm'] = VirtualMachineSerializer(vm).data
                    user_vat_country = obj.order.billing_address.country
                    user_country_vat_rate = get_vat_rate_for_country(
                        user_vat_country)
                    price, vat, vat_percent, discount = get_vm_price_for_given_vat(
                        cpu=context['vm']['cores'],
                        ssd_size=context['vm']['disk_size'],
                        memory=context['vm']['memory'],
                        pricing_name=(obj.order.vm_pricing.name
                                      if obj.order.vm_pricing else 'default'),
                        vat_rate=(
                            user_country_vat_rate * 100
                            if obj.order.vm_id >= settings.FIRST_VM_ID_AFTER_EU_VAT
                            else settings.PRE_EU_VAT_RATE
                        )
                    )
                    context['vm']["after_eu_vat_intro"] = (
                        True if obj.order.vm_id >= settings.FIRST_VM_ID_AFTER_EU_VAT
                        else False
                    )
                    context['vm']["price"] = price
                    context['vm']["vat"] = vat
                    context['vm']["vat_percent"] = vat_percent
                    context['vm']["vat_country"] = user_vat_country
                    context['vm']["discount"] = discount
                    context['vm']["total_price"] = round(
                        price + vat - discount['amount'], 2)
                except TypeError:
                    logger.error("Type error. Probably we "
                                 "came from a generic product. "
                                 "Invoice ID %s" % obj.invoice_id)
                except WrongIdError:
                    logger.error("WrongIdError while accessing "
                                 "invoice {}".format(obj.invoice_id))
                    messages.error(
                        self.request,
                        _('The VM you are looking for is unavailable at the '
                          'moment. Please contact Data Center Light support.')
                    )
                    self.kwargs['error'] = 'WrongIdError'
                    context['error'] = 'WrongIdError'
                    return context

            # add context params from monthly hosting bill
            context['period_start'] = obj.get_period_start()
            context['period_end'] = obj.get_period_end()
            context['paid_at'] = obj.paid_at
            context['total_in_chf'] = obj.total_in_chf()
            context['invoice_number'] = obj.invoice_number
            context['discount_on_stripe'] = obj.discount_in_chf()
            if obj.lines_data_count > 1:
                # special case, we pass the details of each of the line items
                context['line_items'] = obj.hostingbilllineitem_set.all()
            return context
        else:
            raise Http404

    @method_decorator(decorators)
    def get(self, request, *args, **kwargs):
        self.object = self.get_object()
        context = self.get_context_data(object=self.get_object())
        return self.render_to_response(context)


class OrdersHostingDeleteView(LoginRequiredMixin, DeleteView):
    login_url = reverse_lazy('hosting:login')
    success_url = reverse_lazy('hosting:orders')
    model = HostingOrder


class VirtualMachinesPlanListView(LoginRequiredMixin, ListView):
    template_name = "hosting/virtual_machines.html"
    login_url = reverse_lazy('hosting:login')
    context_object_name = "vms"
    paginate_by = 10
    ordering = '-id'

    def get_queryset(self):
        owner = self.request.user
        manager = OpenNebulaManager(email=owner.username,
                                    password=owner.password)
        try:
            queryset = manager.get_vms()
            serializer = VirtualMachineSerializer(queryset, many=True)
            return serializer.data
        except ConnectionRefusedError:
            messages.error(self.request,
                           'We could not load your VMs due to a backend connection \
                error. Please try again in a few minutes'
                           )

            self.kwargs['error'] = 'connection'
            return []

    def get_context_data(self, **kwargs):
        error = self.kwargs.get('error')
        if error is not None:
            print(error)
            context = {'error': 'connection'}
        else:
            context = super(ListView, self).get_context_data(**kwargs)
            if UserHostingKey.objects.filter(user=self.request.user).exists():
                context['show_create_ssh_key_msg'] = False
            else:
                context['show_create_ssh_key_msg'] = True
        return context


class CreateVirtualMachinesView(LoginRequiredMixin, View):
    template_name = "hosting/create_virtual_machine.html"
    login_url = reverse_lazy('hosting:login')

    def validate_cores(self, value):
        if (value > 48) or (value < 1):
            raise ValidationError(_('Invalid number of cores'))

    def validate_memory(self, value):
        if 'pid' in self.request.POST:
            try:
                plugin = DCLCalculatorPluginModel.objects.get(
                             id=self.request.POST['pid']
                         )
            except DCLCalculatorPluginModel.DoesNotExist as dne:
                logger.error(
                    str(dne) + " plugin_id: " + self.request.POST['pid']
                )
                raise ValidationError(_('Invalid calculator properties'))
            if plugin.enable_512mb_ram:
                if value % 1 == 0 or value == 0.5:
                    logger.debug(
                        "Given ram {value} is either 0.5 or a"
                        " whole number".format(value=value)
                    )
                    if (value > 200) or (value < 0.5):
                        raise ValidationError(_('Invalid RAM size'))
                else:
                    raise ValidationError(_('Invalid RAM size'))
            elif (value > 200) or (value < 1) or (value % 1 != 0):
                raise ValidationError(_('Invalid RAM size'))
        else:
            raise ValidationError(_('Invalid RAM size'))

    def validate_storage(self, value):
        if (value > 2000) or (value < 10):
            raise ValidationError(_('Invalid storage size'))

    @method_decorator(decorators)
    def get(self, request, *args, **kwargs):
        context = {
            'templates': VMTemplate.objects.all(),
            'cms_integration': get_cms_integration('default'),
        }
        return render(request, self.template_name, context)

    @method_decorator(decorators)
    def post(self, request):
        cores = request.POST.get('cpu')
        cores_field = forms.IntegerField(validators=[self.validate_cores])
        memory = request.POST.get('ram')
        memory_field = forms.FloatField(validators=[self.validate_memory])
        storage = request.POST.get('storage')
        storage_field = forms.IntegerField(validators=[self.validate_storage])
        template_id = int(request.POST.get('config'))
        pricing_name = request.POST.get('pricing_name')
        vm_pricing = VMPricing.get_vm_pricing_by_name(pricing_name)
        template = VMTemplate.objects.filter(
            opennebula_vm_template_id=template_id).first()
        template_data = VMTemplateSerializer(template).data

        if vm_pricing is None:
            vm_pricing_name_msg = _(
                "Incorrect pricing name. Please contact support"
                "{support_email}".format(
                    support_email=settings.DCL_SUPPORT_FROM_ADDRESS
                )
            )
            messages.add_message(
                self.request, messages.ERROR, vm_pricing_name_msg,
                extra_tags='pricing'
            )
            return redirect(CreateVirtualMachinesView.as_view())
        else:
            vm_pricing_name = vm_pricing.name

        try:
            cores = cores_field.clean(cores)
        except ValidationError as err:
            msg = '{} : {}.'.format(cores, str(err))
            messages.add_message(self.request, messages.ERROR, msg,
                                 extra_tags='cores')
            return redirect(CreateVirtualMachinesView.as_view())

        try:
            memory = memory_field.clean(memory)
        except ValidationError as err:
            msg = '{} : {}.'.format(memory, str(err))
            messages.add_message(self.request, messages.ERROR, msg,
                                 extra_tags='memory')
            return redirect(CreateVirtualMachinesView.as_view())

        try:
            storage = storage_field.clean(storage)
        except ValidationError as err:
            msg = '{} : {}.'.format(storage, str(err))
            messages.add_message(self.request, messages.ERROR, msg,
                                 extra_tags='storage')
            return redirect(CreateVirtualMachinesView.as_view())

        price, vat, vat_percent, discount = get_vm_price_with_vat(
            cpu=cores,
            memory=memory,
            ssd_size=storage,
            pricing_name=vm_pricing_name
        )

        specs = {
            'cpu': cores,
            'memory': memory,
            'disk_size': storage,
            'discount': discount,
            'price': price,
            'vat': vat,
            'vat_percent': vat_percent,
            'total_price': round(price + vat - discount['amount'], 2),
            'pricing_name': vm_pricing_name
        }

        request.session['specs'] = specs
        request.session['template'] = template_data
        return redirect(reverse('hosting:payment'))


class VirtualMachineView(LoginRequiredMixin, View):
    template_name = "hosting/virtual_machine_detail.html"
    login_url = reverse_lazy('hosting:login')

    def get_object(self):
        owner = self.request.user
        vm = None
        manager = OpenNebulaManager(
            email=owner.username,
            password=owner.password
        )
        vm_id = self.kwargs.get('pk')
        try:
            vm = manager.get_vm(vm_id)
            return vm
        except WrongIdError:
            messages.error(self.request,
                           _('We could not find the requested VM. Please \
                           contact Data Center Light Support.')
                           )
            return None
        except ConnectionRefusedError:
            messages.error(self.request,
                           'We could not load your VM due to a backend connection \
                error. Please try again in a few minutes'
                           )
            return None
        except Exception as error:
            logger.error(str(error))
            raise Http404()

    def get_success_url(self):
        final_url = reverse('hosting:virtual_machines')
        return final_url

    @method_decorator(decorators)
    def get(self, request, *args, **kwargs):
        vm = self.get_object()
        if vm is None:
            if self.request.is_ajax():
                storage = messages.get_messages(request)
                for m in storage:
                    pass
                storage.used = True
                return JsonResponse({'text': ugettext('Terminated')})
            else:
                return redirect(reverse('hosting:virtual_machines'))
        elif self.request.is_ajax():
            return HttpResponse()
        context = None
        try:
            serializer = VirtualMachineSerializer(vm)
            hosting_order = HostingOrder.objects.get(
                vm_id=serializer.data['vm_id']
            )
            inv_url = None
            if hosting_order.subscription_id:
                stripe_obj = stripe.Invoice.list(
                   subscription=hosting_order.subscription_id,
                   count=1
                )
                inv_url = stripe_obj.data[0].hosted_invoice_url
            elif hosting_order.stripe_charge_id:
                stripe_obj = stripe.Charge.retrieve(
                    hosting_order.stripe_charge_id
                )
                inv_url = stripe_obj.receipt_url
            context = {
                'virtual_machine': serializer.data,
                'order': hosting_order,
                'keys': UserHostingKey.objects.filter(user=request.user),
                'inv_url': inv_url
            }

        except Exception as ex:
            logger.debug("Exception generated {}".format(str(ex)))
            messages.error(self.request,
                           _('We could not find the requested VM. Please '
                             'contact Data Center Light Support.')
                           )
            return redirect(reverse('hosting:virtual_machines'))

        return render(request, self.template_name, context)

    @method_decorator(decorators)
    def post(self, request, *args, **kwargs):
        response = {'status': False}
        admin_email_body = {}
        owner = self.request.user
        vm = self.get_object()

        manager = OpenNebulaManager(
            email=owner.username,
            password=owner.password
        )
        try:
            vm_data = VirtualMachineSerializer(manager.get_vm(vm.id)).data
            vm_name = vm_data.get('name')
        except WrongIdError as wrong_id_err:
            logger.error(str(wrong_id_err))
            return redirect(reverse('hosting:virtual_machines'))

        # Cancel Stripe subscription
        stripe_utils = StripeUtils()
        hosting_order = None
        stripe_subscription_obj = None
        try:
            hosting_order = HostingOrder.objects.get(
                vm_id=vm.id
            )
            result = stripe_utils.unsubscribe_customer(
                subscription_id=hosting_order.subscription_id
            )
            stripe_subscription_obj = result.get('response_object')
            # Check if the subscription was canceled
            if (stripe_subscription_obj is None or
                    stripe_subscription_obj.status != 'canceled'):
                error_msg = result.get('error')
                logger.error(
                    'Error canceling subscription for {user} and vm id '
                    '{vm_id}'.format(user=owner.username, vm_id=vm.id)
                )
                logger.error(error_msg)
                admin_email_body['stripe_error_msg'] = error_msg
        except HostingOrder.DoesNotExist:
            error_msg = (
                "HostingOrder corresponding to vm_id={vm_id} does"
                "not exist. Hence, can not find subscription to "
                "cancel ".format(vm_id=vm.id)
            )
            logger.error(error_msg)
            admin_email_body['stripe_error_msg'] = error_msg

        terminated = manager.delete_vm(vm.id)

        if not terminated:
            logger.error(
                "manager.delete_vm returned False. Hence, error making "
                "xml-rpc call to delete vm failed."
            )
            response['text'] = str(_('Error terminating VM')) + str(vm.id)
        else:
            for t in range(settings.MAX_TIME_TO_WAIT_FOR_VM_TERMINATE):
                try:
                    manager.get_vm(vm.id)
                except WrongIdError:
                    logger.error(
                        "VM {} not found. So, its terminated.".format(vm.id)
                    )
                    response['status'] = True
                    response['text'] = ugettext('Terminated')
                    vm_detail_obj = VMDetail.objects.filter(
                        vm_id=vm.id
                    ).first()
                    vm_detail_obj.terminated_at = datetime.utcnow()
                    vm_detail_obj.save()
                except BaseException as base_exception:
                    logger.error(
                        "manager.get_vm({vm_id}) returned exception: "
                        "{details}.".format(
                            details=str(base_exception), vm_id=vm.id
                        )
                    )
                    break
                else:
                    logger.debug(
                        'Sleeping 2 seconds for terminate action on VM %s' %
                        vm.id
                    )
                    sleep(2)
            if not response['status']:
                response['text'] = str(_("VM terminate action timed out. "
                                         "Please contact "
                                         "support@datacenterlight.ch for "
                                         "further information."))
            context = {
                'vm_name': vm_name,
                'base_url': "{0}://{1}".format(
                    self.request.scheme, self.request.get_host()
                ),
                'page_header': _('Virtual Machine %(vm_name)s Cancelled') % {
                    'vm_name': vm_name
                }
            }
            email_data = {
                'subject': context['page_header'],
                'to': self.request.user.email,
                'context': context,
                'template_name': 'vm_canceled',
                'template_path': 'hosting/emails/',
                'from_address': settings.DCL_SUPPORT_FROM_ADDRESS,
            }
            email = BaseEmail(**email_data)
            email.send()
        admin_email_body.update(response)
        admin_email_body["customer_email"] = owner.email
        admin_email_body["customer_username"] = owner.username
        admin_email_body["VM_ID"] = vm.id
        admin_email_body["VM_created_at"] = (str(hosting_order.created_at) if
                                             hosting_order is not None
                                             else "unknown")
        content = ""
        total_amount = 0
        if stripe_subscription_obj:
            for line_item in stripe_subscription_obj["items"]["data"]:
                total_amount += (line_item["quantity"] *
                                 line_item.plan["amount"])
                content += "   %s => %s x %s => %s\n" % (
                    line_item.plan["name"], line_item["quantity"],
                    line_item.plan["amount"]/100,
                    (line_item["quantity"] * line_item.plan["amount"])/100
                )
        admin_email_body["subscription_amount"] = total_amount/100
        admin_email_body["subscription_detail"] = content
        admin_msg_sub = "VM and Subscription for VM {} and user: {}, {}".format(
            vm.id,
            owner.email, owner.username
        )
        email_to_admin_data = {
            'subject': ("Deleted " if response['status']
                        else "ERROR deleting ") + admin_msg_sub,
            'from_email': settings.DCL_SUPPORT_FROM_ADDRESS,
            'to': ['info@ungleich.ch'],
            'body': "\n".join(
                ["%s=%s" % (k, v) for (k, v) in admin_email_body.items()]),
        }
        send_plain_email_task.delay(email_to_admin_data)
        return JsonResponse(response)


class HostingBillListView(PermissionRequiredMixin, LoginRequiredMixin,
                          ListView):
    template_name = "hosting/bills.html"
    login_url = reverse_lazy('hosting:login')
    permission_required = ['view_hostingview']
    context_object_name = "users"
    model = StripeCustomer
    paginate_by = 10
    ordering = '-id'


class HostingBillDetailView(PermissionRequiredMixin, LoginRequiredMixin,
                            DetailView):
    template_name = "hosting/bill_detail.html"
    login_url = reverse_lazy('hosting:login')
    permission_required = ['view_hostingview']
    context_object_name = "bill"
    model = HostingBill

    def get_object(self, queryset=None):
        # Get HostingBill for primary key (Select from customer users)
        pk = self.kwargs['pk']
        object = HostingBill.objects.filter(customer__id=pk).first()
        if object is None:
            self.template_name = 'hosting/bill_error.html'
        return object

    def get_context_data(self, **kwargs):
        # Get context
        context = super(DetailView, self).get_context_data(**kwargs)

        owner = self.request.user
        manager = OpenNebulaManager(email=owner.username,
                                    password=owner.password)
        # Get vms
        queryset = manager.get_vms()
        vms = VirtualMachineSerializer(queryset, many=True).data
        # Set total price
        bill = context['bill']
        bill.total_price = 0.0
        for vm in vms:
            bill.total_price += vm['price']
        context['vms'] = vms
        return context


def forbidden_view(request, exception=None, reason=''):
    """
    Handle 403 error
    """
    logger.error(str(exception) if exception else None)
    logger.error('Reason = {reason}'.format(reason=reason))
    err_msg = _('There was an error processing your request. Please try '
                'again.')
    messages.add_message(request, messages.ERROR, err_msg)
    return HttpResponseRedirect(request.get_full_path())


class CheckUserVM(APIView):
    renderer_classes = (JSONRenderer, )

    def get(self, request):
        try:
            email = request.data['email']
            ip = request.data['ip']
            user = request.data['user']
            realm = request.data['realm']
            token = request.data['token']
            if realm != settings.READ_VM_REALM:
                return Response("User not allowed", 403)
            response = check_otp(user, realm, token)
            if response != 200:
                return Response('Invalid token', 403)
            manager = OpenNebulaManager(settings.OPENNEBULA_USERNAME,
                                        settings.OPENNEBULA_PASSWORD)
            # not the best way to lookup vms by ip
            # TODO: make this optimal
            vms = manager.get_vms()
            users_vms = [vm for vm in vms if vm.uname == email]
            if len(users_vms) == 0:
                return Response('No VM found with the given email address',
                                404)
            for vm in users_vms:
                for nic in vm.template.nics:
                    if hasattr(nic, 'ip6_global'):
                        if nic.ip6_global == ip:
                            return Response('success', 200)
                    elif hasattr(nic, 'ip'):
                        if nic.ip == ip:
                            return Response('success', 200)
            return Response('No VM found matching the ip address provided', 404)
        except KeyError:
            return Response('Not enough data provided', 400)