529 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			TypeScript
		
	
	
	
	
	
			
		
		
	
	
			529 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			TypeScript
		
	
	
	
	
	
import autobind from 'autobind-decorator';
 | 
						|
import Vue from 'vue';
 | 
						|
import { EventEmitter } from 'eventemitter3';
 | 
						|
import * as uuid from 'uuid';
 | 
						|
 | 
						|
import initStore from './store';
 | 
						|
import { apiUrl, clientVersion as version, lang } from './config';
 | 
						|
import Progress from './common/scripts/loading';
 | 
						|
 | 
						|
import Err from './common/views/components/connect-failed.vue';
 | 
						|
import Stream from './common/scripts/stream';
 | 
						|
 | 
						|
//#region api requests
 | 
						|
let spinner = null;
 | 
						|
let pending = 0;
 | 
						|
//#endregion
 | 
						|
 | 
						|
/**
 | 
						|
 * Misskey Operating System
 | 
						|
 */
 | 
						|
export default class MiOS extends EventEmitter {
 | 
						|
	/**
 | 
						|
	 * Misskeyの /meta で取得できるメタ情報
 | 
						|
	 */
 | 
						|
	private meta: {
 | 
						|
		data: { [x: string]: any };
 | 
						|
		chachedAt: Date;
 | 
						|
	};
 | 
						|
 | 
						|
	public get instanceName() {
 | 
						|
		return this.meta ? this.meta.data.name : 'Misskey';
 | 
						|
	}
 | 
						|
 | 
						|
	private isMetaFetching = false;
 | 
						|
 | 
						|
	public app: Vue;
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Whether is debug mode
 | 
						|
	 */
 | 
						|
	public get debug() {
 | 
						|
		return this.store ? this.store.state.device.debug : false;
 | 
						|
	}
 | 
						|
 | 
						|
	public store: ReturnType<typeof initStore>;
 | 
						|
 | 
						|
	/**
 | 
						|
	 * A connection manager of home stream
 | 
						|
	 */
 | 
						|
	public stream: Stream;
 | 
						|
 | 
						|
	/**
 | 
						|
	 * A registration of service worker
 | 
						|
	 */
 | 
						|
	private swRegistration: ServiceWorkerRegistration = null;
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Whether should register ServiceWorker
 | 
						|
	 */
 | 
						|
	private shouldRegisterSw: boolean;
 | 
						|
 | 
						|
	/**
 | 
						|
	 * ウィンドウシステム
 | 
						|
	 */
 | 
						|
	public windows = new WindowSystem();
 | 
						|
 | 
						|
	/**
 | 
						|
	 * MiOSインスタンスを作成します
 | 
						|
	 * @param shouldRegisterSw ServiceWorkerを登録するかどうか
 | 
						|
	 */
 | 
						|
	constructor(shouldRegisterSw = false) {
 | 
						|
		super();
 | 
						|
 | 
						|
		this.shouldRegisterSw = shouldRegisterSw;
 | 
						|
 | 
						|
		if (this.debug) {
 | 
						|
			(window as any).os = this;
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	@autobind
 | 
						|
	public log(...args) {
 | 
						|
		if (!this.debug) return;
 | 
						|
		console.log.apply(null, args);
 | 
						|
	}
 | 
						|
 | 
						|
	@autobind
 | 
						|
	public logInfo(...args) {
 | 
						|
		if (!this.debug) return;
 | 
						|
		console.info.apply(null, args);
 | 
						|
	}
 | 
						|
 | 
						|
	@autobind
 | 
						|
	public logWarn(...args) {
 | 
						|
		if (!this.debug) return;
 | 
						|
		console.warn.apply(null, args);
 | 
						|
	}
 | 
						|
 | 
						|
	@autobind
 | 
						|
	public logError(...args) {
 | 
						|
		if (!this.debug) return;
 | 
						|
		console.error.apply(null, args);
 | 
						|
	}
 | 
						|
 | 
						|
	@autobind
 | 
						|
	public signout() {
 | 
						|
		this.store.dispatch('logout');
 | 
						|
		location.href = '/';
 | 
						|
	}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Initialize MiOS (boot)
 | 
						|
	 * @param callback A function that call when initialized
 | 
						|
	 */
 | 
						|
	@autobind
 | 
						|
	public async init(callback) {
 | 
						|
		this.store = initStore(this);
 | 
						|
 | 
						|
		// ユーザーをフェッチしてコールバックする
 | 
						|
		const fetchme = (token, cb) => {
 | 
						|
			let me = null;
 | 
						|
 | 
						|
			// Return when not signed in
 | 
						|
			if (token == null) {
 | 
						|
				return done();
 | 
						|
			}
 | 
						|
 | 
						|
			// Fetch user
 | 
						|
			fetch(`${apiUrl}/i`, {
 | 
						|
				method: 'POST',
 | 
						|
				body: JSON.stringify({
 | 
						|
					i: token
 | 
						|
				})
 | 
						|
			})
 | 
						|
			// When success
 | 
						|
			.then(res => {
 | 
						|
				// When failed to authenticate user
 | 
						|
				if (res.status !== 200 && res.status < 500) {
 | 
						|
					return this.signout();
 | 
						|
				}
 | 
						|
 | 
						|
				// Parse response
 | 
						|
				res.json().then(i => {
 | 
						|
					me = i;
 | 
						|
					me.token = token;
 | 
						|
					done();
 | 
						|
				});
 | 
						|
			})
 | 
						|
			// When failure
 | 
						|
			.catch(() => {
 | 
						|
				// Render the error screen
 | 
						|
				document.body.innerHTML = '<div id="err"></div>';
 | 
						|
				new Vue({
 | 
						|
					render: createEl => createEl(Err)
 | 
						|
				}).$mount('#err');
 | 
						|
 | 
						|
				Progress.done();
 | 
						|
			});
 | 
						|
 | 
						|
			function done() {
 | 
						|
				if (cb) cb(me);
 | 
						|
			}
 | 
						|
		};
 | 
						|
 | 
						|
		// フェッチが完了したとき
 | 
						|
		const fetched = () => {
 | 
						|
			this.emit('signedin');
 | 
						|
 | 
						|
			this.initStream();
 | 
						|
 | 
						|
			// Finish init
 | 
						|
			callback();
 | 
						|
 | 
						|
			// Init service worker
 | 
						|
			//if (this.shouldRegisterSw) this.registerSw();
 | 
						|
		};
 | 
						|
 | 
						|
		// キャッシュがあったとき
 | 
						|
		if (this.store.state.i != null) {
 | 
						|
			if (this.store.state.i.token == null) {
 | 
						|
				this.signout();
 | 
						|
				return;
 | 
						|
			}
 | 
						|
 | 
						|
			// とりあえずキャッシュされたデータでお茶を濁して(?)おいて、
 | 
						|
			fetched();
 | 
						|
 | 
						|
			// 後から新鮮なデータをフェッチ
 | 
						|
			fetchme(this.store.state.i.token, freshData => {
 | 
						|
				this.store.dispatch('mergeMe', freshData);
 | 
						|
			});
 | 
						|
		} else {
 | 
						|
			// Get token from cookie or localStorage
 | 
						|
			const i = (document.cookie.match(/i=(!\w+)/) || [null, null])[1] || localStorage.getItem('i');
 | 
						|
 | 
						|
			fetchme(i, me => {
 | 
						|
				if (me) {
 | 
						|
					this.store.dispatch('login', me);
 | 
						|
					fetched();
 | 
						|
				} else {
 | 
						|
					this.initStream();
 | 
						|
 | 
						|
					// Finish init
 | 
						|
					callback();
 | 
						|
				}
 | 
						|
			});
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	@autobind
 | 
						|
	private initStream() {
 | 
						|
		this.stream = new Stream(this);
 | 
						|
 | 
						|
		if (this.store.getters.isSignedIn) {
 | 
						|
			const main = this.stream.useSharedConnection('main');
 | 
						|
 | 
						|
			// 自分の情報が更新されたとき
 | 
						|
			main.on('meUpdated', i => {
 | 
						|
				this.store.dispatch('mergeMe', i);
 | 
						|
			});
 | 
						|
 | 
						|
			main.on('readAllNotifications', () => {
 | 
						|
				this.store.dispatch('mergeMe', {
 | 
						|
					hasUnreadNotification: false
 | 
						|
				});
 | 
						|
			});
 | 
						|
 | 
						|
			main.on('unreadNotification', () => {
 | 
						|
				this.store.dispatch('mergeMe', {
 | 
						|
					hasUnreadNotification: true
 | 
						|
				});
 | 
						|
			});
 | 
						|
 | 
						|
			main.on('readAllMessagingMessages', () => {
 | 
						|
				this.store.dispatch('mergeMe', {
 | 
						|
					hasUnreadMessagingMessage: false
 | 
						|
				});
 | 
						|
			});
 | 
						|
 | 
						|
			main.on('unreadMessagingMessage', () => {
 | 
						|
				this.store.dispatch('mergeMe', {
 | 
						|
					hasUnreadMessagingMessage: true
 | 
						|
				});
 | 
						|
			});
 | 
						|
 | 
						|
			main.on('unreadMention', () => {
 | 
						|
				this.store.dispatch('mergeMe', {
 | 
						|
					hasUnreadMentions: true
 | 
						|
				});
 | 
						|
			});
 | 
						|
 | 
						|
			main.on('readAllUnreadMentions', () => {
 | 
						|
				this.store.dispatch('mergeMe', {
 | 
						|
					hasUnreadMentions: false
 | 
						|
				});
 | 
						|
			});
 | 
						|
 | 
						|
			main.on('unreadSpecifiedNote', () => {
 | 
						|
				this.store.dispatch('mergeMe', {
 | 
						|
					hasUnreadSpecifiedNotes: true
 | 
						|
				});
 | 
						|
			});
 | 
						|
 | 
						|
			main.on('readAllUnreadSpecifiedNotes', () => {
 | 
						|
				this.store.dispatch('mergeMe', {
 | 
						|
					hasUnreadSpecifiedNotes: false
 | 
						|
				});
 | 
						|
			});
 | 
						|
 | 
						|
			main.on('clientSettingUpdated', x => {
 | 
						|
				this.store.commit('settings/set', {
 | 
						|
					key: x.key,
 | 
						|
					value: x.value
 | 
						|
				});
 | 
						|
			});
 | 
						|
 | 
						|
			main.on('homeUpdated', x => {
 | 
						|
				this.store.commit('settings/setHome', x);
 | 
						|
			});
 | 
						|
 | 
						|
			main.on('mobileHomeUpdated', x => {
 | 
						|
				this.store.commit('settings/setMobileHome', x);
 | 
						|
			});
 | 
						|
 | 
						|
			main.on('widgetUpdated', x => {
 | 
						|
				this.store.commit('settings/setWidget', {
 | 
						|
					id: x.id,
 | 
						|
					data: x.data
 | 
						|
				});
 | 
						|
			});
 | 
						|
 | 
						|
			// トークンが再生成されたとき
 | 
						|
			// このままではMisskeyが利用できないので強制的にサインアウトさせる
 | 
						|
			main.on('myTokenRegenerated', () => {
 | 
						|
				alert('%i18n:common.my-token-regenerated%');
 | 
						|
				this.signout();
 | 
						|
			});
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Register service worker
 | 
						|
	 */
 | 
						|
	@autobind
 | 
						|
	private registerSw() {
 | 
						|
		// Check whether service worker and push manager supported
 | 
						|
		const isSwSupported =
 | 
						|
			('serviceWorker' in navigator) && ('PushManager' in window);
 | 
						|
 | 
						|
		// Reject when browser not service worker supported
 | 
						|
		if (!isSwSupported) return;
 | 
						|
 | 
						|
		// Reject when not signed in to Misskey
 | 
						|
		if (!this.store.getters.isSignedIn) return;
 | 
						|
 | 
						|
		// When service worker activated
 | 
						|
		navigator.serviceWorker.ready.then(registration => {
 | 
						|
			this.log('[sw] ready: ', registration);
 | 
						|
 | 
						|
			this.swRegistration = registration;
 | 
						|
 | 
						|
			// Options of pushManager.subscribe
 | 
						|
			// SEE: https://developer.mozilla.org/en-US/docs/Web/API/PushManager/subscribe#Parameters
 | 
						|
			const opts = {
 | 
						|
				// A boolean indicating that the returned push subscription
 | 
						|
				// will only be used for messages whose effect is made visible to the user.
 | 
						|
				userVisibleOnly: true,
 | 
						|
 | 
						|
				// A public key your push server will use to send
 | 
						|
				// messages to client apps via a push server.
 | 
						|
				applicationServerKey: urlBase64ToUint8Array(this.meta.data.swPublickey)
 | 
						|
			};
 | 
						|
 | 
						|
			// Subscribe push notification
 | 
						|
			this.swRegistration.pushManager.subscribe(opts).then(subscription => {
 | 
						|
				this.log('[sw] Subscribe OK:', subscription);
 | 
						|
 | 
						|
				function encode(buffer: ArrayBuffer) {
 | 
						|
					return btoa(String.fromCharCode.apply(null, new Uint8Array(buffer)));
 | 
						|
				}
 | 
						|
 | 
						|
				// Register
 | 
						|
				this.api('sw/register', {
 | 
						|
					endpoint: subscription.endpoint,
 | 
						|
					auth: encode(subscription.getKey('auth')),
 | 
						|
					publickey: encode(subscription.getKey('p256dh'))
 | 
						|
				});
 | 
						|
			})
 | 
						|
			// When subscribe failed
 | 
						|
			.catch(async (err: Error) => {
 | 
						|
				this.logError('[sw] Subscribe Error:', err);
 | 
						|
 | 
						|
				// 通知が許可されていなかったとき
 | 
						|
				if (err.name == 'NotAllowedError') {
 | 
						|
					this.logError('[sw] Subscribe failed due to notification not allowed');
 | 
						|
					return;
 | 
						|
				}
 | 
						|
 | 
						|
				// 違うapplicationServerKey (または gcm_sender_id)のサブスクリプションが
 | 
						|
				// 既に存在していることが原因でエラーになった可能性があるので、
 | 
						|
				// そのサブスクリプションを解除しておく
 | 
						|
				const subscription = await this.swRegistration.pushManager.getSubscription();
 | 
						|
				if (subscription) subscription.unsubscribe();
 | 
						|
			});
 | 
						|
		});
 | 
						|
 | 
						|
		// The path of service worker script
 | 
						|
		const sw = `/sw.${version}.js`;
 | 
						|
 | 
						|
		// Register service worker
 | 
						|
		navigator.serviceWorker.register(sw).then(registration => {
 | 
						|
			// 登録成功
 | 
						|
			this.logInfo('[sw] Registration successful with scope: ', registration.scope);
 | 
						|
		}).catch(err => {
 | 
						|
			// 登録失敗 :(
 | 
						|
			this.logError('[sw] Registration failed: ', err);
 | 
						|
		});
 | 
						|
	}
 | 
						|
 | 
						|
	public requests = [];
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Misskey APIにリクエストします
 | 
						|
	 * @param endpoint エンドポイント名
 | 
						|
	 * @param data パラメータ
 | 
						|
	 */
 | 
						|
	@autobind
 | 
						|
	public api(endpoint: string, data: { [x: string]: any } = {}, silent = false): Promise<{ [x: string]: any }> {
 | 
						|
		if (!silent) {
 | 
						|
			if (++pending === 1) {
 | 
						|
				spinner = document.createElement('div');
 | 
						|
				spinner.setAttribute('id', 'wait');
 | 
						|
				document.body.appendChild(spinner);
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		const onFinally = () => {
 | 
						|
			if (!silent) {
 | 
						|
				if (--pending === 0) spinner.parentNode.removeChild(spinner);
 | 
						|
			}
 | 
						|
		};
 | 
						|
 | 
						|
		const promise = new Promise((resolve, reject) => {
 | 
						|
			// Append a credential
 | 
						|
			if (this.store.getters.isSignedIn) (data as any).i = this.store.state.i.token;
 | 
						|
 | 
						|
			const req = {
 | 
						|
				id: uuid(),
 | 
						|
				date: new Date(),
 | 
						|
				name: endpoint,
 | 
						|
				data,
 | 
						|
				res: null,
 | 
						|
				status: null
 | 
						|
			};
 | 
						|
 | 
						|
			if (this.debug) {
 | 
						|
				this.requests.push(req);
 | 
						|
			}
 | 
						|
 | 
						|
			// Send request
 | 
						|
			fetch(endpoint.indexOf('://') > -1 ? endpoint : `${apiUrl}/${endpoint}`, {
 | 
						|
				method: 'POST',
 | 
						|
				body: JSON.stringify(data),
 | 
						|
				credentials: endpoint === 'signin' ? 'include' : 'omit',
 | 
						|
				cache: 'no-cache'
 | 
						|
			}).then(async (res) => {
 | 
						|
				const body = res.status === 204 ? null : await res.json();
 | 
						|
 | 
						|
				if (this.debug) {
 | 
						|
					req.status = res.status;
 | 
						|
					req.res = body;
 | 
						|
				}
 | 
						|
 | 
						|
				if (res.status === 200) {
 | 
						|
					resolve(body);
 | 
						|
				} else if (res.status === 204) {
 | 
						|
					resolve();
 | 
						|
				} else {
 | 
						|
					reject(body.error);
 | 
						|
				}
 | 
						|
			}).catch(reject);
 | 
						|
		});
 | 
						|
 | 
						|
		promise.then(onFinally, onFinally);
 | 
						|
 | 
						|
		return promise;
 | 
						|
	}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Misskeyのメタ情報を取得します
 | 
						|
	 */
 | 
						|
	@autobind
 | 
						|
	public getMetaSync() {
 | 
						|
		return this.meta ? this.meta.data : null;
 | 
						|
	}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Misskeyのメタ情報を取得します
 | 
						|
	 * @param force キャッシュを無視するか否か
 | 
						|
	 */
 | 
						|
	@autobind
 | 
						|
	public getMeta(force = false) {
 | 
						|
		return new Promise<{ [x: string]: any }>(async (res, rej) => {
 | 
						|
			if (this.isMetaFetching) {
 | 
						|
				this.once('_meta_fetched_', () => {
 | 
						|
					res(this.meta.data);
 | 
						|
				});
 | 
						|
				return;
 | 
						|
			}
 | 
						|
 | 
						|
			const expire = 1000 * 60; // 1min
 | 
						|
 | 
						|
			// forceが有効, meta情報を保持していない or 期限切れ
 | 
						|
			if (force || this.meta == null || Date.now() - this.meta.chachedAt.getTime() > expire) {
 | 
						|
				this.isMetaFetching = true;
 | 
						|
				const meta = await this.api('meta', {
 | 
						|
					detail: false
 | 
						|
				});
 | 
						|
				this.meta = {
 | 
						|
					data: meta,
 | 
						|
					chachedAt: new Date()
 | 
						|
				};
 | 
						|
				this.isMetaFetching = false;
 | 
						|
				this.emit('_meta_fetched_');
 | 
						|
				res(meta);
 | 
						|
			} else {
 | 
						|
				res(this.meta.data);
 | 
						|
			}
 | 
						|
		});
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
class WindowSystem extends EventEmitter {
 | 
						|
	public windows = new Set();
 | 
						|
 | 
						|
	public add(window) {
 | 
						|
		this.windows.add(window);
 | 
						|
		this.emit('added', window);
 | 
						|
	}
 | 
						|
 | 
						|
	public remove(window) {
 | 
						|
		this.windows.delete(window);
 | 
						|
		this.emit('removed', window);
 | 
						|
	}
 | 
						|
 | 
						|
	public getAll() {
 | 
						|
		return this.windows;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Convert the URL safe base64 string to a Uint8Array
 | 
						|
 * @param base64String base64 string
 | 
						|
 */
 | 
						|
function urlBase64ToUint8Array(base64String: string): Uint8Array {
 | 
						|
	const padding = '='.repeat((4 - base64String.length % 4) % 4);
 | 
						|
	const base64 = (base64String + padding)
 | 
						|
		.replace(/\-/g, '+')
 | 
						|
		.replace(/_/g, '/');
 | 
						|
 | 
						|
	const rawData = window.atob(base64);
 | 
						|
	const outputArray = new Uint8Array(rawData.length);
 | 
						|
 | 
						|
	for (let i = 0; i < rawData.length; ++i) {
 | 
						|
		outputArray[i] = rawData.charCodeAt(i);
 | 
						|
	}
 | 
						|
	return outputArray;
 | 
						|
}
 |