From 393d7fcb5c423be7a1b1f24b35889a47a267fc2c Mon Sep 17 00:00:00 2001
From: Bob Mottram <bob@libreserver.org>
Date: Thu, 12 May 2022 13:39:07 +0100
Subject: [PATCH] Command to get raw ssml

---
 epicyon.py | 38 +++++++++++++++++++++++++++++++++-----
 session.py | 39 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 72 insertions(+), 5 deletions(-)

diff --git a/epicyon.py b/epicyon.py
index b5aa6e348..46dc0b418 100644
--- a/epicyon.py
+++ b/epicyon.py
@@ -46,6 +46,7 @@ from session import create_session
 from session import get_json
 from session import get_vcard
 from session import download_html
+from session import download_ssml
 from newswire import get_rss
 from filters import add_filter
 from filters import remove_filter
@@ -341,6 +342,8 @@ parser.add_argument('--xmlvcard', dest='xmlvcard', type=str, default=None,
                     'activitypub actor url')
 parser.add_argument('--json', dest='json', type=str, default=None,
                     help='Show the json for a given activitypub url')
+parser.add_argument('--ssml', dest='ssml', type=str, default=None,
+                    help='Show the SSML for a given activitypub url')
 parser.add_argument('--htmlpost', dest='htmlpost', type=str, default=None,
                     help='Show the html for a given activitypub url')
 parser.add_argument('--rss', dest='rss', type=str, default=None,
@@ -1040,6 +1043,31 @@ if args.json:
         pprint(test_json)
     sys.exit()
 
+if args.ssml:
+    session = create_session(None)
+    profile_str = 'https://www.w3.org/ns/activitystreams'
+    as_header = {
+        'Accept': 'application/ssml+xml; profile="' + profile_str + '"'
+    }
+    if not args.domain:
+        args.domain = get_config_param(base_dir, 'domain')
+    domain = ''
+    if args.domain:
+        domain = args.domain
+    signing_priv_key_pem = get_instance_actor_key(base_dir, domain)
+    if debug:
+        print('base_dir: ' + str(base_dir))
+        if signing_priv_key_pem:
+            print('Obtained instance actor signing key')
+        else:
+            print('Did not obtain instance actor key for ' + domain)
+    test_ssml = download_ssml(signing_priv_key_pem, session, args.ssml,
+                              as_header, None, debug, __version__,
+                              http_prefix, domain)
+    if test_ssml:
+        print(str(test_ssml))
+    sys.exit()
+
 if args.vcard:
     session = create_session(None)
     if not args.domain:
@@ -1084,11 +1112,11 @@ if args.htmlpost:
             print('Obtained instance actor signing key')
         else:
             print('Did not obtain instance actor key for ' + domain)
-    testHtml = download_html(signing_priv_key_pem, session, args.htmlpost,
-                             as_header, None, debug, __version__,
-                             http_prefix, domain)
-    if testHtml:
-        print(testHtml)
+    test_html = download_html(signing_priv_key_pem, session, args.htmlpost,
+                              as_header, None, debug, __version__,
+                              http_prefix, domain)
+    if test_html:
+        print(test_html)
     sys.exit()
 
 # create cache for actors
diff --git a/session.py b/session.py
index bbea835ba..fe6d2a350 100644
--- a/session.py
+++ b/session.py
@@ -373,6 +373,45 @@ def download_html(signing_priv_key_pem: str,
                              None, quiet, debug, False)
 
 
+def download_ssml(signing_priv_key_pem: str,
+                  session, url: str, headers: {}, params: {}, debug: bool,
+                  version: str = __version__, http_prefix: str = 'https',
+                  domain: str = 'testdomain',
+                  timeout_sec: int = 20, quiet: bool = False) -> {}:
+    if not isinstance(url, str):
+        if debug and not quiet:
+            print('url: ' + str(url))
+            print('ERROR: download_ssml failed, url should be a string')
+        return None
+    session_params = {}
+    session_headers = {}
+    if headers:
+        session_headers = headers
+    if params:
+        session_params = params
+    session_headers['Accept'] = 'application/ssml+xml'
+    session_headers['User-Agent'] = 'Epicyon/' + version
+    if domain:
+        session_headers['User-Agent'] += \
+            '; +' + http_prefix + '://' + domain + '/'
+    if not session:
+        if not quiet:
+            print('WARN: download_ssml failed, no session specified')
+        return None
+
+    if debug:
+        HTTPConnection.debuglevel = 1
+
+    if signing_priv_key_pem:
+        return _get_json_signed(session, url, domain,
+                                session_headers, session_params,
+                                timeout_sec, signing_priv_key_pem,
+                                quiet, debug)
+    return _get_json_request(session, url, domain, session_headers,
+                             session_params, timeout_sec,
+                             None, quiet, debug, False)
+
+
 def _set_user_agent(session, http_prefix: str, domain_full: str) -> None:
     """Sets the user agent
     """