56 lines
1.9 KiB
Python
Executable File
56 lines
1.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import os
|
|
import tempfile
|
|
import shutil
|
|
import glob
|
|
from subprocess import run
|
|
|
|
SOURCE = os.environ.get("PLUGIN_SOURCE", ".")
|
|
TARGET = os.environ['PLUGIN_TARGET']
|
|
|
|
def apply_key_permissions(keyfile):
|
|
os.chmod(keyfile, 0o600)
|
|
if not os.name == "nt":
|
|
return
|
|
|
|
username = os.environ['USERNAME']
|
|
users_directory = "C:\\Users"
|
|
run(["Icacls", keyfile, "/c", "/t", "/Inheritance:d"])
|
|
run(["TakeOwn", "/F", keyfile])
|
|
run(["Icacls", keyfile, "/c", "/t", "/Grant:r", f"{username}:F"])
|
|
run(["Icacls", keyfile, "/c", "/t", "/Remove:g", "Administrator", "Authenticated Users", "BUILTIN\\Administrators", "BUILTIN", "Everyone", "System", "Users"])
|
|
for other_user in [user for user in os.listdir(users_directory) if not user == username]:
|
|
run(["Icacls", keyfile, "/c", "/t", "/Remove:g", other_user])
|
|
run(["Icacls", keyfile])
|
|
|
|
def deploy(source, target, auth):
|
|
for source_file in glob.glob(source):
|
|
deploy_file(source_file, target, auth)
|
|
|
|
def deploy_file(source_file, target, auth):
|
|
print(f">> {source_file} -> {target}")
|
|
if target.startswith("http://") or target.startswith("https://"):
|
|
run(["curl", "--user", auth, target, "--upload-file", source_file], check=True)
|
|
else:
|
|
run(["scp", "-i", auth, "-o", "StrictHostKeyChecking=no", "-o", "PasswordAuthentication=no", "-r", source_file, target], check=True)
|
|
|
|
temp_file_name = None
|
|
auth = None
|
|
try:
|
|
if 'PLUGIN_KEY' in os.environ:
|
|
with tempfile.NamedTemporaryFile(delete=False) as deploy_key:
|
|
temp_file_name = deploy_key.name
|
|
deploy_key.write(os.environ['PLUGIN_KEY'].encode())
|
|
deploy_key.write(b"\n")
|
|
deploy_key.close()
|
|
|
|
apply_key_permissions(deploy_key.name)
|
|
auth = deploy_key.name
|
|
else:
|
|
auth = os.environ['PLUGIN_AUTHENTICATION']
|
|
|
|
deploy(SOURCE, TARGET, auth)
|
|
finally:
|
|
if temp_file_name is not None:
|
|
os.remove(temp_file_name)
|