Add Gyazo direct image url option

README.md

@@ -63,6 +63,7 @@ If an external file is found, Myazo extends the default config with the provided
 | Key                | Default                                | Description                                         |
 |--------------------|----------------------------------------|-----------------------------------------------------|
 | gyazo_server       | False                                  | Controls whether to use Gyazo's servers             |
+| gyazo_direct_link  | True                                   | Controls whether to open Gyazo direct image url     |
 | upload_script      | 'https://myazo.example.com/upload.php' | Full path to the upload.php file                    |
 | secret             | 'hunter2'                              | Secret token                                        |
 | clear_metadata     | True                                   | Controls clearing screenshot metadata before upload |

client/config.ini.example

@@ -1,5 +1,6 @@
 [Myazo]
 gyazo_server = False
+gyazo_direct_link = True
 upload_script = https://myazo.example.com/upload.php
 secret = hunter2
 clear_metadata = True

client/myazo.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 
 from configparser import ConfigParser
+from pathlib import Path
 from subprocess import run
 import os
 import platform
@@ -15,6 +16,7 @@
 config = ConfigParser()
 config.read_dict({'Myazo': {
     'gyazo_server': False, # If True, upload_script and secret are ignored
+    'gyazo_direct_link': True, # Ignored if gyazo_server is False
     'upload_script': 'https://myazo.example.com/upload.php',
     'secret': 'hunter2',
     'clear_metadata': True,
@@ -68,7 +70,6 @@
 img = open(tmp_file, 'rb')
 
 if config.get('gyazo_server'):
-    # If an error occurred, it'll output an error image url
     r = requests.post(
         'https://upload.gyazo.com/upload.cgi',
         files={'imagedata': img}
@@ -84,7 +85,12 @@
     print('Error: Failed to upload screenshot. Server returned status code {}.'.format(r.status_code))
     exit(-2)
 
-url = r.text
+if config.get('gyazo_server') and config.get('gyazo_direct_link'):
+    # Convert the Gyazo link to a direct image
+    # https://gyazo.com/hash > https://i.gyazo.com/hash.extension
+    url = r.text.replace('//', '//i.') + Path(tmp_file).suffix
+else:
+    url = r.text
 
 if config.getboolean('open_browser'):
     webbrowser.open(url)
@@ -94,4 +100,4 @@
     print(url)
 
 img.close()
-os.remove(tmp_file)
+os.remove(tmp_file)