[wip] url shortener support (#19)

Signed-off-by: r4sas <r4sas@i2pmail.org>

pbincli/__init__.py

@@ -2,6 +2,6 @@
 # -*- coding: utf-8 -*-
 
 __author__ = "R4SAS <r4sas@i2pmail.org>"
-__version__ = "0.2.1"
+__version__ = "0.2.2b1"
 __copyright__ = "Copyright (c) R4SAS"
 __license__ = "MIT"

pbincli/actions.py

@@ -1,4 +1,5 @@
 from pbincli.format import Paste
+from pbincli.utils import PBinCLIError
 
 def send(args, api_client):
     if not args.notext:
@@ -7,8 +8,7 @@ def send(args, api_client):
         elif args.stdin:
             text = args.stdin.read()
     elif not args.file:
-        print("Nothing to send!")
-        exit(1)
+        PBinCLIError("Nothing to send!")
     else:
         text = ""
 
@@ -64,11 +64,9 @@ def send(args, api_client):
             result['id'],
             passphrase))
     elif result['status']: # return code is other then zero
-        print("Something went wrong...\nError:\t\t{}".format(result['message']))
-        exit(1)
+        PBinCLIError("Something went wrong...\nError:\t\t{}".format(result['message']))
     else: # or here no status field in response or it is empty
-        print("Something went wrong...\nError: Empty response.")
-        exit(1)
+        PBinCLIError("Something went wrong...\nError: Empty response.")
 
 
 def get(args, api_client):
@@ -77,12 +75,10 @@ def get(args, api_client):
     try:
         pasteid, passphrase = args.pasteinfo.split("#")
     except ValueError:
-        print("PBinCLI error: provided info hasn't contain valid PasteID#Passphrase string")
-        exit(1)
+        PBinCLIError("Provided info hasn't contain valid PasteID#Passphrase string")
 
     if not (pasteid and passphrase):
-        print("PBinCLI error: Incorrect request")
-        exit(1)
+        PBinCLIError("Incorrect request")
 
     if args.debug: print("PasteID:\t{}\nPassphrase:\t{}".format(pasteid, passphrase))
 
@@ -139,11 +135,9 @@ def get(args, api_client):
             api_client.delete(json_encode({'pasteid':pasteid,'deletetoken':'burnafterreading'}))
 
     elif result['status']: # return code is other then zero
-        print("Something went wrong...\nError:\t\t{}".format(result['message']))
-        exit(1)
+        PBinCLIError("Something went wrong...\nError:\t\t{}".format(result['message']))
     else: # or here no status field in response or it is empty
-        print("Something went wrong...\nError: Empty response.")
-        exit(1)
+        PBinCLIError("Something went wrong...\nError: Empty response.")
 
 
 def delete(args, api_client):

pbincli/api.py

@@ -1,4 +1,5 @@
 import requests
+from pbincli.utils import PBinCLIError
 
 class PrivateBin:
     def __init__(self, settings=None):
@@ -27,8 +28,7 @@ class PrivateBin:
         try:
             return result.json()
         except ValueError:
-            print("ERROR: Unable parse response as json. Received (size = {}):\n{}".format(len(result.text), result.text))
-            exit(1)
+            PBinCLIError("Unable parse response as json. Received (size = {}):\n{}".format(len(result.text), result.text))
 
 
     def get(self, request):
@@ -56,11 +56,9 @@ class PrivateBin:
         if not result['status']:
             print("Paste successfully deleted!")
         elif result['status']:
-            print("Something went wrong...\nError:\t\t{}".format(result['message']))
-            exit(1)
+            PBinCLIError("Something went wrong...\nError:\t\t{}".format(result['message']))
         else:
-            print("Something went wrong...\nError: Empty response.")
-            exit(1)
+            PBinCLIError("Something went wrong...\nError: Empty response.")
 
 
     def getVersion(self):
@@ -74,9 +72,24 @@ class PrivateBin:
             else 1
 
 class Shortener:
-    def __init__(self, settings=None):
-        self.server = settings['server']
-        self.headers = {'X-Requested-With': 'JSONHttpRequest'}
+    """Some parts of this class was taken from
+    python-yourls (https://github.com/tflink/python-yourls/) library
+    """
+    def __init__(self, apiurl, username=None, password=None, token=None, settings=None):
+        # we checking which service is used, because some services doesn't require
+        # any authentication, or have only one domain on which it working
+        if settings['api'] == 'yourls':
+            if not apiurl:
+                PBinCLIError("YOURLS: An API URL is required")
+            self.apiurl = apiurl
+
+            if not username or not password:
+                if not self.token:
+                    PBinCLIError("YOURLS: username and password or token are required")
+                else:
+                    self.auth_args = {'signature': token}
+            else:
+                self.auth_args = {'username': self.username, 'password': self.password}
 
         if settings['proxy']:
             self.proxy = {settings['proxy'].split('://')[0]: settings['proxy']}
