105 lines
2.6 KiB
TypeScript
105 lines
2.6 KiB
TypeScript
import {
|
|
Body,
|
|
Controller,
|
|
Get,
|
|
Post,
|
|
Query,
|
|
UseInterceptors,
|
|
} from '@nestjs/common';
|
|
import { UnprocessableEntityException } from '@nestjs/common/exceptions/unprocessable-entity.exception';
|
|
import { ApiOperation, ApiSecurity, ApiTags } from '@nestjs/swagger';
|
|
import { ApiFileAcceptHeader } from '@waha/nestjs/ApiFileAcceptHeader';
|
|
import {
|
|
QRCodeSessionParam,
|
|
SessionApiParam,
|
|
SessionParam,
|
|
} from '@waha/nestjs/params/SessionApiParam';
|
|
|
|
import { SessionManager } from '../core/abc/manager.abc';
|
|
import { WhatsappSession } from '../core/abc/session.abc';
|
|
import { BufferResponseInterceptor } from '../nestjs/BufferResponseInterceptor';
|
|
import {
|
|
CaptchaBody,
|
|
OTPRequest,
|
|
QRCodeFormat,
|
|
QRCodeQuery,
|
|
QRCodeValue,
|
|
RequestCodeRequest,
|
|
} from '../structures/auth.dto';
|
|
import { Base64File } from '../structures/files.dto';
|
|
|
|
@ApiSecurity('api_key')
|
|
@Controller('api/:session/auth')
|
|
@ApiTags('🔑 Auth')
|
|
class AuthController {
|
|
constructor(private manager: SessionManager) {}
|
|
|
|
@Get('qr')
|
|
@ApiOperation({
|
|
summary: 'Get QR code for pairing WhatsApp API.',
|
|
})
|
|
@SessionApiParam
|
|
@ApiFileAcceptHeader(Base64File, QRCodeValue)
|
|
@UseInterceptors(new BufferResponseInterceptor())
|
|
async getQR(
|
|
@QRCodeSessionParam session: WhatsappSession,
|
|
@Query() query: QRCodeQuery,
|
|
): Promise<Buffer | QRCodeValue> {
|
|
const qr = session.getQR();
|
|
if (query.format == QRCodeFormat.RAW) {
|
|
return { value: qr.raw };
|
|
}
|
|
return qr.get();
|
|
}
|
|
|
|
@Post('request-code')
|
|
@SessionApiParam
|
|
@ApiOperation({
|
|
summary: 'Request authentication code.',
|
|
})
|
|
requestCode(
|
|
@SessionParam session: WhatsappSession,
|
|
@Body() request: RequestCodeRequest,
|
|
) {
|
|
return session.requestCode(request.phoneNumber, request.method, request);
|
|
}
|
|
|
|
@Post('authorize-code')
|
|
@SessionApiParam
|
|
@ApiOperation({
|
|
summary: 'Send OTP authentication code.',
|
|
})
|
|
authorizeCode(
|
|
@SessionParam session: WhatsappSession,
|
|
@Body() request: OTPRequest,
|
|
) {
|
|
return session.authorizeCode(request.code);
|
|
}
|
|
|
|
@Get('captcha')
|
|
@SessionApiParam
|
|
@ApiFileAcceptHeader()
|
|
@UseInterceptors(new BufferResponseInterceptor())
|
|
@ApiOperation({
|
|
summary: 'Get captcha image.',
|
|
})
|
|
async getCaptcha(@SessionParam session: WhatsappSession): Promise<Buffer> {
|
|
const image = await session.getCaptcha();
|
|
return image.get();
|
|
}
|
|
|
|
@Post('captcha')
|
|
@SessionApiParam
|
|
@ApiOperation({
|
|
summary: 'Enter captcha code.',
|
|
})
|
|
saveCaptcha(
|
|
@SessionParam session: WhatsappSession,
|
|
@Body() body: CaptchaBody,
|
|
) {
|
|
return session.saveCaptcha(body.code);
|
|
}
|
|
}
|
|
|
|
export { AuthController };
|