wip

package.json

@@ -3,7 +3,8 @@
 	"main": "./built/index.js",
 	"scripts": {
 		"start": "node ./built",
-    "build": "tsc"
+		"build": "tsc",
+		"test": "jest"
 	},
 	"dependencies": {
 		"@types/chalk": "2.2.0",
@@ -22,6 +23,7 @@
 		"lokijs": "1.5.11",
 		"memory-streams": "0.1.3",
 		"misskey-reversi": "0.0.5",
+		"module-alias": "2.2.2",
 		"promise-retry": "2.0.1",
 		"random-seed": "0.3.0",
 		"reconnecting-websocket": "4.4.0",
@@ -34,5 +36,32 @@
 		"typescript": "4.0.2",
 		"uuid": "8.3.0",
 		"ws": "7.3.1"
+	},
+	"devDependencies": {
+		"@types/jest": "26.0.14",
+		"jest": "26.4.2",
+		"ts-jest": "26.3.0"
+	},
+	"_moduleAliases": {
+		"@": "built"
+	},
+	"jest": {
+		"testRegex": "/test/.*",
+		"moduleFileExtensions": [
+			"ts",
+			"js"
+		],
+		"transform": {
+			"^.+\\.ts$": "ts-jest"
+		},
+		"globals": {
+			"ts-jest": {
+				"tsConfig": "test/tsconfig.json"
+			}
+		},
+		"moduleNameMapper": {
+			"^@/(.+)": "<rootDir>/src/$1",
+			"^#/(.+)": "<rootDir>/test/$1"
+		}
 	}
 }

src/ai.ts

