feat: 🚧 Create base page for roomID selection

main.py

@@ -1,9 +1,8 @@
 import requests
 import os
-from dotenv import load_dotenv
 from flask import Flask, request, render_template
 from flask_bootstrap import Bootstrap5
-from spotifySearch import searchSpotify
+from utils.spotifyAPI import searchSpotify
 
 import sqlite3
 
@@ -11,11 +10,24 @@ import sqlite3
 app = Flask(__name__)
 bootstrap = Bootstrap5(app)
 
-database = sqlite3.connect("database.db")
+database = sqlite3.connect("database.db", check_same_thread=False)
+cursor = database.cursor()
+cursor.execute("CREATE TABLE IF NOT EXISTS rooms (roomid TEXT PRIMARY KEY, spotify_id TEXT)")
 
-@app.route("/", methods=['GET'])
+@app.route("/", methods=['GET', 'POST'])
 def main():
-    return "Hello, World!"
+    if request.method == 'POST':
+        roomid = request.form.get('roomID')
+        if len(roomid) != 8:
+            return render_template("index.html", response="Invalid room id", comment="A room ID should be composed of number and have a lenght of 8 characters", type="error", roomid=roomid)
+        roomSearch = cursor.execute("SELECT * FROM rooms WHERE roomid = ?", (roomid,)).fetchone()
+        if roomSearch:
+            return render_template("index.html", response="We found your collaborative playlist", comment="It's great news, click the button below to access it.", type="success", roomid=roomid)
+        else:
+            cursor.execute("INSERT INTO rooms (roomid, spotify_id) VALUES (?, ?)", (roomid, "test"))
+            database.commit()
+            return render_template("index.html", response="We couldn't find your collaborative playlist", comment="Please check the room id and try again or create one.", type="error", roomid=roomid)
+    return render_template("index.html")
 
 @app.route('/search/<string:roomid>', methods=['GET', 'POST'])
 def main_app(roomid):
@@ -26,7 +38,7 @@ def main_app(roomid):
         return render_template("found.html", query=search, tracks=result, roomid=roomid)
 
     # otherwise handle the GET request
-    return render_template("index.html")
+    return render_template("search.html")
 
 @app.route('/add/<string:roomid>/<string:trackid>', methods=['GET'])
 def add_to_playlist(roomid, trackid):
@@ -46,4 +58,10 @@ def add_to_playlist(roomid, trackid):
 
 
 if __name__ == '__main__':
-    app.run(debug=True, port="3000", host="0.0.0.0")
+    if not os.getenv("client_id") or not os.getenv("client_secret") or not os.getenv("n8n_webhook"):
+        print("Please provide client_id, client_secret and n8n_webhook in the .env file")
+        exit(1)
+    if os.getenv("enviroment") != "production":
+        app.run(debug=True, port="3000", host="0.0.0.0")
+    else:
+        app.run(debug=False, port="3000", host="0.0.0.0")