mirror of https://github.com/msgbyte/tailchat
feat(server): add getuiclient lib and plugin
moonrailgun
2 years ago
parent
82a08d65ef
commit
1605b2d186
@ -0,0 +1,167 @@
|
||||
import got from 'got';
|
||||
import crypto from 'crypto';
|
||||
import _ from 'lodash';
|
||||
|
||||
function sha256(data: string) {
|
||||
return crypto.createHash('sha256').update(data).digest('hex');
|
||||
}
|
||||
|
||||
interface AuthResPayload {
|
||||
msg: string;
|
||||
code: number;
|
||||
data: {
|
||||
expire_time: string;
|
||||
token: string;
|
||||
};
|
||||
}
|
||||
|
||||
interface SinglePushResPayload {
|
||||
msg: string;
|
||||
code: number;
|
||||
data: {
|
||||
[taskid: string]: {
|
||||
[cid: string]:
|
||||
| 'successed_offline'
|
||||
| 'successed_online'
|
||||
| 'successed_ignore';
|
||||
};
|
||||
};
|
||||
}
|
||||
|
||||
interface AllPushResPayload {
|
||||
msg: string;
|
||||
code: number;
|
||||
data: {
|
||||
[taskid: string]: string;
|
||||
};
|
||||
}
|
||||
|
||||
export class GetuiClient {
|
||||
token: string;
|
||||
expireTime: number;
|
||||
|
||||
constructor(
|
||||
public appId: string,
|
||||
public appkey: string,
|
||||
public mastersecret: string
|
||||
) {}
|
||||
|
||||
get baseUrl() {
|
||||
return `https://restapi.getui.com/v2/${this.appId}`;
|
||||
}
|
||||
|
||||
/**
|
||||
* Generate Request ID with fixed prefix and timestamp and random number
|
||||
*/
|
||||
generateRequestId(): string {
|
||||
return (
|
||||
'tailchat' +
|
||||
String(Date.now()) +
|
||||
_.padStart(String(Math.floor(Math.random() * 1e8)), 8, '0')
|
||||
);
|
||||
}
|
||||
|
||||
signBody() {
|
||||
const timestamp = String(Date.now());
|
||||
|
||||
return {
|
||||
sign: sha256(this.appkey + String(Date.now()) + this.mastersecret),
|
||||
timestamp,
|
||||
appkey: this.appkey,
|
||||
};
|
||||
}
|
||||
|
||||
private async fetchToken() {
|
||||
try {
|
||||
const res = await got(`${this.baseUrl}/auth`, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'content-type': 'application/json;charset=utf-8',
|
||||
},
|
||||
json: {
|
||||
...this.signBody(),
|
||||
},
|
||||
}).json<AuthResPayload>();
|
||||
|
||||
if (res.code === 0) {
|
||||
this.token = res.data.token;
|
||||
this.expireTime = Number(res.data.expire_time) - 60 * 1000; // 提前60s过期
|
||||
}
|
||||
} catch (err) {
|
||||
console.error(err);
|
||||
}
|
||||
}
|
||||
|
||||
async getToken(): Promise<string> {
|
||||
if (!this.token || Date.now() > this.expireTime) {
|
||||
// 如果token不存在或者token过期
|
||||
await this.fetchToken();
|
||||
}
|
||||
|
||||
return this.token;
|
||||
}
|
||||
|
||||
async singlePush(
|
||||
userId: string,
|
||||
title: string,
|
||||
body: string,
|
||||
payload: {}
|
||||
): Promise<SinglePushResPayload> {
|
||||
const token = await this.getToken();
|
||||
const requestId = this.generateRequestId();
|
||||
const res = await got(`${this.baseUrl}/push/single/alias`, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'content-type': 'application/json;charset=utf-8',
|
||||
token,
|
||||
},
|
||||
json: {
|
||||
request_id: requestId,
|
||||
audience: {
|
||||
alias: [userId],
|
||||
},
|
||||
push_message: {
|
||||
notification: {
|
||||
title: title,
|
||||
body: body,
|
||||
click_type: 'payload',
|
||||
payload: JSON.stringify(payload),
|
||||
},
|
||||
},
|
||||
},
|
||||
}).json<SinglePushResPayload>();
|
||||
|
||||
return res;
|
||||
}
|
||||
|
||||
async allPush(
|
||||
title: string,
|
||||
body: string,
|
||||
payload: {}
|
||||
): Promise<AllPushResPayload> {
|
||||
const token = await this.getToken();
|
||||
console.log('token', token);
|
||||
const requestId = this.generateRequestId();
|
||||
const res = await got(`${this.baseUrl}/push/all`, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'content-type': 'application/json;charset=utf-8',
|
||||
token,
|
||||
},
|
||||
json: {
|
||||
request_id: requestId,
|
||||
audience: 'all',
|
||||
push_message: {
|
||||
notification: {
|
||||
title: title,
|
||||
body: body,
|
||||
click_type: 'payload',
|
||||
payload: JSON.stringify(payload),
|
||||
},
|
||||
},
|
||||
},
|
||||
}).json<AllPushResPayload>();
|
||||
|
||||
return res;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,20 @@
|
||||
import { db } from 'tailchat-server-sdk';
|
||||
const { getModelForClass, prop, modelOptions, TimeStamps } = db;
|
||||
|
||||
@modelOptions({
|
||||
options: {
|
||||
customName: 'p_getui_log',
|
||||
},
|
||||
})
|
||||
export class GetuiLog extends TimeStamps implements db.Base {
|
||||
_id: db.Types.ObjectId;
|
||||
id: string;
|
||||
}
|
||||
|
||||
export type GetuiLogDocument = db.DocumentType<GetuiLog>;
|
||||
|
||||
const model = getModelForClass(GetuiLog);
|
||||
|
||||
export type GetuiLogModel = typeof model;
|
||||
|
||||
export default model;
|
||||
@ -0,0 +1,18 @@
|
||||
{
|
||||
"name": "tailchat-plugin-getui",
|
||||
"version": "1.0.0",
|
||||
"main": "index.js",
|
||||
"author": "moonrailgun",
|
||||
"description": "Support Getui Notify in chinese mainland",
|
||||
"license": "MIT",
|
||||
"private": true,
|
||||
"scripts": {},
|
||||
"dependencies": {
|
||||
"got": "^11.8.3",
|
||||
"lodash": "^4.17.21",
|
||||
"tailchat-server-sdk": "*"
|
||||
},
|
||||
"devDependencies": {
|
||||
"@types/lodash": "^4.14.191"
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,20 @@
|
||||
import { TcService, TcDbService } from 'tailchat-server-sdk';
|
||||
import type { GetuiLogDocument, GetuiLogModel } from '../models/log';
|
||||
|
||||
/**
|
||||
* Support Getui Notify in chinese mainland
|
||||
*/
|
||||
interface GetuiService
|
||||
extends TcService,
|
||||
TcDbService<GetuiLogDocument, GetuiLogModel> {}
|
||||
class GetuiService extends TcService {
|
||||
get serviceName() {
|
||||
return 'plugin:com.msgbyte.getui';
|
||||
}
|
||||
|
||||
onInit() {
|
||||
this.registerLocalDb(require('../models/log').default);
|
||||
}
|
||||
}
|
||||
|
||||
export default GetuiService;
|
||||
@ -0,0 +1,13 @@
|
||||
import * as dotenv from 'dotenv';
|
||||
dotenv.config();
|
||||
import { GetuiClient } from '../../../plugins/com.msgbyte.getui/lib/GetuiClient';
|
||||
|
||||
const client = new GetuiClient(
|
||||
process.env.GETUI_APPID,
|
||||
process.env.GETUI_APPKEY,
|
||||
process.env.GETUI_MASTERSECRET
|
||||
);
|
||||
|
||||
client.allPush('title', 'body', {}).then((res) => {
|
||||
console.log('res', res);
|
||||
});
|
||||
Loading…
Reference in New Issue