@@ -80,9 +80,11 @@ export default class 藍 {
 		this.account = account;
 		this.modules = modules;
 
-		this.log('Lodaing the memory...');
+		const file = process.env.NODE_ENV === 'test' ? 'test.memory.json' : 'memory.json';
 
-		this.db = new loki('memory.json', {
+		this.log(`Lodaing the memory from ${file}...`);
+
+		this.db = new loki(file, {
 			autoload: true,
 			autosave: true,
 			autosaveInterval: 1000,

src/index.ts

@@ -1,5 +1,7 @@
 // AiOS bootstrapper
 
+import 'module-alias/register';
+
 import * as chalk from 'chalk';
 import * as request from 'request-promise-native';
 const promiseRetry = require('promise-retry');

src/misskey/user.ts

@@ -2,7 +2,7 @@ export type User = {
 	id: string;
 	name: string;
 	username: string;
-	host: string;
-	isFollowing: boolean;
+	host?: string | null;
+	isFollowing?: boolean;
 	isBot: boolean;
 };

src/modules/birthday/index.ts

@@ -1,7 +1,7 @@
 import autobind from 'autobind-decorator';
-import Module from '../../module';
+import Module from '@/module';
 import Friend from '../../friend';
-import serifs from '../../serifs';
+import serifs from '@/serifs';
 
 function zeroPadding(num: number, length: number): string {
 	return ('0000000000' + num).slice(-length);

src/modules/chart/index.ts

@@ -1,7 +1,7 @@
 import autobind from 'autobind-decorator';
-import Module from '../../module';
-import serifs from '../../serifs';
-import Message from '../../message';
+import Module from '@/module';
+import serifs from '@/serifs';
+import Message from '@/message';
 import { renderChart } from './render-chart';
 import { items } from '../../vocabulary';
 import config from '../../config';

src/modules/core/index.ts

@@ -1,7 +1,7 @@
 import autobind from 'autobind-decorator';
-import Module from '../../module';
-import Message from '../../message';
-import serifs from '../../serifs';
+import Module from '@/module';
+import Message from '@/message';
+import serifs from '@/serifs';
 import { safeForInterpolate } from '../../utils/safe-for-interpolate';
 
 const titles = ['さん', 'くん', '君', 'ちゃん', '様', '先生'];

src/modules/dice/index.ts

@@ -1,7 +1,7 @@
 import autobind from 'autobind-decorator';
-import Module from '../../module';
-import Message from '../../message';
-import serifs from '../../serifs';
+import Module from '@/module';
+import Message from '@/message';
+import serifs from '@/serifs';
 
 export default class extends Module {
 	public readonly name = 'dice';

src/modules/emoji-react/index.ts

@@ -2,7 +2,7 @@ import autobind from 'autobind-decorator';
 import { parse } from 'twemoji-parser';
 
 import { Note } from '../../misskey/note';
-import Module from '../../module';
+import Module from '@/module';
 import Stream from '../../stream';
 import includes from '../../utils/includes';
 

src/modules/emoji/index.ts

@@ -1,7 +1,7 @@
 import autobind from 'autobind-decorator';
-import Module from '../../module';
-import Message from '../../message';
-import serifs from '../../serifs';
+import Module from '@/module';
+import Message from '@/message';
+import serifs from '@/serifs';
 
 const hands = [
 	'👏',

src/modules/follow/index.ts

@@ -1,6 +1,6 @@
 import autobind from 'autobind-decorator';
-import Module from '../../module';
-import Message from '../../message';
+import Module from '@/module';
+import Message from '@/message';
 
 export default class extends Module {
 	public readonly name = 'follow';

src/modules/fortune/index.ts

@@ -1,7 +1,7 @@
 import autobind from 'autobind-decorator';
-import Module from '../../module';
-import Message from '../../message';
-import serifs from '../../serifs';
+import Module from '@/module';
+import Message from '@/message';
+import serifs from '@/serifs';
 import * as seedrandom from 'seedrandom';
 import { genItem } from '../../vocabulary';
 

src/modules/guessing-game/index.ts

@@ -1,8 +1,8 @@
 import autobind from 'autobind-decorator';
 import * as loki from 'lokijs';
-import Module from '../../module';
-import Message from '../../message';
-import serifs from '../../serifs';
+import Module from '@/module';
+import Message from '@/message';
+import serifs from '@/serifs';
 
 export default class extends Module {
 	public readonly name = 'guessingGame';

src/modules/kazutori/index.ts

@@ -1,18 +1,17 @@
 import autobind from 'autobind-decorator';
 import * as loki from 'lokijs';
-import Module from '../../module';
-import Message from '../../message';
-import serifs from '../../serifs';
-
-type User = {
-	id: string;
-	username: string;
-	host: string;
-};
+import Module from '@/module';
+import Message from '@/message';
+import serifs from '@/serifs';
+import { User } from '@/misskey/user';
 
 type Game = {
 	votes: {
-		user: User;
+		user: {
+			id: string;
+			username: string;
+			host: User['host'];
+		};
 		number: number;
 	}[];
 	isEnded: boolean;
@@ -173,14 +172,14 @@ export default class extends Module {
 			return;
 		}
 
-		function acct(user: User): string {
+		function acct(user: Game['votes'][0]['user']): string {
 			return user.host
 				? `@${user.username}@${user.host}`
 				: `@${user.username}`;
 		}
 
 		let results: string[] = [];
-		let winner: User | null = null;
+		let winner: Game['votes'][0]['user'] | null = null;
 
 		for (let i = 100; i >= 0; i--) {
 			const users = game.votes

src/modules/keyword/index.ts

@@ -1,8 +1,8 @@
 import autobind from 'autobind-decorator';
 import * as loki from 'lokijs';
-import Module from '../../module';
+import Module from '@/module';
 import config from '../../config';
-import serifs from '../../serifs';
+import serifs from '@/serifs';
 import { mecab } from './mecab';
 
 function kanaToHira(str: string) {

src/modules/maze/index.ts

@@ -1,9 +1,9 @@
 import autobind from 'autobind-decorator';
-import Module from '../../module';
-import serifs from '../../serifs';
+import Module from '@/module';
+import serifs from '@/serifs';
 import { genMaze } from './gen-maze';
 import { renderMaze } from './render-maze';
-import Message from '../../message';
+import Message from '@/message';
 
 export default class extends Module {
 	public readonly name = 'maze';

src/modules/noting/index.ts

@@ -1,6 +1,6 @@
 import autobind from 'autobind-decorator';
-import Module from '../../module';
-import serifs from '../../serifs';
+import Module from '@/module';
+import serifs from '@/serifs';
 import { genItem } from '../../vocabulary';
 
 export default class extends Module {

src/modules/ping/index.ts

@@ -1,6 +1,6 @@
 import autobind from 'autobind-decorator';
-import Module from '../../module';
-import Message from '../../message';
+import Module from '@/module';
+import Message from '@/message';
 
 export default class extends Module {
 	public readonly name = 'ping';

src/modules/poll/index.ts

@@ -1,7 +1,7 @@
 import autobind from 'autobind-decorator';
-import Message from '../../message';
-import Module from '../../module';
-import serifs from '../../serifs';
+import Message from '@/message';
+import Module from '@/module';
+import serifs from '@/serifs';
 import { genItem } from '../../vocabulary';
 import config from '../../config';
 import { Note } from '../../misskey/note';

src/modules/reversi/back.ts

@@ -9,7 +9,7 @@
 import * as request from 'request-promise-native';
 import Reversi, { Color } from 'misskey-reversi';
 import config from '../../config';
-import serifs from '../../serifs';
+import serifs from '@/serifs';
 import { User } from '../../misskey/user';
 
 const db = {};

src/modules/reversi/index.ts

@@ -1,9 +1,9 @@
 import * as childProcess from 'child_process';
 import autobind from 'autobind-decorator';
-import Module from '../../module';
-import serifs from '../../serifs';
+import Module from '@/module';
+import serifs from '@/serifs';
 import config from '../../config';
-import Message from '../../message';
+import Message from '@/message';
 import Friend from '../../friend';
 import getDate from '../../utils/get-date';
 

src/modules/server/index.ts

@@ -1,6 +1,6 @@
 import autobind from 'autobind-decorator';
-import Module from '../../module';
-import serifs from '../../serifs';
+import Module from '@/module';
+import serifs from '@/serifs';
 import config from '../../config';
 
 export default class extends Module {

src/modules/sleep-report/index.ts

@@ -1,6 +1,6 @@
 import autobind from 'autobind-decorator';
-import Module from '../../module';
-import serifs from '../../serifs';
+import Module from '@/module';
+import serifs from '@/serifs';
 
 export default class extends Module {
 	public readonly name = 'sleepReport';

src/modules/talk/index.ts

@@ -1,7 +1,7 @@
 import autobind from 'autobind-decorator';
 import { HandlerResult } from '../../ai';
-import Module from '../../module';
-import Message from '../../message';
+import Module from '@/module';
+import Message from '@/message';
 import serifs, { getSerif } from '../../serifs';
 import getDate from '../../utils/get-date';
 

src/modules/timer/index.ts

@@ -1,7 +1,7 @@
 import autobind from 'autobind-decorator';
-import Module from '../../module';
-import Message from '../../message';
-import serifs from '../../serifs';
+import Module from '@/module';
+import Message from '@/message';
+import serifs from '@/serifs';
 
 export default class extends Module {
 	public readonly name = 'timer';

src/modules/valentine/index.ts

@@ -1,7 +1,7 @@
 import autobind from 'autobind-decorator';
-import Module from '../../module';
+import Module from '@/module';
 import Friend from '../../friend';
-import serifs from '../../serifs';
+import serifs from '@/serifs';
 
 export default class extends Module {
 	public readonly name = 'valentine';

src/modules/welcome/index.ts

@@ -1,5 +1,5 @@
 import autobind from 'autobind-decorator';
-import Module from '../../module';
+import Module from '@/module';
 
 export default class extends Module {
 	public readonly name = 'welcome';

test/_mocks_/account.ts

@@ -0,0 +1,7 @@
+export const account = {
+	id: '0',
+	name: '藍',
+	username: 'ai',
+	host: null,
+	isBot: true,
+};

test/_mocks_/ws.ts

@@ -0,0 +1,17 @@
+import * as websocket from 'websocket';
+
+export class StreamingApi {
+	private ws: WS;
+
+	constructor() {
+		this.ws = new WS('ws://localhost/streaming');
+	}
+
+	public async waitForMainChannelConnected() {
+		await expect(this.ws).toReceiveMessage("hello");
+	}
+
+	public send(message) {
+		this.ws.send(JSON.stringify(message));
+	}
+}

test/_modules_/test.ts

@@ -0,0 +1,26 @@
+import autobind from 'autobind-decorator';
+import Module from '@/module';
+import Message from '@/message';
+
+export default class extends Module {
+	public readonly name = 'test';
+
+	@autobind
+	public install() {
+		return {
+			mentionHook: this.mentionHook
+		};
+	}
+
+	@autobind
+	private async mentionHook(msg: Message) {
+		if (msg.text && msg.text.includes('ping')) {
+			msg.reply('PONG!', {
+				immediate: true
+			});
+			return true;
+		} else {
+			return false;
+		}
+	}
+}

test/core.ts

@@ -0,0 +1,20 @@
+import 藍 from '@/ai';
+import { account } from '#/_mocks_/account';
+import TestModule from '#/_modules_/test';
+import { StreamingApi } from '#/_mocks_/ws';
+
+process.env.NODE_ENV = 'test';
+
+let ai: 藍;
+
+beforeEach(() => {
+  ai = new 藍(account, [
+		new TestModule(),
+	]);
+});
+
+test('mention hook', async () => {
+	const streaming = new StreamingApi();
+
+	
+});

test/tsconfig.json

@@ -0,0 +1,15 @@
+{
+	"extends": "../tsconfig.json",
+	"compilerOptions": {
+		"baseUrl": ".",
+		"rootDir": "../",
+		"paths": {
+			"@/*": ["../src/*"],
+			"#/*": ["./*"]
+		},
+	},
+	"compileOnSave": false,
+	"include": [
+		"**/*.ts"
+	]
+}

tsconfig.json

@@ -8,12 +8,16 @@
 		"strictNullChecks": true,
 		"experimentalDecorators": true,
 		"sourceMap": false,
-		"target": "es2017",
+		"target": "es2020",
 		"module": "commonjs",
 		"removeComments": false,
 		"noLib": false,
 		"outDir": "built",
-		"rootDir": "src"
+		"rootDir": "src",
+		"baseUrl": ".",
+		"paths": {
+			"@/*": ["src/*"]
+		},
 	},
 	"compileOnSave": false,
 	"include": [