mirror of
https://github.com/9001/copyparty.git
synced 2026-06-18 20:22:27 -06:00
70 lines
1.9 KiB
Python
70 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import os
|
|
|
|
|
|
_ = r"""
|
|
rename incoming uploads according to the "wark" (the file identifier)
|
|
which is basically but not exactly a sha512 hash of the file contents
|
|
|
|
NOTE: this only works for up2k uploads (dragdrop into browser);
|
|
combine this with reloc-by-wark-xau.py to cover the other protocols
|
|
|
|
example usage as global config:
|
|
-e2d --xbu I,c,bin/hooks/reloc-by-wark-xbu.py
|
|
|
|
parameters explained,
|
|
e2d = enable up2k database (mandatory for xbu hooks)
|
|
xbu = execute before upload
|
|
I = import this hook for performance; do not fork / subprocess
|
|
c = "check"; reject upload if this hook crashes due to a bug
|
|
|
|
example usage as a volflag (per-volume config):
|
|
-v srv/inc:inc:r:rw,ed:c,e2d,xbu=I,c,bin/hooks/reloc-by-wark-xbu.py
|
|
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
(share filesystem-path srv/inc as volume /inc,
|
|
readable by everyone, read-write for user 'ed',
|
|
running this plugin on all uploads with the params explained above)
|
|
|
|
example usage as a volflag in a copyparty config file:
|
|
[/inc]
|
|
srv/inc
|
|
accs:
|
|
r: *
|
|
rw: ed
|
|
flags:
|
|
e2d, xbu: I,c,bin/hooks/reloc-by-wark-xbu.py
|
|
"""
|
|
|
|
|
|
def main(inf):
|
|
wark = inf.get("wark")
|
|
if not wark:
|
|
# not an up2k upload, so we don't have the hash;
|
|
|
|
# option 1: let upload proceed with original filename
|
|
return {}
|
|
|
|
# option 2: reject the upload
|
|
return {"rejectmsg": "only up2k uploads are allowed in this volume"}
|
|
|
|
# grab the original filename from the vpath...
|
|
vdir, fn = os.path.split(inf["vp"])
|
|
|
|
# ...to retain the original file extension, if any
|
|
try:
|
|
fn, ext = fn.rsplit(".", 1)
|
|
except:
|
|
ext = ""
|
|
|
|
# use the first 16 characters; 12 bytes of entropy,
|
|
# roughly one collision for every 26 million files
|
|
fn = wark[:16]
|
|
|
|
if ext:
|
|
ext = ext.lower()
|
|
fn += "." + ext
|
|
|
|
return {"reloc": {"fn": fn}}
|