85 lines
2.6 KiB
TypeScript
85 lines
2.6 KiB
TypeScript
import { WxTokenResponse, fetchToken, getUserInfo } from '@/modules/wx.ts';
|
||
import { useContextKey } from '@kevisual/use-config/context';
|
||
import { UserModel } from '@kevisual/code-center-module';
|
||
import { Buffer } from 'buffer';
|
||
|
||
const User = useContextKey<typeof UserModel>('UserModel');
|
||
export class WxServices {
|
||
token: WxTokenResponse;
|
||
// 创建一个webToken，用户登录
|
||
webToken: string;
|
||
isNew: boolean;
|
||
// @ts-ignore
|
||
user: User;
|
||
constructor() {
|
||
//
|
||
}
|
||
async login(code: string) {
|
||
const token = await fetchToken(code);
|
||
this.token = token;
|
||
if (!token.unionid) {
|
||
throw new Error('unionid is required');
|
||
}
|
||
const unionid = token.unionid;
|
||
let user = await User.findOne({ where: { username: unionid } });
|
||
if (!user) {
|
||
user = await User.createUser(unionid, unionid.slice(0, 8));
|
||
this.isNew = true;
|
||
}
|
||
const tokenInfo = await user.createToken(null, 'plugin');
|
||
this.webToken = tokenInfo.token;
|
||
this.user = user;
|
||
return this.webToken;
|
||
}
|
||
|
||
async checkHasUser() {}
|
||
async getUserInfo() {
|
||
const userInfo = await getUserInfo(this.token.access_token, this.token.openid);
|
||
const { nickname, headimgurl } = userInfo;
|
||
this.user.nickname = nickname;
|
||
try {
|
||
const downloadImgUrl = await this.downloadImg(headimgurl);
|
||
this.user.avatar = downloadImgUrl;
|
||
} catch (error) {
|
||
console.error('Error downloading or converting image:', error);
|
||
}
|
||
await this.user.save();
|
||
}
|
||
/**
|
||
* 转成base64
|
||
* @param url
|
||
*/
|
||
async downloadImg(url: string): Promise<string> {
|
||
try {
|
||
return await downloadImag(url);
|
||
} catch (error) {
|
||
console.error('Error downloading or converting image:', error);
|
||
throw error;
|
||
}
|
||
}
|
||
}
|
||
|
||
// https://thirdwx.qlogo.cn/mmopen/vi_32/WvOPpbDwUKEVJvSst8Z91Y68m7CsBeecMqRGlqey5HejByePD89boYGaVCM8vESsYmokk1jABUDsK08IrfI6JEkibZkDIC2zsb96DGBTEF7E/132
|
||
export const downloadImag = async (url: string) => {
|
||
try {
|
||
const res = await fetch(url);
|
||
if (!res.ok) {
|
||
throw new Error(`HTTP error! Status: ${res.status}`);
|
||
}
|
||
const arrayBufferToBase64 = (buffer: ArrayBuffer): string => {
|
||
const bytes = new Uint8Array(buffer);
|
||
let binary = '';
|
||
for (let i = 0; i < bytes.byteLength; i++) {
|
||
binary += String.fromCharCode(bytes[i]);
|
||
}
|
||
return Buffer.from(binary, 'binary').toString('base64');
|
||
};
|
||
|
||
const buffer = await res.arrayBuffer();
|
||
return `data:image/jpeg;base64,${arrayBufferToBase64(buffer)}`;
|
||
} catch (error) {
|
||
console.error('Error downloading or converting image:', error);
|
||
throw error;
|
||
}
|
||
};
|