shdir 0.1.0__tar.gz

shdir-0.1.0/PKG-INFO

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: shdir
+Version: 0.1.0
+Summary: Share your files in local network with secure connection
+License: GNU GPL v3
+Author: Rogov Anton
+Author-email: akldrix@gmail.com
+Requires-Python: >=3.15
+Classifier: License :: Other/Proprietary License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.15

shdir-0.1.0/pyproject.toml

@@ -0,0 +1,16 @@
+[project]
+name = "shdir"
+version = "0.1.0"
+description = "Share your files in local network with secure connection"
+authors = [{ name = "Rogov Anton", email = "akldrix@gmail.com" }]
+license = "GNU GPL v3"
+requires-python = ">=3.15"
+dependencies = []
+
+
+[build-system]
+requires = ["poetry-core>=2.0.0,<3.0.0"]
+build-backend = "poetry.core.masonry.api"
+
+[tool.poetry.scripts]
+APPLICATION-NAME = 'shdir.app:main'

shdir-0.1.0/src/shdir/.gitignore

@@ -0,0 +1,220 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[codz]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#   Usually these files are written by a python script from a template
+#   before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py.cover
+*.lcov
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+# Pipfile.lock
+
+# uv
+#   Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+# uv.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+# poetry.lock
+# poetry.toml
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#   pdm recommends including project-wide configuration in pdm.toml, but excluding .pdm-python.
+#   https://pdm-project.org/en/latest/usage/project/#working-with-version-control
+# pdm.lock
+# pdm.toml
+.pdm-python
+.pdm-build/
+
+# pixi
+#   Similar to Pipfile.lock, it is generally recommended to include pixi.lock in version control.
+# pixi.lock
+#   Pixi creates a virtual environment in the .pixi directory, just like venv module creates one
+#   in the .venv directory. It is recommended not to include this directory in version control.
+.pixi/*
+!.pixi/config.toml
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule*
+celerybeat.pid
+
+# Redis
+*.rdb
+*.aof
+*.pid
+
+# RabbitMQ
+mnesia/
+rabbitmq/
+rabbitmq-data/
+
+# ActiveMQ
+activemq-data/
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.envrc
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs/Zensical documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#   JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#   be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#   and can be added to the global gitignore or merged into this file.  For a more nuclear
+#   option (not recommended) you can uncomment the following to ignore the entire idea folder.
+# .idea/
+
+# Abstra
+#   Abstra is an AI-powered process automation framework.
+#   Ignore directories containing user credentials, local state, and settings.
+#   Learn more at https://abstra.io/docs
+.abstra/
+
+# Visual Studio Code
+#   Visual Studio Code specific template is maintained in a separate VisualStudioCode.gitignore that
+#   can be found at https://github.com/github/gitignore/blob/main/Global/VisualStudioCode.gitignore
+#   and can be added to the global gitignore or merged into this file. However, if you prefer, you
+#   could uncomment the following to ignore the entire vscode folder
+# .vscode/
+# Temporary file for partial code execution
+tempCodeRunnerFile.py
+
+# Ruff stuff:
+.ruff_cache/
+
+# PyPI configuration file
+.pypirc
+
+# Marimo
+marimo/_static/
+marimo/_lsp/
+__marimo__/
+
+# Streamlit
+.streamlit/secrets.toml

shdir-0.1.0/src/shdir/app.py

@@ -0,0 +1,280 @@
+import io
+import os
+import secrets
+import shutil
+import urllib
+import zipfile
+from contextlib import asynccontextmanager
+from pathlib import Path
+
+import uvicorn
+from fastapi import (
+    FastAPI,
+    Form,
+    HTTPException,
+    Request,
+    UploadFile,
+)
+from fastapi.responses import (
+    FileResponse,
+    HTMLResponse,
+    RedirectResponse,
+    StreamingResponse,
+)
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
+
+from cli import args, make_qr
+from utils.colors import BLUE, CYAN, RESET
+from utils.logs import log_config, logger
+from utils.security import temp_tls_cert
+
+COOKIE_NAME = "shdir_session"
+IS_AUTH_ENABLED = args.auth
+IS_HTTPS_ENABLED = args.https
+IS_UPLOAD_DISABLED = args.no_upload
+if args.dir:
+    dir_path = Path(args.dir)
+    if dir_path.is_dir():
+        BASE_DIR = dir_path.resolve()
+    else:
+        raise NotADirectoryError(
+            f"This path is not a dir or does not exist: {dir_path}"
+        )
+else:
+    BASE_DIR = Path.cwd().resolve()
+
+logger.info(f"\n{CYAN} Sharing Directory: {BASE_DIR}{RESET}")
+
+TEMPLATE_DIR = Path(__file__).resolve().parent / "templates"
+STATIC_DIR = Path(__file__).resolve().parent / "static"
+
+
+if args.code:
+    IS_AUTH_ENABLED = True
+    SERVER_CODE = args.code
+    logger.info(f"\n{BLUE} ACCESS CODE: {SERVER_CODE}{RESET}")
+elif IS_AUTH_ENABLED:
+    SERVER_CODE = "".join(secrets.choice("0123456789") for _ in range(6))
+    logger.info(f"\n{BLUE} ACCESS CODE: {SERVER_CODE}{RESET}")
+
+
+def check_authenticated(request: Request):
+    if IS_AUTH_ENABLED:
+        session_token = request.cookies.get(COOKIE_NAME)
+
+        if not session_token or not secrets.compare_digest(session_token, SERVER_CODE):
+            raise HTTPException(status_code=401, detail="Not authenticated")
+
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    make_qr()
+    logger.info(f"\n{BLUE} Server is running. Press Ctrl + C to stop...{RESET}\n")
+    yield
+
+
+app = FastAPI(lifespan=lifespan)
+
+app.mount("/static", StaticFiles(directory=STATIC_DIR), name="static")
+
+templates = Jinja2Templates(directory=TEMPLATE_DIR)
+
+
+@app.get("/login", response_class=HTMLResponse)
+async def login_page(request: Request):
+    if IS_AUTH_ENABLED and request.cookies.get(COOKIE_NAME) == SERVER_CODE:
+        return RedirectResponse(url="/", status_code=303)
+
+    return templates.TemplateResponse(request=request, name="login-page.html")
+
+
+@app.post("/login")
+async def login_process(request: Request, code: str = Form(...)):
+
+    if secrets.compare_digest(code, SERVER_CODE):
+        response = RedirectResponse("/", status_code=303)
+        response.set_cookie(
+            key=COOKIE_NAME,
+            value=SERVER_CODE,
+            httponly=True,
+            samesite="lax",
+        )
+        return response
+    return templates.TemplateResponse(
+        request=request,
+        name="login-page.html",
+        context={"error": True},
+        status_code=401,
+    )
+
+
+@app.get("/", response_class=HTMLResponse)
+@app.get("/{subpath:path}", response_class=HTMLResponse)
+async def get_files_page(request: Request, subpath: str = ""):
+    try:
+        check_authenticated(request)
+    except HTTPException as e:
+        if e.status_code == 401:
+            return RedirectResponse(url="/login", status_code=303)
+        raise e
+
+    zip_target = request.query_params.get("zip")
+    if zip_target:
+        target_path = (BASE_DIR / zip_target).resolve()
+
+        if (
+            not target_path.relative_to(BASE_DIR)
+            or not target_path.exists()
+            or not target_path.is_dir()
+        ):
+            raise HTTPException(
+                status_code=403, detail="Access denied or invalid directory"
+            )
+
+        zip_buffer = io.BytesIO()
+        with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zf:
+            for f_path in target_path.rglob("*"):
+                archive_name = f_path.relative_to(target_path)
+                if f_path.is_file():
+                    zf.write(f_path, archive_name)
+
+        zip_buffer.seek(0)
+        archive_filename = f"{target_path.name or 'root'}.zip"
+
+        encoded_filename = urllib.parse.quote(archive_filename)
+
+        return StreamingResponse(
+            zip_buffer,
+            media_type="application/x-zip-compressed",
+            headers={
+                "Content-Disposition": f"attachment; filename*=UTF-8''{encoded_filename}"
+            },
+        )
+
+    target_path = (BASE_DIR / subpath).resolve()
+
+    if not target_path.is_relative_to(BASE_DIR):
+        raise HTTPException(status_code=403, detail="Access denied")
+    if not target_path.exists():
+        raise HTTPException(status_code=404, detail="Not found")
+    if target_path.is_file():
+        return FileResponse(
+            path=target_path,
+            filename=target_path.name,
+            media_type="application/octet-stream",
+        )
+    if target_path != BASE_DIR:
+        parent_rel = target_path.parent.relative_to(BASE_DIR)
+        parent_path = (
+            str(parent_rel).replace("\\", "/") if str(parent_rel) != "." else ""
+        )
+    else:
+        parent_path = ""
+
+    dirs = []
+    files = []
+
+    for path in target_path.iterdir():
+        rel_path = path.relative_to(BASE_DIR)
+        web_rel_path = str(rel_path).replace("\\", "/")
+
+        if path.is_dir():
+            dirs.append(
+                {
+                    "name": path.name,
+                    "path": web_rel_path,
+                }
+            )
+        elif path.is_file():
+            _, ext = os.path.splitext(path)
+            files.append(
+                {
+                    "name": path.name,
+                    "path": web_rel_path,
+                    "ext": ext,
+                }
+            )
+    path_len = [p for p in subpath.split("/") if p]
+    depth = len(path_len)
+    context = {
+        "files": files,
+        "dirs": dirs,
+        "subpath": subpath,
+        "parent_path": parent_path,
+        "is_root": target_path == BASE_DIR,
+        "root": "",
+        "show_home": depth >= 2,
+        "upload": not IS_UPLOAD_DISABLED,
+    }
+
+    return templates.TemplateResponse(
+        request=request,
+        name="index.html",
+        context=context,
+    )
+
+
+@app.post("/")
+@app.post("/{subpath:path}")
+async def upload_file(
+    request: Request,
+    files: list[UploadFile],
+    subpath: str = "",
+):
+    if not IS_UPLOAD_DISABLED:
+        try:
+            check_authenticated(request)
+        except HTTPException as e:
+            if e.status_code == 401:
+                return RedirectResponse(url="/login", status_code=303)
+            raise e
+
+        target_path = (BASE_DIR / subpath).resolve()
+
+        if not target_path.is_relative_to(BASE_DIR):
+            raise HTTPException(status_code=403, detail="Access denied")
+        if not target_path.exists():
+            raise HTTPException(status_code=404, detail="Not found")
+        for file in files:
+            if not file.filename:
+                continue
+
+            dest_path = target_path / file.filename
+
+            try:
+                with dest_path.open("wb") as buf:
+                    shutil.copyfileobj(file.file, buf)
+            except Exception as e:
+                logger.error(f"Error occured while loading file: {e}")
+                raise HTTPException(status_code=500, detail="Could not save file")
+            finally:
+                await file.close()
+
+        return RedirectResponse(url=request.url.path, status_code=303)
+
+
+def run_app():
+    if IS_HTTPS_ENABLED:
+        with temp_tls_cert() as (cert_path, key_path):
+            uvicorn.run(
+                "app:app",
+                host="0.0.0.0",
+                port=8000,
+                reload=False,
+                log_config=log_config,
+                ssl_certfile=cert_path,
+                ssl_keyfile=key_path,
+            )
+    else:
+        uvicorn.run(
+            "app:app",
+            host="0.0.0.0",
+            port=8000,
+            reload=False,
+            log_config=log_config,
+        )
+
+
+if __name__ == "__main__":
+    run_app()

shdir-0.1.0/src/shdir/cli.py

@@ -0,0 +1,75 @@
+import argparse
+
+import qrcode
+
+from utils.colors import GREEN, RESET, UNDERLINE
+from utils.logs import logger
+from utils.net import get_local_ip
+
+parser = argparse.ArgumentParser(description="Shdir file manager")
+parser.add_argument(
+    "--code",
+    "-c",
+    type=str,
+    default=None,
+    help="Access code for current session",
+)
+
+parser.add_argument(
+    "--auth",
+    "-a",
+    action="store_true",
+    help="Run server with authentication",
+)
+parser.add_argument(
+    "--https",
+    action="store_true",
+    help="Run server with temporary tsl certificate",
+)
+parser.add_argument(
+    "--dir",
+    "-d",
+    type=str,
+    default=None,
+    help="Set up a directory to share",
+)
+parser.add_argument(
+    "--no-upload",
+    "-u",
+    action="store_true",
+    help="Run without file upload",
+)
+args, _ = parser.parse_known_args()
+
+
+def make_qr():
+    is_https = args.https
+    qr = qrcode.QRCode(version=1, box_size=1, border=1)
+
+    if is_https:
+        qr.add_data(f"https://{get_local_ip()}:8000/")
+        logger.info(
+            f"\n{UNDERLINE}{GREEN} Copy Link: https://{get_local_ip()}:8000/{RESET}\n\n"
+        )
+    else:
+        qr.add_data(f"http://{get_local_ip()}:8000/")
+        logger.info(
+            f"\n{UNDERLINE}{GREEN} Copy Link: http://{get_local_ip()}:8000/{RESET}\n\n"
+        )
+    qr.make(fit=True)
+
+    matrix = qr.get_matrix()
+
+    WHITE_BLOCK = "\033[7m  \033[m"
+    BLACK_BLOCK = "  "
+
+    indent = "  "
+
+    qr_lines = []
+    for row in matrix:
+        line = "".join(BLACK_BLOCK if cell else WHITE_BLOCK for cell in row)
+        qr_lines.append(f"{indent}{line}")
+
+    tabulated_qr = "\n".join(qr_lines)
+
+    logger.info(f"{tabulated_qr}")

shdir-0.1.0/src/shdir/icons/arrow_up.svg

@@ -0,0 +1,2 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><path d="m18 15-6-6-6 6"/></svg>
+

shdir-0.1.0/src/shdir/icons/file.svg

@@ -0,0 +1 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="lucide lucide-file preview-icon"><path d="M6 22a2 2 0 0 1-2-2V4a2 2 0 0 1 2-2h8a2.4 2.4 0 0 1 1.704.706l3.588 3.588A2.4 2.4 0 0 1 20 8v12a2 2 0 0 1-2 2z"/><path d="M14 2v5a1 1 0 0 0 1 1h5"/></svg>

shdir-0.1.0/src/shdir/icons/folder.svg

@@ -0,0 +1 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="lucide lucide-folder preview-icon"><path d="M20 20a2 2 0 0 0 2-2V8a2 2 0 0 0-2-2h-7.9a2 2 0 0 1-1.69-.9L9.6 3.9A2 2 0 0 0 7.93 3H4a2 2 0 0 0-2 2v13a2 2 0 0 0 2 2Z"/></svg>

shdir-0.1.0/src/shdir/icons/python.svg

@@ -0,0 +1 @@
+<svg role="img" viewBox="0 0 24 24" xmlns="http://www.w3.org/2000/svg"><title>Python</title><path d="M14.25.18l.9.2.73.26.59.3.45.32.34.34.25.34.16.33.1.3.04.26.02.2-.01.13V8.5l-.05.63-.13.55-.21.46-.26.38-.3.31-.33.25-.35.19-.35.14-.33.1-.3.07-.26.04-.21.02H8.77l-.69.05-.59.14-.5.22-.41.27-.33.32-.27.35-.2.36-.15.37-.1.35-.07.32-.04.27-.02.21v3.06H3.17l-.21-.03-.28-.07-.32-.12-.35-.18-.36-.26-.36-.36-.35-.46-.32-.59-.28-.73-.21-.88-.14-1.05-.05-1.23.06-1.22.16-1.04.24-.87.32-.71.36-.57.4-.44.42-.33.42-.24.4-.16.36-.1.32-.05.24-.01h.16l.06.01h8.16v-.83H6.18l-.01-2.75-.02-.37.05-.34.11-.31.17-.28.25-.26.31-.23.38-.2.44-.18.51-.15.58-.12.64-.1.71-.06.77-.04.84-.02 1.27.05zm-6.3 1.98l-.23.33-.08.41.08.41.23.34.33.22.41.09.41-.09.33-.22.23-.34.08-.41-.08-.41-.23-.33-.33-.22-.41-.09-.41.09zm13.09 3.95l.28.06.32.12.35.18.36.27.36.35.35.47.32.59.28.73.21.88.14 1.04.05 1.23-.06 1.23-.16 1.04-.24.86-.32.71-.36.57-.4.45-.42.33-.42.24-.4.16-.36.09-.32.05-.24.02-.16-.01h-8.22v.82h5.84l.01 2.76.02.36-.05.34-.11.31-.17.29-.25.25-.31.24-.38.2-.44.17-.51.15-.58.13-.64.09-.71.07-.77.04-.84.01-1.27-.04-1.07-.14-.9-.2-.73-.25-.59-.3-.45-.33-.34-.34-.25-.34-.16-.33-.1-.3-.04-.25-.02-.2.01-.13v-5.34l.05-.64.13-.54.21-.46.26-.38.3-.32.33-.24.35-.2.35-.14.33-.1.3-.06.26-.04.21-.02.13-.01h5.84l.69-.05.59-.14.5-.21.41-.28.33-.32.27-.35.2-.36.15-.36.1-.35.07-.32.04-.28.02-.21V6.07h2.09l.14.01zm-6.47 14.25l-.23.33-.08.41.08.41.23.33.33.23.41.08.41-.08.33-.23.23-.33.08-.41-.08-.41-.23-.33-.33-.23-.41-.08-.41.08z"/></svg>

shdir-0.1.0/src/shdir/icons/search.svg

@@ -0,0 +1 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="lucide lucide-search preview-icon"><path d="m21 21-4.34-4.34"/><circle cx="11" cy="11" r="8"/></svg>

shdir-0.1.0/src/shdir/icons/zip.svg

@@ -0,0 +1 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="lucide lucide-file-archive preview-icon"><path d="M13.659 22H18a2 2 0 0 0 2-2V8a2.4 2.4 0 0 0-.706-1.706l-3.588-3.588A2.4 2.4 0 0 0 14 2H6a2 2 0 0 0-2 2v11.5"/><path d="M14 2v5a1 1 0 0 0 1 1h5"/><path d="M8 12v-1"/><path d="M8 18v-2"/><path d="M8 7V6"/><circle cx="8" cy="20" r="2"/></svg>