shellmfs.py

import os
import argparse
import readline
import glob
import getpass
from prettytable import PrettyTable
import humanize
import datetime

from libmfs import MatrixFs


def get_completions(text, state, options):
    opts = list(options) if not isinstance(options, (list, tuple)) else options
    matches = [x for x in opts if x.startswith(text)]
    return matches[state] if state < len(matches) else None


def print_help():
    print("Available commands:")
    print("  put [filename]         Upload a local file")
    print("  get [filename] [dest]  Download a file to dest or current dir")
    print("  del [filename]         Delete a remote file")
    print("  ls                     List remote files")
    print("  help                   Show this help")
    print("  exit                   Exit the shell")


def print_error(e):
    print(f"Error: {e}")


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--server_url", help="Matrix homeserver URL")
    parser.add_argument("--username", help="Username")
    parser.add_argument("--password", help="Password")
    parser.add_argument("--room_name", help="Room name")
    parser.add_argument("--encryption_key", default="", help="Encryption key")
    parser.add_argument("--max_file_size", type=int, default=1 * 1024 * 1024)
    parser.add_argument("--variance", type=float, default=0.10, help="Chunk size variance (e.g. 0.10 for ±10%)")
    args = parser.parse_args()

    if not args.server_url:
        args.server_url = input("Server URL: ")
    if not args.username:
        args.username = input("Username: ")
    if not args.password:
        args.password = getpass.getpass("Password: ")
    if not args.room_name:
        args.room_name = input("Room name: ")

    fs = MatrixFs(
        server_url=args.server_url,
        username=args.username,
        password=args.password,
        room_name=args.room_name,
        encryption_key=args.encryption_key,
        max_file_size=args.max_file_size,
        variance=args.variance,
    )

    def completer(text, state):
        buffer = readline.get_line_buffer()
        line = buffer.strip().split()

        if not line:
            return None

        cmd = line[0]
        if cmd == "put":
            return get_completions(text, state, glob.glob(text + "*"))
        elif cmd in {"get", "del", "delete"}:
            try:
                return get_completions(text, state, list(fs.list_files().keys()))
            except Exception:
                return None
        else:
            return get_completions(text, state, ["put", "get", "del", "ls", "exit", "help"])

    readline.set_completer(completer)
    readline.parse_and_bind("tab: complete")

    print_help()
    while True:
        try:
            cmd_line = input(f"MatrixFS@{args.username}:{args.server_url}> ").strip()
        except (EOFError, KeyboardInterrupt):
            print("\nExiting.")
            break

        if not cmd_line:
            continue

        parts = cmd_line.split()
        command = parts[0]

        if command == "put" and len(parts) == 2:
            print("Uploading...")
            try:
                fs.send_file(parts[1])
            except Exception as e:
                print_error(e)

        elif command == "get" and len(parts) >= 2:
            filename = parts[1]
            dest = parts[2] if len(parts) == 3 else os.path.join(".", filename)
            try:
                print("Downloading...")
                content = fs.find_file(filename)
                with open(dest, "wb") as f:
                    f.write(content)
                print(f"Downloaded to {dest}")
            except Exception as e:
                print_error(e)

        elif command in {"del", "delete"} and len(parts) == 2:
            try:
                fs.delete(parts[1])
                print(f"{parts[1]} deleted.")
            except Exception as e:
                print_error(e)

        elif command == "ls":
            try:
                files = fs.list_files()
                table = PrettyTable()
                table.field_names = [
                    "Filename",
                    "Size",
                    "# Chunks",
                    "Date Added",
                    "ID",
                ]
                table.border = True
                table.header = True
                for col in table.field_names:
                    table.align[col] = "l"

                for filename, info in files.items():
                    size = humanize.naturalsize(info.get("size", "N/A"), binary=False)
                    chunks = info.get("chunks", "N/A")
                    date = humanize.naturaltime(
                        datetime.datetime.fromtimestamp(int(info.get("date_added", 0)))
                    )
                    media_id = info.get("media_id", "N/A")
                    table.add_row([filename, size, chunks, date, media_id])

                print(table)
            except Exception as e:
                print_error(e)

        elif command in {"exit", "quit"}:
            break

        elif command == "help":
            print_help()

        else:
            print(f"Unknown command: {command}")


if __name__ == "__main__":
    main()