app.py

from flask import Flask, request, jsonify, redirect
import redis
import string
import random
import os
from urllib.parse import urlparse


app = Flask(__name__)
redis_host = os.environ.get('REDIS_HOST', 'localhost')
redis_port = int(os.environ.get('REDIS_PORT', 6379))
r = redis.Redis(host=redis_host, port=redis_port, decode_responses=True)

@app.route('/')
def index():
    return '🎉 URL Shortener is running! Try POSTing to /shorten'


@app.route('/shorten', methods=['POST'])
def shorten():
    data = request.get_json()
    original_url = data['url']

    # Extract domain from the URL
    parsed_url = urlparse(original_url)
    domain = parsed_url.netloc.replace(":", "_")  # replace ':' for ports, if any

    # Generate short ID
    short_id = ''.join(random.choices(string.ascii_letters + string.digits, k=6))
    
    # Use domain in the key
    redis_key = f"{domain}/{short_id}"
    r.set(redis_key, original_url)

    # Create full short URL including the domain
    short_url = request.host_url + redis_key
    return jsonify({'short_url': short_url})


@app.route('/<domain>/<short_id>')
def redirect_short_url_with_domain(domain, short_id):
    redis_key = f"{domain}/{short_id}"
    original_url = r.get(redis_key)

    if original_url:
        return redirect(original_url)
    else:
        return 'URL not found', 404

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=8080)