iceshrimp/src/api/service/twitter.ts

import * as express from 'express';
import * as cookie from 'cookie';
import * as uuid from 'uuid';
// import * as Twitter from 'twitter';
// const Twitter = require('twitter');
import autwh from 'autwh';
import redis from '../../db/redis';
import User from '../models/user';
import serialize from '../serializers/user';
import event from '../event';
import config from '../../conf';
import signin from '../common/signin';

module.exports = (app: express.Application) => {
	app.get('/disconnect/twitter', async (req, res): Promise<any> => {
		if (res.locals.user == null) return res.send('plz signin');
		const user = await User.findOneAndUpdate({
			token: res.locals.user
		}, {
				$set: {
					twitter: null
				}
			});

		res.send(`Twitterの連携を解除しました :v:`);

		// Publish i updated event
		event(user._id, 'i_updated', await serialize(user, user, {
			detail: true,
			includeSecrets: true
		}));
	});

	if (config.twitter == null) {
		app.get('/connect/twitter', (req, res) => {
			res.send('現在Twitterへ接続できません (このインスタンスではTwitterはサポートされていません)');
		});

		app.get('/signin/twitter', (req, res) => {
			res.send('現在Twitterへ接続できません (このインスタンスではTwitterはサポートされていません)');
		});

		return;
	}

	const twAuth = autwh({
		consumerKey: config.twitter.consumer_key,
		consumerSecret: config.twitter.consumer_secret,
		callbackUrl: `${config.api_url}/tw/cb`
	});

	const twAuthSignin = autwh({
		consumerKey: config.twitter.consumer_key,
		consumerSecret: config.twitter.consumer_secret,
		callbackUrl: `${config.api_url}/signin/twitter/cb`
	});

	app.get('/connect/twitter', async (req, res): Promise<any> => {
		if (res.locals.user == null) return res.send('plz signin');
		const ctx = await twAuth.begin();
		redis.set(res.locals.user, JSON.stringify(ctx));
		res.redirect(ctx.url);
	});

	app.get('/tw/cb', (req, res): any => {
		if (res.locals.user == null) return res.send('plz signin');
		redis.get(res.locals.user, async (_, ctx) => {
			const result = await twAuth.done(JSON.parse(ctx), req.query.oauth_verifier);

			const user = await User.findOneAndUpdate({
				token: res.locals.user
			}, {
				$set: {
					twitter: {
						access_token: result.accessToken,
						access_token_secret: result.accessTokenSecret,
						user_id: result.userId,
						screen_name: result.screenName
					}
				}
			});

			res.send(`Twitter: @${result.screenName} を、Misskey: @${user.username} に接続しました！`);

			// Publish i updated event
			event(user._id, 'i_updated', await serialize(user, user, {
				detail: true,
				includeSecrets: true
			}));
		});
	});

	app.get('/signin/twitter', async (req, res): Promise<any> => {
		const ctx = await twAuthSignin.begin();

		const sessid = uuid();

		redis.set(sessid, JSON.stringify(ctx));

		const expires = 1000 * 60 * 60; // 1h
		res.cookie('signin_with_twitter_session_id', sessid, {
			path: '/',
			domain: `.${config.host}`,
			secure: config.url.substr(0, 5) === 'https',
			httpOnly: true,
			expires: new Date(Date.now() + expires),
			maxAge: expires
		});

		res.redirect(ctx.url);
	});

	app.get('/signin/twitter/cb', (req, res): any => {
		// req.headers['cookie'] は常に string ですが、型定義の都合上
		// string | string[] になっているので string を明示しています
		const cookies = cookie.parse((req.headers['cookie'] as string || ''));

		const sessid = cookies['signin_with_twitter_session_id'];

		if (sessid == undefined) {
			res.status(400).send('invalid session');
		}

		redis.get(sessid, async (_, ctx) => {
			const result = await twAuthSignin.done(JSON.parse(ctx), req.query.oauth_verifier);

			const user = await User.findOne({
				'twitter.user_id': result.userId
			});

			if (user == null) {
				res.status(404).send(`@${result.screenName}と連携しているMisskeyアカウントはありませんでした...`);
			}

			signin(res, user, true);
		});
	});
};