From eb137b85d32465b15f43fe442bc4ade91c2f5323 Mon Sep 17 00:00:00 2001
From: Nathan Woodburn <github@nathan.woodburn.au>
Date: Thu, 24 Aug 2023 16:53:39 +1000
Subject: [PATCH] main: Added email test for stripe

---
 README.md      | 13 +++++++++++++
 master/main.py | 35 +++++++++++++++++++++++++++--------
 2 files changed, 40 insertions(+), 8 deletions(-)

diff --git a/README.md b/README.md
index e00aad3..a3364ac 100644
--- a/README.md
+++ b/README.md
@@ -50,6 +50,19 @@ Then to start the master api server
 screen -dmS hnshosting-master python3 main.py
 ```
 
+## Stripe webhook
+Create a new webhook endpoint on stripe and set the url to https://master-server-domain:5000/stripe-webhook (note: you need to add a https proxy to your master server if you want to use stripe)
+Add these environment variables to your master server
+```yaml
+STRIPE_SECRET: your-stripe-secret-key
+STRIPE_ENDPOINT_SECRET: your-stripe-endpoint-secret
+SMTP_HOST: smtp-server
+SMTP_PORT: smtp-port
+SMTP_USER: smtp-user
+SMTP_PASS: smtp-pass
+SMTP_FROM: smtp-from <Optional>
+```
+
 
 ## Worker server install
 
diff --git a/master/main.py b/master/main.py
index 8a5cad0..f314b0b 100644
--- a/master/main.py
+++ b/master/main.py
@@ -2,7 +2,8 @@ from flask import Flask, request, jsonify
 import dotenv
 import os
 import requests
-import stripe
+import stripe # For stripe payments
+import smtplib, ssl # For sending emails
 
 dotenv.load_dotenv()
 
@@ -236,8 +237,6 @@ def tlsa():
 
 @app.route('/stripe', methods=['POST'])
 def stripeapi():
-    print(request.json, flush=True)
-    print(request.headers, flush=True)
     payload = request.data
     stripe.api_key = os.getenv('STRIPE_SECRET')
     endpoint_secret = os.getenv('STRIPE_ENDPOINT_SECRET')
@@ -255,12 +254,32 @@ def stripeapi():
 
     # Handle the event
     if event.type == 'payment_intent.succeeded':
-        payment_intent = event.data.object # contains a stripe.PaymentIntent
+        payment_intent = event.data.object
+        # Get email
+        email = payment_intent['receipt_email']
+        # Create licence key
+        licence_key = os.urandom(16).hex()
+        # Add licence key to file
+        key_file = open('/data/licence_key.txt', 'a')
+        key_file.write(licence_key + '\n')
+        key_file.close()
+        # Send email
+        host = os.getenv('SMTP_HOST')
+        port = os.getenv('SMTP_PORT')
+        user = os.getenv('SMTP_USER')
+        password = os.getenv('SMTP_PASS')
+        from_email = os.getenv('SMTP_FROM')
+        if from_email == None:
+            from_email = user
+        
+        context = ssl.create_default_context()
+        with smtplib.SMTP_SSL(host, port, context=context) as server:
+            server.login(user, password)
+            server.sendmail(from_email, email, "Your licence key is: " + licence_key)
+
+
+
         print('PaymentIntent was successful!', flush=True)
-    elif event.type == 'payment_method.attached':
-        payment_method = event.data.object # contains a stripe.PaymentMethod
-        print('PaymentMethod was attached to a Customer!', flush=True)
-    # ... handle other event types
     else:
         print('Unhandled event type {}'.format(event.type))
     return jsonify({'success': 'true'})