|
|
|
@ -32,12 +32,6 @@ from os.path import join as path_join
|
|
|
|
|
import dbus.service
|
|
|
|
|
from dbus.mainloop.glib import DBusGMainLoop
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
from thefuzz import fuzz
|
|
|
|
|
from thefuzz import process
|
|
|
|
|
except ModuleNotFoundError:
|
|
|
|
|
from fuzzywuzzy import fuzz
|
|
|
|
|
from fuzzywuzzy import process
|
|
|
|
|
from gi.repository import GLib
|
|
|
|
|
|
|
|
|
|
# Convenience shorthand for declaring dbus interface methods.
|
|
|
|
@ -116,12 +110,8 @@ class SearchPassService(dbus.service.Object):
|
|
|
|
|
[self.password_executable, "list"], universal_newlines=True
|
|
|
|
|
).split("\n")[:-1]
|
|
|
|
|
|
|
|
|
|
results = [
|
|
|
|
|
e[0]
|
|
|
|
|
for e in process.extract(
|
|
|
|
|
name, password_list, limit=5, scorer=fuzz.partial_ratio
|
|
|
|
|
)
|
|
|
|
|
]
|
|
|
|
|
results = list(filter(lambda x: x.startswith(name), password_list))[:5]
|
|
|
|
|
|
|
|
|
|
if field is not None:
|
|
|
|
|
results = [f":{field} {r}" for r in results]
|
|
|
|
|
return results
|
|
|
|
@ -149,51 +139,51 @@ class SearchPassService(dbus.service.Object):
|
|
|
|
|
path = path_join(dir_path, filename)[:-4]
|
|
|
|
|
password_list.append(path)
|
|
|
|
|
|
|
|
|
|
results = [
|
|
|
|
|
e[0]
|
|
|
|
|
for e in process.extract(
|
|
|
|
|
name, password_list, limit=5, scorer=fuzz.partial_ratio
|
|
|
|
|
)
|
|
|
|
|
]
|
|
|
|
|
results = list(filter(lambda x: x.startswith(name), password_list))[:5]
|
|
|
|
|
|
|
|
|
|
if field == "otp":
|
|
|
|
|
results = [f"otp {r}" for r in results]
|
|
|
|
|
elif field is not None:
|
|
|
|
|
results = [f":{field} {r}" for r in results]
|
|
|
|
|
return results
|
|
|
|
|
|
|
|
|
|
def get_value_from_output(self, output, field=None):
|
|
|
|
|
if field is not None:
|
|
|
|
|
match = re.search(
|
|
|
|
|
rf"^{field}:\s*(?P<value>.+?)$", output, flags=re.I | re.M
|
|
|
|
|
)
|
|
|
|
|
if match:
|
|
|
|
|
value = match.group("value")
|
|
|
|
|
else:
|
|
|
|
|
raise RuntimeError(f"The field {field} was not found in the password entry.")
|
|
|
|
|
else:
|
|
|
|
|
value = output.split("\n", 1)[0]
|
|
|
|
|
return value
|
|
|
|
|
|
|
|
|
|
def send_password_to_gpaste(self, base_args, name, field=None):
|
|
|
|
|
gpaste = self.session_bus.get_object(
|
|
|
|
|
"org.gnome.GPaste.Daemon", "/org/gnome/GPaste"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
output = subprocess.check_output(base_args + [name], universal_newlines=True)
|
|
|
|
|
if field is not None:
|
|
|
|
|
match = re.search(
|
|
|
|
|
rf"^{field}:\s*(?P<value>.+?)$", output, flags=re.I | re.M
|
|
|
|
|
)
|
|
|
|
|
if match:
|
|
|
|
|
password = match.group("value")
|
|
|
|
|
else:
|
|
|
|
|
raise RuntimeError(f"The field {field} was not found in the pass file.")
|
|
|
|
|
else:
|
|
|
|
|
password = output.split("\n", 1)[0]
|
|
|
|
|
value = self.get_value_from_output(output, field)
|
|
|
|
|
try:
|
|
|
|
|
gpaste.AddPassword(name, password, dbus_interface="org.gnome.GPaste1")
|
|
|
|
|
gpaste.AddPassword(name, value, dbus_interface="org.gnome.GPaste1")
|
|
|
|
|
except dbus.DBusException:
|
|
|
|
|
gpaste.AddPassword(name, password, dbus_interface="org.gnome.GPaste2")
|
|
|
|
|
gpaste.AddPassword(name, value, dbus_interface="org.gnome.GPaste2")
|
|
|
|
|
|
|
|
|
|
def send_password_to_native_clipboard(self, base_args, name, field=None):
|
|
|
|
|
if field is not None:
|
|
|
|
|
raise RuntimeError("This feature requires GPaste.")
|
|
|
|
|
|
|
|
|
|
if self.password_mode == "bw":
|
|
|
|
|
p1 = subprocess.Popen(base_args + [name], stdout=subprocess.PIPE)
|
|
|
|
|
p2 = subprocess.run(self.clipboard_executable, stdin=p1.stdout)
|
|
|
|
|
if p1.returncode or p2.returncode:
|
|
|
|
|
output = subprocess.check_output(base_args + [name], universal_newlines=True)
|
|
|
|
|
value = self.get_value_from_output(output, field)
|
|
|
|
|
p1 = subprocess.run(self.clipboard_executable, input=value, text=True)
|
|
|
|
|
if p1.returncode:
|
|
|
|
|
raise RuntimeError(
|
|
|
|
|
f"Error while running rbw: got return codes: {p1.returncode} {p2.returncode}."
|
|
|
|
|
f"Error while running copying to clipboard: got return code: {p1.returncode}."
|
|
|
|
|
)
|
|
|
|
|
else:
|
|
|
|
|
if field is not None:
|
|
|
|
|
raise RuntimeError("This feature requires GPaste.")
|
|
|
|
|
result = subprocess.run(base_args + ["-c", name])
|
|
|
|
|
if result.returncode:
|
|
|
|
|
raise RuntimeError(
|
|
|
|
|