import stripe from django.conf import settings from django.contrib import messages from django.contrib.auth.decorators import login_required from django.db.models import Max, Q from django.http import HttpResponse, HttpResponseForbidden, JsonResponse from django.shortcuts import redirect, render from django.views.decorators.csrf import ensure_csrf_cookie from apps.applets.models import Applet, UserApplet from apps.dashboard.forms import ExistingListItemForm, ItemForm from apps.dashboard.models import Item, List from apps.lyric.models import PaymentMethod, Token, User, Wallet APPLET_ORDER = ["wallet", "new-list", "my-lists", "username", "palette"] UNLOCKED_PALETTES = frozenset(["palette-default"]) PALETTES = [ {"name": "palette-default", "label": "Earthman", "locked": False}, {"name": "palette-nirvana", "label": "Nirvana", "locked": True}, {"name": "palette-sheol", "label": "Sheol", "locked": True}, {"name": "palette-inferno", "label": "Inferno", "locked": True}, {"name": "palette-terrestre", "label": "Terrestre", "locked": True}, {"name": "palette-celestia", "label": "Celestia", "locked": True}, ] def _recent_lists(user, limit=3): return ( List .objects .filter(Q(owner=user) | Q(shared_with=user)) .annotate(last_item=Max('item__id')) .order_by('-last_item') .distinct()[:limit] ) def _applet_context(user): ua_map = {ua.applet_id: ua.visible for ua in user.user_applets.all()} applets = {a.slug: a for a in Applet.objects.all()} return [ {"applet": applets[slug], "visible": ua_map.get(applets[slug].pk, applets[slug].default_visible)} for slug in APPLET_ORDER if slug in applets ] def home_page(request): context = { "form": ItemForm(), "palettes": PALETTES, "page_class": "page-dashboard", } if request.user.is_authenticated: context["applets"] = _applet_context(request.user) context["recent_lists"] = _recent_lists(request.user) return render(request, "apps/dashboard/home.html", context) def new_list(request): form = ItemForm(data=request.POST) if form.is_valid(): nulist = List.objects.create() if request.user.is_authenticated: nulist.owner = request.user nulist.save() form.save(for_list=nulist) return redirect(nulist) else: context = { "form": form, "palettes": PALETTES, "page_class": "page-dashboard", } if request.user.is_authenticated: context["applets"] = _applet_context(request.user) context["recent_lists"] = _recent_lists(request.user) return render(request, "apps/dashboard/home.html", context) def view_list(request, list_id): our_list = List.objects.get(id=list_id) if our_list.owner: if not request.user.is_authenticated: return redirect("/") if request.user != our_list.owner and request.user not in our_list.shared_with.all(): return HttpResponseForbidden() form = ExistingListItemForm(for_list=our_list) if request.method == "POST": form = ExistingListItemForm(for_list=our_list, data=request.POST) if form.is_valid(): form.save() return redirect(our_list) return render(request, "apps/dashboard/list.html", {"list": our_list, "form": form}) def my_lists(request, user_id): owner = User.objects.get(id=user_id) if not request.user.is_authenticated: return redirect("/") if request.user.id != owner.id: return HttpResponseForbidden() return render(request, "apps/dashboard/my_lists.html", {"owner": owner}) def share_list(request, list_id): our_list = List.objects.get(id=list_id) try: recipient = User.objects.get(email=request.POST["recipient"]) if recipient == request.user: return redirect(our_list) our_list.shared_with.add(recipient) except User.DoesNotExist: pass messages.success(request, "An invite has been sent if that address is registered.") return redirect(our_list) @login_required(login_url="/") def set_palette(request): if request.method == "POST": palette = request.POST.get("palette", "") if palette in UNLOCKED_PALETTES: request.user.palette = palette request.user.save(update_fields=["palette"]) return redirect("home") @login_required(login_url="/") def set_profile(request): if request.method == "POST": username = request.POST.get("username", "") request.user.username = username request.user.save(update_fields=["username"]) return redirect("/") @login_required(login_url="/") def toggle_applets(request): checked = request.POST.getlist("applets") for applet in Applet.objects.all(): UserApplet.objects.update_or_create( user=request.user, applet=applet, defaults={"visible": applet.slug in checked}, ) if request.headers.get("HX-Request"): return render(request, "apps/dashboard/_partials/_applets.html", { "applets": _applet_context(request.user), "palettes": PALETTES, "form": ItemForm(), "recent_lists": _recent_lists(request.user), }) return redirect("home") @login_required(login_url="/") @ensure_csrf_cookie def wallet(request): wallet = request.user.wallet coin = request.user.tokens.filter(token_type=Token.COIN).first() free_tokens = list(request.user.tokens.filter(token_type=Token.FREE)) return render(request, "apps/dashboard/wallet.html", { "wallet": wallet, "coin": coin, "free_tokens": free_tokens, }) @login_required(login_url="/") def setup_intent(request): stripe.api_key = settings.STRIPE_SECRET_KEY user = request.user if not user.stripe_customer_id: customer = stripe.Customer.create(email=user.email) user.stripe_customer_id = customer.id user.save(update_fields=["stripe_customer_id"]) intent = stripe.SetupIntent.create(customer=user.stripe_customer_id) return JsonResponse({ "client_secret": intent.client_secret, "publishable_key": settings.STRIPE_PUBLISHABLE_KEY, }) @login_required(login_url="/") def save_payment_method(request): stripe.api_key = settings.STRIPE_SECRET_KEY pm_id = request.POST.get("payment_method_id") pm = stripe.PaymentMethod.retrieve(pm_id) stripe.PaymentMethod.attach(pm_id, customer=request.user.stripe_customer_id) PaymentMethod.objects.create( user=request.user, stripe_pm_id=pm_id, last4=pm.card.last4, brand=pm.card.brand, ) return JsonResponse({"last4": pm.card.last4, "brand": pm.card.brand})