daniel156161
42064de633
Build and Push Docker Container / build-and-push (push) Successful in 1m3s
- Add a separate installable NanoShare CLI package under cli/. - Implement browser login with local callback and refresh-token config. - Add upload, list, download, sync, and watch CLI commands. - Add private owner-only file download endpoint for CLI downloads. - Add CLI auth endpoints for browser login and token refresh. - Return file_id from ServiceLink uploads for reliable sync state. - Exclude the CLI package from NanoShare container builds. - Include tests for private downloads and sync state updates.
55 lines
2.0 KiB
Python
55 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
import base64
|
|
import mimetypes
|
|
from pathlib import Path
|
|
from urllib.parse import quote
|
|
|
|
import httpx
|
|
|
|
def content_type(path: Path) -> str:
|
|
return mimetypes.guess_type(path.name)[0] or 'application/octet-stream'
|
|
|
|
def upload(client, node: str, path: Path, remote_name: str, note: str, expires: str) -> dict:
|
|
content_b64 = base64.b64encode(path.read_bytes()).decode('ascii')
|
|
params = {
|
|
'file_name': remote_name,
|
|
'content_b64': content_b64,
|
|
'content_type': content_type(path),
|
|
'note': note,
|
|
}
|
|
if expires:
|
|
params['expires'] = expires
|
|
result = client.call(node, 'files.upload', params)
|
|
return result if isinstance(result, dict) else {'result': result}
|
|
|
|
def delete_remote(client, node: str, file_id: str) -> None:
|
|
client.call(node, 'files.delete', {'file_id': file_id})
|
|
|
|
def list_remote(client, node: str) -> list[dict]:
|
|
result = client.call(node, 'files.list', {})
|
|
files = result.get('files', []) if isinstance(result, dict) else []
|
|
return files if isinstance(files, list) else []
|
|
|
|
def download_url(client, node: str, file_id: str) -> str:
|
|
base = client.registry.get(node)
|
|
if not base:
|
|
raise RuntimeError(f'unknown node: {node}')
|
|
return f"{base.rstrip('/')}/api/files/{quote(file_id, safe='')}/download"
|
|
|
|
def download(client, node: str, file_id: str, path: Path) -> None:
|
|
path.parent.mkdir(parents=True, exist_ok=True)
|
|
headers = {'Accept': '*/*', **client.auth_headers()}
|
|
with httpx.Client(timeout=client.timeout) as http:
|
|
with http.stream('GET', download_url(client, node, file_id), headers=headers) as response:
|
|
if response.status_code == 404:
|
|
raise FileNotFoundError(file_id)
|
|
if response.status_code == 410:
|
|
raise RuntimeError(f'remote file expired: {file_id}')
|
|
response.raise_for_status()
|
|
tmp = path.with_name(f'{path.name}.{Path.cwd().stat().st_ino}.tmp')
|
|
with tmp.open('wb') as handle:
|
|
for chunk in response.iter_bytes():
|
|
handle.write(chunk)
|
|
tmp.replace(path)
|