@@ -90,3 +103,33 @@ class Shortener:
         self.session = requests.Session()
         self.session.verify = settings['nocheckcert']
 
+    def yourls(self, url):
+        data_format = 'json'
+        args = {'action': 'shorturl', 'format': data_format, 'url': url}
+        reqest = dict(args.items() + self.auth_args.items())
+
+        result = self.session.post(
+            url = self.apiurl,
+            proxies = self.proxy,
+            data = request)
+
+        try:
+            result.json()
+            if result['status'] == 'fail' and result['code'] == 'error:keyword':
+                PBinCLIError("YOURLS: Received error from API: {}".format(result['message']))
+            if not 'shorturl' in result:
+                PBinCLIError("YOURLS: Unknown error: {}".format(result['message']))
+
+            print("Paste short URL: {}".format(result['shorturl']))
+        except ValueError:
+            PBinCLIError("YOURLS: Unable parse response. Received (size = {}):\n{}".format(len(result.text), result.text))
+
+    def clck(self, url):
+        # from urllib.parse import quote_plus
+        request = {'url': url}
+
+        result = self.session.post(
+            url = "https://clck.ru/--",
+            proxies = self.proxy,
+            data = request)
+        print("Paste short URL: {}".format(result))

pbincli/cli.py

@@ -39,7 +39,7 @@ def main():
     # URL shortener
     send_parser.add_argument("-s", "--shorten", default=False, action="store_true", help="use URL shortener")
     send_parser.add_argument("--shorten-api", default="yourls", action="store",
-        choices=["yourls"], help="select API used for selected URL shortener service")
+        choices=["clckru", "yourls"], help="API used by shortener service (default: YOURLS)")
     send_parser.add_argument("--shorten-url", help="URL of shortener service API")
     send_parser.add_argument("--shorten-user", help="Shortener username")
     send_parser.add_argument("--shorten-pass", help="Shortener password")
@@ -89,7 +89,7 @@ def main():
         if var in os.environ: CONFIG[key] = os.getenv(var)
 
     SETTINGS = {
-        "server" : validate_url(CONFIG["server"])
+        "server" : validate_url(CONFIG["server"]),
         "proxy": CONFIG["proxy"],
         "nocheckcert": args.no_check_certificate,
         "noinsecurewarn": args.no_insecure_warning
@@ -101,8 +101,7 @@ def main():
         try:
             args.func(args, api_client)
         except PBinCLIException as pe:
-            print("PBinCLI error: {}".format(pe))
-            sys.exit(1)
+            raise PBinCLIException("error: {}".format(pe))
     else:
         parser.print_help()
 

pbincli/format.py

@@ -1,7 +1,7 @@
 from Crypto.Random import get_random_bytes
 from Crypto.Cipher import AES
 from base64 import b64encode, b64decode
-from pbincli.utils import PBinCLIException
+from pbincli.utils import PBinCLIError
 import zlib
 
 CIPHER_ITERATION_COUNT = 100000
@@ -143,7 +143,7 @@ class Paste:
         elif self._version == 1:
             return zlib.decompress(bytearray(map(lambda c:ord(c)&255, b64decode(s.encode('utf-8')).decode('utf-8'))), -zlib.MAX_WBITS)
         else:
-            raise PBinCLIException('Unknown compression type provided in paste!')
+            PBinCLIError('Unknown compression type provided in paste!')
 
 
     def __compress(self, s):
@@ -160,7 +160,7 @@ class Paste:
             b = co.compress(s) + co.flush()
             return b64encode(''.join(map(chr, b)).encode('utf-8'))
         else:
-            raise PBinCLIException('Unknown compression type provided!')
+            PBinCLIError('Unknown compression type provided!')
 
 
     def decrypt(self):

pbincli/utils.py

@@ -1,9 +1,14 @@
-import json, ntpath, os
+import json, ntpath, os, sys
 
 class PBinCLIException(Exception):
     pass
 
 
+def PBinCLIError(message):
+    print("PBinCLI Error: {}".format(message), file=sys.stderr)
+    exit(1)
+
+
 def path_leaf(path):
     head, tail = ntpath.split(path)
     return tail or ntpath.basename(head)
@@ -12,13 +17,13 @@ def path_leaf(path):
 def check_readable(f):
     # Checks if path exists and readable
     if not os.path.exists(f) or not os.access(f, os.R_OK):
-        raise PBinCLIException("Error accessing path: {}".format(f))
+        PBinCLIError("Error accessing path: {}".format(f))
 
 
 def check_writable(f):
     # Checks if path is writable
     if not os.access(os.path.dirname(f) or ".", os.W_OK):
-        raise PBinCLIException("Path is not writable: {}".format(f))
+        PBinCLIError("Path is not writable: {}".format(f))
 
 
 def json_encode(s):