121 lines
4.6 KiB
TypeScript
121 lines
4.6 KiB
TypeScript
import { Injectable, NotFoundException, Logger, ForbiddenException } from '@nestjs/common';
|
||
import { InjectModel } from '@nestjs/sequelize';
|
||
import { Checkin, CheckinStatus } from './models/checkin.model';
|
||
import { CreateCheckinDto, UpdateCheckinDto, CompleteCheckinDto, RemoveCheckinDto, CheckinResponseDto } from './dto/checkin.dto';
|
||
import { ResponseCode } from '../base.dto';
|
||
import * as dayjs from 'dayjs';
|
||
import { Op } from 'sequelize';
|
||
|
||
@Injectable()
|
||
export class CheckinsService {
|
||
private readonly logger = new Logger(CheckinsService.name);
|
||
|
||
constructor(
|
||
@InjectModel(Checkin)
|
||
private readonly checkinModel: typeof Checkin,
|
||
) { }
|
||
|
||
async create(dto: CreateCheckinDto): Promise<CheckinResponseDto> {
|
||
const record = await this.checkinModel.create({
|
||
userId: dto.userId,
|
||
workoutId: dto.workoutId || null,
|
||
planId: dto.planId || null,
|
||
title: dto.title || null,
|
||
checkinDate: dto.checkinDate || null,
|
||
startedAt: dto.startedAt ? new Date(dto.startedAt) : null,
|
||
notes: dto.notes || null,
|
||
metrics: dto.metrics || null,
|
||
status: CheckinStatus.PENDING,
|
||
});
|
||
|
||
return { code: ResponseCode.SUCCESS, message: 'success', data: record.toJSON() };
|
||
}
|
||
|
||
async update(dto: UpdateCheckinDto, userId: string): Promise<CheckinResponseDto> {
|
||
const record = await this.checkinModel.findByPk(dto.id);
|
||
if (!record) {
|
||
throw new NotFoundException('打卡记录不存在');
|
||
}
|
||
if (record.userId !== userId) {
|
||
throw new ForbiddenException('无权操作该打卡记录');
|
||
}
|
||
|
||
if (dto.workoutId !== undefined) record.workoutId = dto.workoutId;
|
||
if (dto.planId !== undefined) record.planId = dto.planId;
|
||
if (dto.title !== undefined) record.title = dto.title;
|
||
if (dto.checkinDate !== undefined) record.checkinDate = dto.checkinDate as any;
|
||
if (dto.startedAt !== undefined) record.startedAt = dto.startedAt ? new Date(dto.startedAt) : null;
|
||
if (dto.notes !== undefined) record.notes = dto.notes;
|
||
if (dto.metrics !== undefined) record.metrics = dto.metrics as any;
|
||
if (dto.status !== undefined) record.status = dto.status;
|
||
if (dto.completedAt !== undefined) record.completedAt = dto.completedAt ? new Date(dto.completedAt) : null;
|
||
if (dto.durationSeconds !== undefined) record.durationSeconds = dto.durationSeconds;
|
||
|
||
await record.save();
|
||
return { code: ResponseCode.SUCCESS, message: 'success', data: record.toJSON() };
|
||
}
|
||
|
||
async complete(dto: CompleteCheckinDto, userId: string): Promise<CheckinResponseDto> {
|
||
const record = await this.checkinModel.findByPk(dto.id);
|
||
if (!record) {
|
||
throw new NotFoundException('打卡记录不存在');
|
||
}
|
||
if (record.userId !== userId) {
|
||
throw new ForbiddenException('无权操作该打卡记录');
|
||
}
|
||
|
||
record.status = CheckinStatus.COMPLETED;
|
||
record.completedAt = dto.completedAt ? new Date(dto.completedAt) : new Date();
|
||
if (dto.durationSeconds !== undefined) record.durationSeconds = dto.durationSeconds;
|
||
if (dto.notes !== undefined) record.notes = dto.notes;
|
||
if (dto.metrics !== undefined) record.metrics = dto.metrics as any;
|
||
|
||
await record.save();
|
||
return { code: ResponseCode.SUCCESS, message: 'success', data: record.toJSON() };
|
||
}
|
||
|
||
async remove(dto: RemoveCheckinDto, userId: string): Promise<CheckinResponseDto> {
|
||
const record = await this.checkinModel.findByPk(dto.id);
|
||
if (!record) {
|
||
throw new NotFoundException('打卡记录不存在');
|
||
}
|
||
if (record.userId !== userId) {
|
||
throw new ForbiddenException('无权操作该打卡记录');
|
||
}
|
||
await record.destroy();
|
||
return { code: ResponseCode.SUCCESS, message: 'success', data: { id: dto.id } };
|
||
}
|
||
|
||
async getDaily(userId: string, date?: string): Promise<CheckinResponseDto> {
|
||
const target = date ? dayjs(date) : dayjs();
|
||
if (!target.isValid()) {
|
||
return { code: ResponseCode.ERROR, message: '无效日期', data: [] };
|
||
}
|
||
const start = target.startOf('day').toDate();
|
||
const end = target.endOf('day').toDate();
|
||
|
||
// 覆盖两类数据：
|
||
// 1) checkinDate == YYYY-MM-DD（精确日）
|
||
// 2) startedAt/ completedAt 落在该日范围内（更鲁棒）
|
||
const rows = await this.checkinModel.findAll({
|
||
where: {
|
||
userId,
|
||
[Op.or]: [
|
||
{ checkinDate: target.format('YYYY-MM-DD') as any },
|
||
{
|
||
[Op.or]: [
|
||
{ startedAt: { [Op.between]: [start, end] } },
|
||
{ completedAt: { [Op.between]: [start, end] } },
|
||
],
|
||
},
|
||
],
|
||
},
|
||
order: [['createdAt', 'ASC']],
|
||
});
|
||
|
||
return { code: ResponseCode.SUCCESS, message: 'success', data: rows.map(r => r.toJSON()) };
|
||
}
|
||
}
|
||
|
||
|