Loopbot/optin.py at main · alexdnaz/Loopbot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
#!/usr/bin/env python3
"""
optin.py

A simple Flask app to collect email opt-ins with double opt-in confirmation.
Subscribers are added to recipients.csv upon confirmation.
"""

import os
from pathlib import Path

# Optionally load a .env file if python-dotenv is installed
try:
    from dotenv import load_dotenv, find_dotenv
except ImportError:
    load_dotenv = find_dotenv = None

if load_dotenv:
    env_path = Path(__file__).parent / '.env'
    # Prefer .env next to this script, else any .env in parent dirs
    if env_path.exists():
        load_dotenv(env_path)
    else:
        load_dotenv(find_dotenv())
import uuid
import sqlite3
import smtplib
from datetime import datetime
from email.message import EmailMessage
from flask import Flask, request, redirect, url_for, render_template_string
import sys

# Configuration from environment
SMTP_HOST = os.getenv('SMTP_HOST')
SMTP_PORT = int(os.getenv('SMTP_PORT', 587))
SMTP_USER = os.getenv('SMTP_USER')
SMTP_PASS = os.getenv('SMTP_PASS')
EMAIL_FROM = os.getenv('EMAIL_FROM')
BASE_URL = os.getenv('BASE_URL', 'http://localhost:5000')
INVITE_LINK = os.getenv('INVITE_LINK', '')
SERVER_NAME = os.getenv('SERVER_NAME', 'Discord Server')


# Setup DB for pending tokens
conn = sqlite3.connect('optin.db', check_same_thread=False)
cur = conn.cursor()
cur.execute(
    'CREATE TABLE IF NOT EXISTS pending (token TEXT PRIMARY KEY, name TEXT, email TEXT, created TIMESTAMP)'
)
conn.commit()

# Flask app
app = Flask(__name__)
app.secret_key = os.getenv('SECRET_KEY', os.urandom(16))

SUBSCRIBE_FORM = '''
<!doctype html>
<title>Get Your Free Guide & Join Our Discord</title>
<h2>Claim Your Free Guide to LoopBot + Community Invite</h2>
<p>Enter your name and email below to receive our free PDF guide on mastering LoopBot, plus an invite to our Discord server.</p>
<form method=post>
  <label>Name: <input type=text name=name required></label><br>
  <label>Email: <input type=email name=email required></label><br>
  <input type=submit value='Send Me the Guide & Invite'>
</form>
'''

@app.route('/', methods=['GET', 'POST'])
def subscribe():
    if request.method == 'POST':
        name = request.form.get('name').strip()
        email = request.form.get('email').strip().lower()
        token = str(uuid.uuid4())
        now = datetime.utcnow().isoformat()
        cur.execute(
            'INSERT INTO pending(token,name,email,created) VALUES (?,?,?,?)',
            (token, name, email, now)
        )
        conn.commit()
        try:
            send_confirmation(email, name, token)
        except Exception as e:
            print(f"Error sending confirmation email: {e}", file=sys.stderr)
            return '<h3>Oops—could not send confirmation email. Please try again later.</h3>', 500
        return redirect(url_for('thanks'))
    return render_template_string(SUBSCRIBE_FORM)

@app.route('/confirm/<token>')
def confirm(token):
    cur.execute('SELECT name,email FROM pending WHERE token=?', (token,))
    row = cur.fetchone()
    if not row:
        return '<h3>Invalid or expired token.</h3>', 400
    name, email = row
    # append to CSV if not exists
    csv_path = 'recipients.csv'
    seen = set()
    if os.path.exists(csv_path):
        with open(csv_path) as f:
            for line in f:
                parts = line.strip().split(',', 1)
                if len(parts) == 2:
                    seen.add(parts[1].lower())
    if email not in seen:
        with open(csv_path, 'a') as f:
            f.write(f'{name},{email}\n')
    # send the actual invite link
    try:
        send_invite(name, email)
    except Exception:
        pass
    # cleanup pending token
    cur.execute('DELETE FROM pending WHERE token=?', (token,))
    conn.commit()
    return f'<h3>Thank you {name}! A Discord invite has been sent to {email}.</h3>'

def send_confirmation(email, name, token):
    confirm_url = f"{BASE_URL}{url_for('confirm', token=token)}"
    subject = f"Confirm your subscription to our Discord"
    body = (
        f"Hi {name},\n\n"
        "Thanks for subscribing to join our Discord community!\n"
        f"Please confirm your email address by clicking the link below:\n{confirm_url}\n\n"
        "Once confirmed, we’ll send you the invite link.\n\n"
        "If you didn’t request this, you can ignore this message.\n"
    )
    msg = EmailMessage()
    msg['Subject'] = subject
    msg['From'] = EMAIL_FROM
    msg['To'] = email
    msg.set_content(body)

    with smtplib.SMTP(SMTP_HOST, SMTP_PORT) as s:
        s.starttls()
        s.login(SMTP_USER, SMTP_PASS)
        s.send_message(msg)

def send_invite(name, email):
    """Send the Discord invite link after confirmation."""
    subject = f"Your invite to {SERVER_NAME} on Discord"
    body = (
        f"Hi {name},\n\n"
        "Thanks for confirming your subscription!\n\n"
        f"Here’s your invite link to join {SERVER_NAME}:\n{INVITE_LINK}\n\n"
        f"You can also download your free guide here: {BASE_URL}/static/lead_magnet.pdf\n\n"
        "See you inside!\n"
    )
    msg = EmailMessage()
    msg['Subject'] = subject
    msg['From'] = EMAIL_FROM
    msg['To'] = email
    msg.set_content(body)
    # Attach the lead-magnet PDF if present
    pdf_path = Path(__file__).parent / 'static' / 'lead_magnet.pdf'
    if pdf_path.exists():
        with open(pdf_path, 'rb') as pdf_file:
            msg.add_attachment(
                pdf_file.read(),
                maintype='application',
                subtype='pdf',
                filename=pdf_path.name,
            )
    else:
        print(f"Warning: PDF not found at {pdf_path}", file=sys.stderr)

    with smtplib.SMTP(SMTP_HOST, SMTP_PORT) as s:
        s.starttls()
        s.login(SMTP_USER, SMTP_PASS)
        s.send_message(msg)

if __name__ == '__main__':
    # Ensure required environment variables
    missing = [v for v in ('SMTP_HOST','SMTP_USER','SMTP_PASS','EMAIL_FROM','INVITE_LINK') if not os.getenv(v)]
    if missing:
        raise RuntimeError(f"Missing environment variables: {', '.join(missing)}")
    # preload CSV header if missing
    if not os.path.exists('recipients.csv'):
        with open('recipients.csv', 'w') as f:
            f.write('name,email\n')
    port = int(os.getenv('PORT', '5000'))
    app.run(host='0.0.0.0', port=port)


@app.route('/thanks')
def thanks():
    return (
        '<!doctype html>'
        '<title>Thank You!</title>'
        '<h3>Thank you! A confirmation email has been sent. '
        'Please check your inbox (or spam folder) to confirm and receive your PDF & Discord invite.</h3>'
    )