__filename__ = "xmpp.py" __author__ = "Bob Mottram" __license__ = "AGPL3+" __version__ = "1.4.0" __maintainer__ = "Bob Mottram" __email__ = "bob@libreserver.org" __status__ = "Production" __module_group__ = "Profile Metadata" from utils import get_attachment_property_value def get_xmpp_address(actor_json: {}) -> str: """Returns xmpp address for the given actor """ if not actor_json.get('attachment'): return '' for property_value in actor_json['attachment']: name_value = None if property_value.get('name'): name_value = property_value['name'].lower() elif property_value.get('schema:name'): name_value = property_value['schema:name'].lower() if not name_value: continue if not (name_value.startswith('xmpp') or name_value.startswith('jabber')): continue if not property_value.get('type'): continue prop_value_name, _ = \ get_attachment_property_value(property_value) if not prop_value_name: continue if not property_value['type'].endswith('PropertyValue'): continue if '@' not in property_value[prop_value_name]: continue if '"' in property_value[prop_value_name]: continue return property_value[prop_value_name] return '' def set_xmpp_address(actor_json: {}, xmpp_address: str) -> None: """Sets an xmpp address for the given actor """ not_xmpp_address = False if '@' not in xmpp_address: not_xmpp_address = True if '.' not in xmpp_address: not_xmpp_address = True if '"' in xmpp_address: not_xmpp_address = True if '<' in xmpp_address: not_xmpp_address = True if not actor_json.get('attachment'): actor_json['attachment'] = [] # remove any existing value property_found = None for property_value in actor_json['attachment']: name_value = None if property_value.get('name'): name_value = property_value['name'] elif property_value.get('schema:name'): name_value = property_value['schema:name'] if not name_value: continue if not property_value.get('type'): continue if not (name_value.lower().startswith('xmpp') or name_value.lower().startswith('jabber')): continue property_found = property_value break if property_found: actor_json['attachment'].remove(property_found) if not_xmpp_address: return for property_value in actor_json['attachment']: name_value = None if property_value.get('name'): name_value = property_value['name'] elif property_value.get('schema:name'): name_value = property_value['schema:name'] if not name_value: continue if not property_value.get('type'): continue name_value = name_value.lower() if not (name_value.startswith('xmpp') or name_value.startswith('jabber')): continue if not property_value['type'].endswith('PropertyValue'): continue prop_value_name, _ = \ get_attachment_property_value(property_value) if not prop_value_name: continue property_value[prop_value_name] = xmpp_address return new_xmpp_address = { "name": "XMPP", "type": "PropertyValue", "value": xmpp_address } actor_json['attachment'].append(new_xmpp_address)