post to twitter

Author: raylu

config.yaml.example

@@ -13,6 +13,13 @@ twitter: # set this to null to disable
     bearer_token: null
     accounts:
         RiskofRain: '282441291327864834'
+twitter_post: # set this to null to disable
+    consumer_key: null # get these from https://developer.twitter.com/en/apps/
+    consumer_secret: null
+    token: null # get these from twitter_key.py
+    token_secret: null
+    server: '109469702010478592' # secret hiding room
+    channel: '316959344681943040' # food
 twitch: # set this to null to disable
     client_id: ''
     client_secret: ''

mock_cmd.py

@@ -2,7 +2,10 @@
 
 from pprint import pprint
 
-import animal_crossing
+import requests
+
+import config
+import twitter
 
 class MockCmd:
 	def __init__(self):
@@ -26,6 +29,14 @@ def __init__(self):
 	def send_message(self, channel_id, text, embed=None, files=None):
 		print(channel_id, text, embed, files)
 
+	def get(self, path):
+		response = requests.get('https://discord.com/api' + path, headers={
+			'Authorization': 'Bot ' + config.bot.token,
+			'User-Agent': 'DiscordBot (https://github.com/raylu/sbot 0.0)',
+		})
+		response.raise_for_status()
+		return response.json()
+
 class MockGuild:
 	def __init__(self):
 		self.roles = {
@@ -34,4 +45,4 @@ def __init__(self):
 			'cats': {'position': 2, 'name': 'cats', 'color': 13369480, 'id': '2222'},
 		}
 
-animal_crossing.stalk_market(MockCmd())
+twitter.post(MockBot())

twitter.py

@@ -1,5 +1,9 @@
 import base64
+import copy
+import hashlib
+import mimetypes
 import hmac
+import os
 import time
 import urllib.parse
 
@@ -47,13 +51,93 @@ def new_tweets(bot):
 		config.state.tweet_ids[account] = tweets[0]['id']
 	config.state.save()
 
+def post(bot):
+	rs = requests.Session()
+	message_id = '316960031314804737'
+	channel = config.bot.twitter_post['channel']
+	message = bot.get('/channels/%s/messages/%s' % (channel, message_id))
+	media_ids = []
+	upload_url = 'https://upload.twitter.com/1.1/media/upload.json'
+	for attachment in message['attachments'][:4]:
+		media = rs.get(attachment['url']).content
+
+		media_type, _ = mimetypes.guess_type(attachment['filename'])
+		rs = requests.Session()
+		response = signed_request(rs, 'POST', upload_url, params={
+			'command': 'INIT',
+			'media_type': media_type,
+			'total_bytes': str(attachment['size']),
+		})
+		media_id = response.json()['media_id_string']
+
+		response = signed_request(rs, 'POST', upload_url, params={
+			'command': 'APPEND',
+			'media_id': media_id,
+			'segment_index': '0',
+		}, files={'media': media})
+
+		response = signed_request(rs, 'POST', upload_url, params={
+			'command': 'FINALIZE',
+			'media_id': media_id,
+		})
+
+		media_ids.append(media_id)
+
+	discord_link = 'https://discord.com/channels/%s/%s/%s' % (
+			config.bot.twitter_post['server'], config.bot.twitter_post['channel'], message_id)
+	tweet = '%s on %s\n%s' % (message['author']['username'], message['timestamp'][:10], discord_link)
+	response = signed_request(rs, 'POST', 'https://api.twitter.com/1.1/statuses/update.json', {
+		'status': tweet,
+		'media_ids': ','.join(media_ids),
+		'trim_user': '1',
+	}, {})
+
 def tweet_id_to_ts(tweet_id):
 	# https://github.com/client9/snowflake2time#snowflake-layout
 	return (tweet_id & 0x7fffffffffffffff) >> 22
 
-def sign(method, url, params, consumer_secret, token_secret):
+def signed_request(rs, method, url, params=None, data=None, files=None):
+	signing_params = {
+		'oauth_consumer_key': config.bot.twitter_post['consumer_key'],
+		'oauth_nonce': base64.b64encode(os.getrandom(16)),
+		'oauth_signature_method': 'HMAC-SHA1',
+		'oauth_timestamp': str(int(time.time())),
+		'oauth_version': '1.0',
+		'oauth_token': config.bot.twitter_post['token'],
+	}
+	final_data = None
+	headers = {}
+	if files:
+		# https://github.com/oauthlib/oauthlib/blob/bda81b3cb6306dec19a6e60113e21b2933d0950c/oauthlib/oauth1/rfc5849/__init__.py#L212
+		request = requests.Request(method, url, data=data, files=files).prepare()
+		signing_params['oauth_body_hash'] = base64.b64encode(hashlib.sha1(request.body).digest())
+		final_data = request.body
+		headers['Content-Type'] = request.headers['Content-Type']
+	auth_params = copy.copy(signing_params)
+	if params:
+		signing_params.update(params)
+	if data:
+		signing_params.update(data)
+		if not files:
+			final_data = data
+
+	consumer_secret = config.bot.twitter_post['consumer_secret']
+	token_secret = config.bot.twitter_post['token_secret']
+	auth_params['oauth_signature'] = sign(method, url,
+			signing_params, consumer_secret, token_secret)
+	# https://developer.twitter.com/en/docs/authentication/oauth-1-0a/authorizing-a-request
+	auth = 'OAuth '
+	auth += ', '.join('%s="%s"' % (k, urllib.parse.quote(v)) for k, v in auth_params.items())
+	headers['Authorization'] = auth
+	response = rs.request(method, url, params=params, data=final_data, headers=headers)
+	response.raise_for_status()
+	return response
+
+def sign(method, url, signing_params, consumer_secret, token_secret):
 	# https://developer.twitter.com/en/docs/authentication/oauth-1-0a/creating-a-signature
-	parameter_string = urllib.parse.urlencode(sorted(params.items()), quote_via=urllib.parse.quote)
+	# https://github.com/oauthlib/oauthlib/blob/d54965b86ce4ede956db70baff0b3d5e9182a007/oauthlib/oauth1/rfc5849/utils.py#L52
+	parameter_string = urllib.parse.urlencode(sorted(signing_params.items()),
+			quote_via=urllib.parse.quote, safe='~')
 	# that's right! we re-quote the parameter string
 	encoded_params = urllib.parse.quote_plus(parameter_string)
 	base_string = '%s&%s&%s' % (method, urllib.parse.quote_plus(url), encoded_params)

twitter_key.py

@@ -19,7 +19,6 @@
 	print(r.content)
 elif mode == '2':
 	# https://developer.twitter.com/en/docs/authentication/oauth-1-0a/pin-based-oauth
-	# https://developer.twitter.com/en/docs/authentication/api-reference/request_token
 	params = {
 		'oauth_callback': 'oob',
 		'oauth_consumer_key': consumer_key,
@@ -28,6 +27,7 @@
 		'oauth_timestamp': str(int(time.time())),
 		'oauth_version': '1.0',
 	}
+	# https://developer.twitter.com/en/docs/authentication/api-reference/request_token
 	url = 'https://api.twitter.com/oauth/request_token'
 	params['oauth_signature'] = twitter.sign('POST', url, params, consumer_secret, '')
 	# https://developer.twitter.com/en/docs/authentication/oauth-1-0a/authorizing-a-request