239 lines
9.1 KiB
Python
239 lines
9.1 KiB
Python
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.utils import timezone
|
|
from django.views.decorators.csrf import ensure_csrf_cookie
|
|
|
|
from apps.applets.models import Applet, UserApplet
|
|
from apps.applets.utils import applet_context
|
|
from apps.dashboard.forms import ExistingNoteItemForm, ItemForm
|
|
from apps.dashboard.models import Item, Note
|
|
from apps.lyric.models import PaymentMethod, Token, User, Wallet
|
|
|
|
|
|
APPLET_ORDER = ["wallet", "new-note", "my-notes", "username", "palette"]
|
|
UNLOCKED_PALETTES = frozenset([
|
|
"palette-default",
|
|
"palette-sepia",
|
|
"palette-oblivion-light",
|
|
"palette-monochrome-dark",
|
|
])
|
|
PALETTES = [
|
|
{"name": "palette-default", "label": "Earthman", "locked": False},
|
|
{"name": "palette-sepia", "label": "Sepia", "locked": False},
|
|
{"name": "palette-oblivion-light", "label": "Oblivion (Light)", "locked": False},
|
|
{"name": "palette-monochrome-dark", "label": "Monochrome (Dark)", "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_notes(user, limit=3):
|
|
return (
|
|
Note
|
|
.objects
|
|
.filter(Q(owner=user) | Q(shared_with=user))
|
|
.annotate(last_item=Max('item__id'))
|
|
.order_by('-last_item')
|
|
.distinct()[:limit]
|
|
)
|
|
|
|
def home_page(request):
|
|
context = {
|
|
"form": ItemForm(),
|
|
"palettes": PALETTES,
|
|
"page_class": "page-dashboard",
|
|
}
|
|
if request.user.is_authenticated:
|
|
context["applets"] = applet_context(request.user, "dashboard")
|
|
context["recent_notes"] = _recent_notes(request.user)
|
|
return render(request, "apps/dashboard/home.html", context)
|
|
|
|
def new_note(request):
|
|
form = ItemForm(data=request.POST)
|
|
if form.is_valid():
|
|
nunote = Note.objects.create()
|
|
if request.user.is_authenticated:
|
|
nunote.owner = request.user
|
|
nunote.save()
|
|
form.save(for_note=nunote)
|
|
return redirect(nunote)
|
|
else:
|
|
context = {
|
|
"form": form,
|
|
"palettes": PALETTES,
|
|
"page_class": "page-dashboard",
|
|
}
|
|
if request.user.is_authenticated:
|
|
context["applets"] = applet_context(request.user, "dashboard")
|
|
context["recent_notes"] = _recent_notes(request.user)
|
|
return render(request, "apps/dashboard/home.html", context)
|
|
|
|
def view_note(request, note_id):
|
|
our_note = Note.objects.get(id=note_id)
|
|
|
|
if our_note.owner:
|
|
if not request.user.is_authenticated:
|
|
return redirect("/")
|
|
if request.user != our_note.owner and request.user not in our_note.shared_with.all():
|
|
return HttpResponseForbidden()
|
|
|
|
form = ExistingNoteItemForm(for_note=our_note)
|
|
|
|
if request.method == "POST":
|
|
form = ExistingNoteItemForm(for_note=our_note, data=request.POST)
|
|
if form.is_valid():
|
|
form.save()
|
|
return redirect(our_note)
|
|
return render(request, "apps/dashboard/note.html", {"note": our_note, "form": form})
|
|
|
|
def my_notes(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_notes.html", {"owner": owner})
|
|
|
|
def share_note(request, note_id):
|
|
our_note = Note.objects.get(id=note_id)
|
|
try:
|
|
recipient = User.objects.get(email=request.POST["recipient"])
|
|
if recipient == request.user:
|
|
return redirect(our_note)
|
|
our_note.shared_with.add(recipient)
|
|
except User.DoesNotExist:
|
|
pass
|
|
messages.success(request, "An invite has been sent if that address is registered.")
|
|
return redirect(our_note)
|
|
|
|
@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"])
|
|
if "application/json" in request.headers.get("Accept", ""):
|
|
return JsonResponse({"palette": request.user.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.filter(context="dashboard"):
|
|
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, "dashboard"),
|
|
"palettes": PALETTES,
|
|
"form": ItemForm(),
|
|
"recent_notes": _recent_notes(request.user),
|
|
})
|
|
return redirect("home")
|
|
|
|
@login_required(login_url="/")
|
|
@ensure_csrf_cookie
|
|
def wallet(request):
|
|
return render(request, "apps/dashboard/wallet.html", {
|
|
"wallet": request.user.wallet,
|
|
"pass_token": request.user.tokens.filter(token_type=Token.PASS).first(),
|
|
"coin": request.user.tokens.filter(token_type=Token.COIN).first(),
|
|
"free_tokens": list(request.user.tokens.filter(
|
|
token_type=Token.FREE, expires_at__gt=timezone.now()
|
|
).order_by("expires_at")),
|
|
"tithe_tokens": list(request.user.tokens.filter(token_type=Token.TITHE)),
|
|
"free_count": request.user.tokens.filter(
|
|
token_type=Token.FREE, expires_at__gt=timezone.now()
|
|
).count(),
|
|
"tithe_count": request.user.tokens.filter(token_type=Token.TITHE).count(),
|
|
"applets": applet_context(request.user, "wallet"),
|
|
"page_class": "page-wallet",
|
|
})
|
|
|
|
|
|
@login_required(login_url="/")
|
|
def kit_bag(request):
|
|
tokens = list(request.user.tokens.all())
|
|
free_tokens = sorted(
|
|
[t for t in tokens if t.token_type == Token.FREE and t.expires_at and t.expires_at > timezone.now()],
|
|
key=lambda t: t.expires_at,
|
|
)
|
|
tithe_tokens = [t for t in tokens if t.token_type == Token.TITHE]
|
|
return render(request, "core/_partials/_kit_bag_panel.html", {
|
|
"equipped_deck": request.user.equipped_deck,
|
|
"equipped_trinket": request.user.equipped_trinket,
|
|
"free_token": free_tokens[0] if free_tokens else None,
|
|
"free_count": len(free_tokens),
|
|
"tithe_token": tithe_tokens[0] if tithe_tokens else None,
|
|
"tithe_count": len(tithe_tokens),
|
|
})
|
|
|
|
@login_required(login_url="/")
|
|
def toggle_wallet_applets(request):
|
|
checked = request.POST.getlist("applets")
|
|
for applet in Applet.objects.filter(context="wallet"):
|
|
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/wallet/_partials/_applets.html", {
|
|
"applets": applet_context(request.user, "wallet"),
|
|
"wallet": request.user.wallet,
|
|
"pass_token": request.user.tokens.filter(token_type=Token.PASS).first(),
|
|
"coin": request.user.tokens.filter(token_type=Token.COIN).first(),
|
|
"free_tokens": list(request.user.tokens.filter(token_type=Token.FREE)),
|
|
"tithe_tokens": list(request.user.tokens.filter(token_type=Token.TITHE)),
|
|
})
|
|
return redirect("wallet")
|
|
|
|
@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})
|