|
|
@ -1,11 +1,16 @@ |
|
|
|
import { Catch, Injectable, NotFoundException } from '@nestjs/common' |
|
|
|
import { InjectRepository } from '@nestjs/typeorm' |
|
|
|
import { EntityNotFoundError, QueryFailedError, Repository } from 'typeorm' |
|
|
|
import { User } from './entity/user.entity' |
|
|
|
import { type UserDto } from './dto/user.dto' |
|
|
|
import { type Channel } from 'src/chat/entity/channel.entity' |
|
|
|
import type Result from 'src/pong/entity/result.entity' |
|
|
|
import { Cron } from '@nestjs/schedule' |
|
|
|
import { |
|
|
|
BadRequestException, |
|
|
|
Catch, |
|
|
|
Injectable, |
|
|
|
NotFoundException, |
|
|
|
} from "@nestjs/common"; |
|
|
|
import { InjectRepository } from "@nestjs/typeorm"; |
|
|
|
import { EntityNotFoundError, QueryFailedError, Repository } from "typeorm"; |
|
|
|
import { User } from "./entity/user.entity"; |
|
|
|
import { type UserDto } from "./dto/user.dto"; |
|
|
|
import { type Channel } from "src/chat/entity/channel.entity"; |
|
|
|
import type Result from "src/pong/entity/result.entity"; |
|
|
|
import { Cron } from "@nestjs/schedule"; |
|
|
|
|
|
|
|
@Injectable() |
|
|
|
@Catch(QueryFailedError, EntityNotFoundError) |
|
|
@ -15,96 +20,96 @@ export class UsersService { |
|
|
|
) {} |
|
|
|
|
|
|
|
save(user: User) { |
|
|
|
this.usersRepository.save(user) |
|
|
|
this.usersRepository.save(user); |
|
|
|
} |
|
|
|
|
|
|
|
async findUsers(): Promise<User[]> { |
|
|
|
return await this.usersRepository.find({}) |
|
|
|
return await this.usersRepository.find({}); |
|
|
|
} |
|
|
|
|
|
|
|
async findUserByName(username: string): Promise<User | null> { |
|
|
|
const user = await this.usersRepository.findOne({ |
|
|
|
where: { username }, |
|
|
|
relations: { results: true } |
|
|
|
}) |
|
|
|
return user |
|
|
|
relations: { results: true }, |
|
|
|
}); |
|
|
|
return user; |
|
|
|
} |
|
|
|
|
|
|
|
@Cron('0 * * * * *') |
|
|
|
@Cron("0 * * * * *") |
|
|
|
async updateStatus() { |
|
|
|
const users = await this.usersRepository.find({}) |
|
|
|
const users = await this.usersRepository.find({}); |
|
|
|
users.forEach((usr) => { |
|
|
|
if (Date.now() - usr.lastAccess > 60000) { |
|
|
|
usr.status = 'offline' |
|
|
|
this.usersRepository.save(usr) |
|
|
|
usr.status = "offline"; |
|
|
|
this.usersRepository.save(usr); |
|
|
|
} |
|
|
|
}) |
|
|
|
}); |
|
|
|
} |
|
|
|
|
|
|
|
async findUser(ftId: number): Promise<User | null> { |
|
|
|
const user = await this.usersRepository.findOneBy({ ftId }) |
|
|
|
if (user == null) return null |
|
|
|
user.lastAccess = Date.now() |
|
|
|
user.status = 'online' |
|
|
|
this.usersRepository.save(user) |
|
|
|
return user |
|
|
|
const user = await this.usersRepository.findOneBy({ ftId }); |
|
|
|
if (user == null) return null; |
|
|
|
user.lastAccess = Date.now(); |
|
|
|
user.status = "online"; |
|
|
|
this.usersRepository.save(user); |
|
|
|
return user; |
|
|
|
} |
|
|
|
|
|
|
|
async findOnlineUsers(): Promise<User[]> { |
|
|
|
return await this.usersRepository.find({ where: { status: 'online' } }) |
|
|
|
return await this.usersRepository.find({ where: { status: "online" } }); |
|
|
|
} |
|
|
|
|
|
|
|
async create(userData: UserDto) { |
|
|
|
try { |
|
|
|
const newUser = this.usersRepository.create(userData) |
|
|
|
return await this.usersRepository.save(newUser) |
|
|
|
const newUser = this.usersRepository.create(userData); |
|
|
|
return await this.usersRepository.save(newUser); |
|
|
|
} catch (err) { |
|
|
|
throw new Error(`Error creating user ${err}`) |
|
|
|
throw new Error(`Error creating user ${err}`); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
async findOnlineInChannel(channel: Channel): Promise<User[]> { |
|
|
|
return await this.usersRepository |
|
|
|
.createQueryBuilder('user') |
|
|
|
.where('user.channel = :chan', { chan: channel }) |
|
|
|
.andWhere('user.status := status)', { status: 'online' }) |
|
|
|
.getMany() |
|
|
|
.createQueryBuilder("user") |
|
|
|
.where("user.channel = :chan", { chan: channel }) |
|
|
|
.andWhere("user.status := status)", { status: "online" }) |
|
|
|
.getMany(); |
|
|
|
} |
|
|
|
|
|
|
|
async update(user: User, changes: UserDto): Promise<User | null> { |
|
|
|
this.usersRepository.merge(user, changes) |
|
|
|
return await this.usersRepository.save(user) |
|
|
|
this.usersRepository.merge(user, changes); |
|
|
|
return await this.usersRepository.save(user); |
|
|
|
} |
|
|
|
|
|
|
|
async addAvatar(ftId: number, filename: string) { |
|
|
|
return await this.usersRepository.update( |
|
|
|
{ ftId }, |
|
|
|
{ |
|
|
|
avatar: filename |
|
|
|
avatar: filename, |
|
|
|
} |
|
|
|
) |
|
|
|
); |
|
|
|
} |
|
|
|
|
|
|
|
async getFriends(ftId: number): Promise<User[]> { |
|
|
|
const user = await this.usersRepository.findOne({ |
|
|
|
where: { ftId }, |
|
|
|
relations: { |
|
|
|
friends: true |
|
|
|
} |
|
|
|
}) |
|
|
|
if (user != null) return user.friends |
|
|
|
return [] |
|
|
|
friends: true, |
|
|
|
}, |
|
|
|
}); |
|
|
|
if (user != null) return user.friends; |
|
|
|
return []; |
|
|
|
} |
|
|
|
|
|
|
|
async getInvits(ftId: number): Promise<User[]> { |
|
|
|
const user = await this.usersRepository.findOne({ |
|
|
|
where: { ftId }, |
|
|
|
relations: { |
|
|
|
followers: true |
|
|
|
} |
|
|
|
}) |
|
|
|
if (user != null) return user.followers |
|
|
|
return [] |
|
|
|
followers: true, |
|
|
|
}, |
|
|
|
}); |
|
|
|
if (user != null) return user.followers; |
|
|
|
return []; |
|
|
|
} |
|
|
|
|
|
|
|
async getResults(ftId: number): Promise<Result[]> { |
|
|
@ -112,68 +117,53 @@ export class UsersService { |
|
|
|
where: { ftId }, |
|
|
|
relations: { |
|
|
|
results: { |
|
|
|
players: true |
|
|
|
} |
|
|
|
} |
|
|
|
}) |
|
|
|
if (user != null) return user.results |
|
|
|
return [] |
|
|
|
players: true, |
|
|
|
}, |
|
|
|
}, |
|
|
|
}); |
|
|
|
if (user != null) return user.results; |
|
|
|
return []; |
|
|
|
} |
|
|
|
|
|
|
|
async getLeader (): Promise<User[]> { |
|
|
|
async getLeaderboard(): Promise<User[]> { |
|
|
|
return await this.usersRepository.find({ |
|
|
|
order: { |
|
|
|
winrate: 'DESC' |
|
|
|
} |
|
|
|
}) |
|
|
|
winrate: "DESC", |
|
|
|
}, |
|
|
|
}); |
|
|
|
} |
|
|
|
|
|
|
|
async getRank(ftId: number): Promise<number> { |
|
|
|
const leader = await this.getLeader() |
|
|
|
return leader.findIndex((user) => user.ftId === ftId) |
|
|
|
const leader = await this.getLeaderboard(); |
|
|
|
return leader.findIndex((user) => user.ftId === ftId); |
|
|
|
} |
|
|
|
|
|
|
|
async invit(ftId: number, targetFtId: number): Promise<any> { |
|
|
|
const user = await this.usersRepository.findOne({ |
|
|
|
const user: User = (await this.usersRepository.findOne({ |
|
|
|
where: { ftId }, |
|
|
|
relations: { |
|
|
|
followers: true, |
|
|
|
friends: true |
|
|
|
} |
|
|
|
}) |
|
|
|
if (user == null) { |
|
|
|
return new NotFoundException(`Error: user id ${ftId} isn't in our db.`) |
|
|
|
} |
|
|
|
if (user.friends.findIndex((friend) => friend.ftId === targetFtId) !== -1) { |
|
|
|
return null |
|
|
|
} |
|
|
|
const target = await this.usersRepository.findOne({ |
|
|
|
friends: true, |
|
|
|
}, |
|
|
|
}))!; |
|
|
|
if (user.friends.findIndex((friend) => friend.ftId === targetFtId) !== -1) |
|
|
|
return new BadRequestException("You are already friends."); |
|
|
|
const target = (await this.usersRepository.findOne({ |
|
|
|
where: { ftId: targetFtId }, |
|
|
|
relations: { |
|
|
|
followers: true, |
|
|
|
friends: true |
|
|
|
} |
|
|
|
}) |
|
|
|
if (target == null) { |
|
|
|
return new NotFoundException( |
|
|
|
`Error: user id ${targetFtId} isn't in our db.` |
|
|
|
) |
|
|
|
} |
|
|
|
friends: true, |
|
|
|
}, |
|
|
|
}))!; |
|
|
|
const id = user.followers.findIndex( |
|
|
|
(follower) => follower.ftId === targetFtId |
|
|
|
) |
|
|
|
); |
|
|
|
if (id !== -1) { |
|
|
|
console.log( |
|
|
|
`Friend relation complete between ${user.username} and ${target.username}` |
|
|
|
) |
|
|
|
user.friends.push(target) |
|
|
|
if (user.ftId !== target.ftId) target.friends.push(user) |
|
|
|
user.followers.slice(id, 1) |
|
|
|
await this.usersRepository.save(user) |
|
|
|
} else { |
|
|
|
console.log(`You asked ${target.username} to be your friend.`) |
|
|
|
target.followers.push(user) |
|
|
|
} |
|
|
|
await this.usersRepository.save(target) |
|
|
|
user.friends.push(target); |
|
|
|
if (user.ftId !== target.ftId) target.friends.push(user); |
|
|
|
user.followers.slice(id, 1); |
|
|
|
await this.usersRepository.save(user); |
|
|
|
} else target.followers.push(user); |
|
|
|
await this.usersRepository.save(target); |
|
|
|
} |
|
|
|
} |
|
|
|