GitHubMirror/sshyp-labs
1
0
Fork 0
mirror of https://github.com/rwinkhart/sshyp-labs.git synced 2026-08-28 04:46:48 -04:00
Code Issues Packages Projects Releases Wiki Activity

[export-to-libmutton] Fix various issues

Browse Source
This commit is contained in:
RandyTheSilly
2024-05-21 13:17:15 -04:00
parent df9d32c715
commit 9b44f438d4
2 changed files with 25 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extensions/export-to-libmutton/export-to-libmutton.ini
+1 -1
View File
@@ -1,3 +1,3 @@
[config] [config]
input = export input = export
output = /usr/lib/sshyp/export-to-libmutton.py output = /usr/lib/sshyp/export-to-libmutton
extensions/export-to-libmutton/export-to-libmutton.py
+24 -16
View File
@@ -2,17 +2,15 @@
from os import path, remove, system, walk from os import path, remove, system, walk
from pathlib import Path from pathlib import Path
from shutil import move, rmtree from shutil import rmtree
from sshyp import decrypt
# NOTE LARGELY UNTESTED, ALPHA EXTENSION
# TODO strip .gpg extension, if present
# TODO extract and move MFA data to dedicated field, if present # TODO extract and move MFA data to dedicated field, if present
if __name__ == "__main__": if __name__ == "__main__":
_pasture = path.expanduser('~/.local/share/sshyp') _pasture = path.expanduser('~/.local/share/sshyp')
# set new gpg key; note that this changes sshyp's gpg key, not just libmutton's _gpg_id = input('\nWhat is the ID of the GPG key you would like to use for re-encryption?\n\nID: ')
gpg_config()
# remove previous export if it exists # remove previous export if it exists
if Path(f"{_pasture}.export").exists(): if Path(f"{_pasture}.export").exists():
@@ -20,21 +18,24 @@ if __name__ == "__main__":
# prompt for unlock and display do not close warning # prompt for unlock and display do not close warning
decrypt(None) decrypt(None)
curses_radio(['okay'], 'entry conversion may take some time (especially on slower devices)\n\nselect "okay" ' print('entry conversion may take some time (especially on slower devices)\n\ndo not terminate this process!')
'to start\n\ndo not terminate this process!')
# iterate over entires, exporting them in the libmutton format # iterate over entires, exporting them in the libmutton format
if path.isdir(_pasture): if path.isdir(_pasture):
_gpg_id = sshyp_data.get('CLIENT-GENERAL', 'gpg_id')
for _dirPath, _dirNames, _filenames in walk(_pasture): for _dirPath, _dirNames, _filenames in walk(_pasture):
for _filename in sorted(_filenames): for _filename in sorted(_filenames):
# ensure parent directory within export tree exists # ensure parent directory within export tree exists
Path(dirPath.replace(_pasture, _pasture + '.export')).mkdir(0o700, parents=True, exist_ok=True) Path(_dirPath.replace(_pasture, _pasture + '.export')).mkdir(0o700, parents=True, exist_ok=True)
# manually decrypt due to potential lack of .gpg extension # remove the .gpg file extension, if it is present
system(f"gpg -d '{_dirPath}/{_filename}' > '{_dirPath.replace(_pasture, _pasture + '.export')}/{_filename}'") # prevents saving with the extension, as libmutton does not require it
if _filename.endswith('.gpg'):
_filename = _filename[:-4]
_old_contents, _new_contents = open(f"{_dirPath.replace(_pasture, _pasture + '.export')}/{_filename}", 'r').readlines(), '' # decrypt to export directory, append .gpg to decrypted file to differentiate from new file
system(f"gpg -d '{_dirPath}/{_filename}.gpg' > '{_dirPath.replace(_pasture, _pasture + '.export')}/{_filename}.gpg'")
_old_contents, _new_contents = open(f"{_dirPath.replace(_pasture, _pasture + '.export')}/{_filename}.gpg", 'r').readlines(), ''
# add a new line reserved for libmutton's TOTP support (between username and URL) # add a new line reserved for libmutton's TOTP support (between username and URL)
for _num in range(len(_old_contents)): for _num in range(len(_old_contents)):
@@ -50,10 +51,17 @@ if __name__ == "__main__":
case _: case _:
_new_contents += _old_contents[_num] _new_contents += _old_contents[_num]
# write and manually encrypt the output file # remove any trailing whitespace
open(f"{_dirPath.replace(_pasture, _pasture + '.export')}/{_filename}", 'w').writelines(_new_contents) for _num in reversed(range(len(_new_contents))):
system(f"gpg -qr {str(_gpg_id)} -e '{_dirPath.replace(_pasture, _pasture + '.export')}/{_filename}'") if _new_contents[_num] in ('\n', '', ' '):
remove(f"{_dirPath.replace(_pasture, _pasture + '.export')}/{_filename}") _new_contents = _new_contents[:-1]
else:
break
# write and encrypt the output file
open(f"{_dirPath.replace(_pasture, _pasture + '.export')}/{_filename}.gpg", 'w').writelines(_new_contents)
system(f"gpg -er {str(_gpg_id)} -o '{_dirPath.replace(_pasture, _pasture + '.export')}/{_filename}' '{_dirPath.replace(_pasture, _pasture + '.export')}/{_filename}.gpg'")
remove(f"{_dirPath.replace(_pasture, _pasture + '.export')}/{_filename}.gpg")
else: else:
print(f"\n\u001b[38;5;9merror: no entry folder ({_pasture}) found\u001b[0m\n") print(f"\n\u001b[38;5;9merror: no entry folder ({_pasture}) found\u001b[0m\n")
s_exit(2) s_exit(2)