share-host/share-via-http.py

#!/usr/bin/env python

import os
import argparse
import base64
import http.server
import socket
import socketserver
import qrcode
import random
import string


def get_current_ip_address():
    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    s.connect(("8.8.8.8", 80))
    ip_address = s.getsockname()[0]
    s.close()
    return ip_address


# Function to generate a default username
def generate_default_username():
    return "".join(random.choices(string.ascii_lowercase + string.digits, k=8))


# Function to generate a default password
def generate_default_password():
    return "".join(random.choices(string.ascii_letters + string.digits, k=12))


# Define the command-line arguments
parser = argparse.ArgumentParser(
    description="Start a simple HTTP server with basic authentication."
)

parser.add_argument(
    "--port",
    type=int,
    default=8000,
    help="Port to serve the directory over. Defaults to 8000.",
)

parser.add_argument(
    "--username",
    default=generate_default_username(),
    help="Username for basic authentication. Generates one if not set",
)

parser.add_argument(
    "--password",
    default=generate_default_password(),
    help="Password for basic authentication. Generates one if not set",
)

parser.add_argument(
    "--directory",
    default=os.getcwd(),
    help="Directory to serve. Defaults to the current directory.",
)

args = parser.parse_args()

# Expand the directory path (e.g., if it contains ~)
expanded_directory = os.path.expanduser(args.directory)

# Change the working directory to serve the specified directory
os.chdir(expanded_directory)


# Define a request handler with basic authentication
class AuthHandler(http.server.SimpleHTTPRequestHandler):
    def __init__(self, username, passsword, *args, **kwargs):
        encoded_bytes = base64.b64encode(f"{username}:{passsword}".encode("utf-8"))
        # Convert bytes back to string
        self.encoded_str = encoded_bytes.decode("utf-8")
        super().__init__(*args, **kwargs)

    def do_AUTHHEAD(self):
        self.send_response(401)
        self.send_header("WWW-Authenticate", 'Basic realm="Server Access"')
        self.send_header("Content-type", "text/html")
        self.end_headers()

    def do_GET(self):
        auth_header = self.headers.get("Authorization")
        if auth_header is None or not self.check_auth(auth_header):
            self.do_AUTHHEAD()
            self.wfile.write(b"Not authenticated")
        else:
            super().do_GET()

    def check_auth(self, auth_header):
        encoded_credentials = auth_header.split()[1]
        return encoded_credentials == self.encoded_str
        # decoded_credentials = base64.b64decode(encoded_credentials).decode("utf-8")
        # username, password = decoded_credentials.split(":")
        # return username == args.username and password == args.password


def save_qr_code(url):
    # Generate the QR code
    qr = qrcode.QRCode(
        version=1,
        error_correction=qrcode.constants.ERROR_CORRECT_L,
        box_size=10,
        border=4,
    )
    qr.add_data(url)
    qr.make(fit=True)
    # Create an image from the QR Code instance
    img = qr.make_image(fill_color="black", back_color="white").convert("RGB")
    # Save the image
    img.save("qrcode.png")


from functools import partial

# Set up the server with the AuthHandler and the defined port
auth_handler = partial(AuthHandler, args.username, args.password)
with socketserver.TCPServer(("", args.port), auth_handler) as httpd:
    print("qrcode.png is saved in the current directory.")
    ip_address = get_current_ip_address()
    print(
        f"http://{args.username}:{args.password}@{ip_address}:{args.port}/ < the page"
    )
    print(
        f"http://{args.username}:{args.password}@{ip_address}:{args.port}/qrcode.png < qrcode"
    )
    save_qr_code(f"http://{args.username}:{args.password}@{ip_address}:{args.port}/")
    httpd.serve_forever()