app.py

import requests, re
from threading import Lock
from flask_limiter import Limiter
from platforms.tiktok import TikTokv1
from platforms.tiktok import TikTokv2
from platforms.facebook import Facebook
from datetime import datetime, timedelta
from platforms.instagram import Instagram
from flask_limiter.util import get_remote_address
from flask import Flask, render_template, request, jsonify

app = Flask(__name__, static_folder='website/static', template_folder='website')

limiter = Limiter(
    get_remote_address, 
    app=app,
    default_limits=[]
)

RATE_LIMIT = 2
RATE_LIMIT_PERIOD = timedelta(minutes=15)

lock = Lock()
# (https://github.com/devfemibadmus/webmedia/issues/1#issue-2725260727) instagram = None
application = app
request_timestamps = []

class Validator:
    tiktok_pattern = r'tiktok\.com/.*/'
    instag_pattern = r'instagram\.com/(p|reel|tv)/([A-Za-z0-9_-]+)/?'
    facebook_pattern = r'(facebook\.com/.*/|fb\.watch/.*/)'
    @staticmethod
    def validate(url):
        if re.search(Validator.tiktok_pattern, url):
            return "TikTok", url
        if re.search(Validator.facebook_pattern, url):
            return "Facebook", url
        insta_match = re.search(Validator.instag_pattern, url)
        if insta_match:
            return "Instagram", insta_match.group(2)
        
        return "Invalid URL", None

@app.route('/webmedia/api/', methods=['POST', 'GET'])
@app.route('/api/', methods=['POST', 'GET'])
@limiter.limit("300 per minute")
def api():
    print('remote address: ', request.remote_addr)
    url = request.form.get('url') if request.method == 'POST' else request.args.get('url')
    cut = request.form.get('cut') if request.method == 'POST' else request.args.get('cut')
    if not url:
        return jsonify({'error': True, 'message': 'URL is required'}), 400
    source, item_id = Validator.validate(url)
    
    if source == "TikTok":
        tiktok = TikTokv2(url, cut)
        data, status = tiktok.getData()
        if status == 200:
            return jsonify({'success': True, 'data': data}), 200
        return jsonify({'error': True, 'message': data['message'], 'error_message': data['error_message']}), status

    elif source == "Facebook":
        facebook = Facebook(url, cut)
        data, status = facebook.getVideo()
        if status == 200:
            return jsonify({'success': True, 'data': data}), 200
        return jsonify({'error': True, 'message': data['message'], 'error_message': data['error_message']}), status
    
    elif source == "Instagram":
        # global instagram
        # with lock:
        #    if not instagram:
        instagram = Instagram()
        if item_id:
            data, status = instagram.getData(item_id, cut)
            if status == 200:
                return jsonify({'success': True, 'data': data}), 200
            return jsonify({'error': True, 'message': data['message'], 'error_message': data['error_message']}), status
        else:
            return jsonify({'error': True, 'message': 'Invalid Instagram video URL'}), 400
        
    return jsonify({'error': True, 'message': 'Unsupported URL'}), 400

# @app.route('/webmedia/sleep', methods=['GET'])
# @app.route('/sleep', methods=['GET'])
def sleep():
    global request_timestamps, instagram
    now = datetime.now()
    request_timestamps = [timestamp for timestamp in request_timestamps if now - timestamp < RATE_LIMIT_PERIOD]
    if len(request_timestamps) >= RATE_LIMIT:
        return jsonify(error=True, message="Too Many Requests", details="You have exceeded the rate limit. Please wait 15 minutes and try again."), 429
    request_timestamps.append(now)
    try:
        if instagram:
            instagram.close()
            instagram = None
            return jsonify(success=True, message="Instagram instance closed and put to sleep."), 200
        else:
            return jsonify(error=True, message="Instagram instance already close.", details="No active Instagram instance to close."), 404
    except Exception as e:
        print(f"An error occurred: {e}")
        return jsonify(error=True, message="An error occurred while closing Instagram instance", details=str(e)), 500

@app.errorhandler(429)
def ratelimit_error(e):
    return jsonify({
        'error': True,
        'message': 'Too Many Requests',
        'details': 'Rate limit exceeded. Please wait a minute and try again.'
    }), 429

@app.route('/', methods=['GET'])
def home():
    return render_template("home.html")

@app.route('/<path:path>')
def catch_all(path):
    return render_template("home.html")

if __name__ == '__main__':
    app.run(debug=True)