db.js

const { sql } = require('./config/server');

class SqlError extends Error {
    constructor(message) {
        super(message);
        this.name = 'SqlError';
    }
}

class SqlConnector {
    constructor(pool) {
        this.connPool = pool;
        this.user = null;
    }

    /**
     * 사용자 이메일을 등록하는 함수
     * OAuth 연결 이후 바로 실행 요망
     * @param {String} email 사용자 이메일
     */
    setUser(email) {
        this.user = email;
    }

    /**
     * 친구를 검색해서 배열로 가져오는 함수
     * @param {String} partialEmail 부분 이메일
     * @returns {String[]}
     */
    async searchFriends(partialEmail) {
        const result = await this.connPool
            .request()
            .input('user', sql.VarChar, `%${partialEmail}%`)
            .query(
                'SELECT DISTINCT [user] FROM Goals WHERE [user] LIKE @user;'
            );

        return result.recordset.map((x) => x.user);
    }

    /**
     * 사용자의 팔로잉 목록을 가져오는 함수
     * @returns {String[]} 팔로잉 목록
     */
    async getFollowings() {
        if (this.user == null)
            throw new SqlError('SqlConnector.user is empty.');

        const result = await this.connPool
            .request()
            .input('user', sql.VarChar, this.user)
            .query('SELECT following FROM Following WHERE [user] = @user;');

        return result.recordset.map((x) => x.following);
    }

    /**
     * 팔로우하는 친구를 추가하는 함수
     * @param {String} following 팔로잉 하는 친구 이메일
     */
        async addFollowing(following) {
            if (this.user == null)
                throw new SqlError('SqlConnector.user is empty.');
    
            await this.connPool
                .request()
                .input('user', sql.VarChar, this.user)
                .input('following', sql.VarChar, following)
                .query(
                    'INSERT INTO Following VALUES(@user, @following);'
                );
        }

    /**
     * 특정 사용자의 목표 목록을 받아오는 함수
     * @param {String} user 사용자 이메일
     * @param {Number} month 월
     * @returns {Object[]}
     */
    async getGoals(user, month) {
        const result = await this.connPool
            .request()
            .input('user', sql.VarChar, user)
            .input('month', sql.Int, month)
            .query(
                'SELECT id, goal FROM Goals WHERE [user] = @user AND month = @month;'
            );

        return result.recordset;
    }

    /**
     * 해당 달에 모든 목표를 달성했는지 확인하는 함수
     * @param {String} user 사용자 이메일
     * @param {Number} month 월
     * @returns {Boolean}
     */
    async achievedAll(user, month) {
        const result = await this.connPool
            .request()
            .input('user', sql.VarChar, user)
            .input('month', sql.Int, month)
            .query(
                'SELECT achieved FROM Goals WHERE [user] = @user AND month = @month;'
            );

        // 리스트로 변환한 후, 전체가 true인지 확인
        return result.recordset
            .map((x) => x.following)
            .every((x) => x === true);
    }

    /**
     * 목표 하나를 수정하는 함수
     * @param {Number} month 월
     * @param {Number} id 목표 ID
     * @param {String} newGoal 새 목표
     */
    async modifyGoal(month, id, newGoal) {
        if (this.user == null)
            throw new SqlError('SqlConnector.user is empty.');

        await this.connPool
            .request()
            .input('user', sql.VarChar, this.user)
            .input('month', sql.Int, month)
            .input('id', sql.Int, id)
            .input('newGoal', sql.NVarChar, newGoal)
            .query(
                'UPDATE Goals SET goal = @newGoal WHERE [user] = @user AND month = @month AND id = @id;'
            );
    }

    /**
     * 목표 하나를 삭제하는 함수
     * @param {Number} month 월
     * @param {Number} id 목표 ID
     */
    async deleteGoal(month, id) {
        if (this.user == null)
            throw new SqlError('SqlConnector.user is empty.');

        await this.connPool
            .request()
            .input('user', sql.VarChar, this.user)
            .input('month', sql.Int, month)
            .input('id', sql.Int, id)
            .query(
                'DELETE Goals WHERE [user] = @user AND month = @month AND id = @id;'
            );
    }

    /**
     * 목록에 목표 추가하는 함수
     * @param {Number} month 월
     * @param {Number} id 아이디
     * @param {String} goal 목표
     */
    async addGoal(month, id, goal) {
        if (this.user == null)
            throw new SqlError('SqlConnector.user is empty.');

        await this.connPool
            .request()
            .input('user', sql.VarChar, this.user)
            .input('month', sql.Int, month)
            .input('id', sql.Int, id)
            .input('goal', sql.NVarChar, goal)
            .query(
                'INSERT INTO Goals VALUES(@user, @month, @id, @goal, 0);'
            );
    }
}

module.exports = {
    SqlConnector,
};