All files / server/src/features/user/lottery-signup lotterySignupRepository.ts

56.52% Statements 26/46
50% Branches 4/8
100% Functions 6/6
56.52% Lines 26/46

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166                      54x     200x   200x 200x             200x       200x   200x 200x                 200x                               54x       37x 37x                         37x               37x       37x 37x                 37x                                 54x     24x 24x 125x                           24x   24x 122x         24x                     54x                            
import { UserModel, UserSchemaDb } from "server/features/user/userSchema";
import { UserLotterySignups } from "server/types/resultTypes";
import { logger } from "server/utils/logger";
import { MongoDbError } from "shared/types/api/errors";
import { LotterySignup, User } from "shared/types/models/user";
import {
  Result,
  makeErrorResult,
  makeSuccessResult,
} from "shared/utils/result";
 
export const saveLotterySignups = async (
  signupData: UserLotterySignups,
): Promise<Result<User, MongoDbError>> => {
  const { lotterySignups, username } = signupData;
 
  try {
    const response = await UserModel.findOneAndUpdate(
      { username },
      {
        lotterySignups,
      },
      { returnDocument: "after" },
    ).lean();
    Iif (!response) {
      logger.error(new Error("Error saving lottery signups"));
      return makeErrorResult(MongoDbError.SIGNUP_NOT_FOUND);
    }
    logger.debug(`MongoDB: Signup data stored for user ${username}`);
 
    const result = UserSchemaDb.safeParse(response);
    Iif (!result.success) {
      logger.error(
        new Error(
          `Error validating saveLotterySignups DB value: ${JSON.stringify(result.error)}`,
        ),
      );
      return makeErrorResult(MongoDbError.UNKNOWN_ERROR);
    }
 
    return makeSuccessResult(result.data);
  } catch (error) {
    logger.error(
      new Error(`MongoDB: Error storing signup data for user ${username}`, {
        cause: error,
      }),
    );
    return makeErrorResult(MongoDbError.UNKNOWN_ERROR);
  }
};
 
interface SaveLotterySignupParams {
  lotterySignup: LotterySignup;
  username: string;
}
 
export const saveLotterySignup = async ({
  lotterySignup,
  username,
}: SaveLotterySignupParams): Promise<Result<User, MongoDbError>> => {
  try {
    const response = await UserModel.findOneAndUpdate(
      { username },
      {
        $addToSet: {
          lotterySignups: {
            programItemId: lotterySignup.programItemId,
            priority: lotterySignup.priority,
            signedToStartTime: lotterySignup.signedToStartTime,
          },
        },
      },
      { returnDocument: "after" },
    ).lean();
    Iif (!response) {
      logger.error(
        new Error(
          `Error saving lottery signup ${lotterySignup.programItemId} for user ${username}, user not found`,
        ),
      );
      return makeErrorResult(MongoDbError.SIGNUP_NOT_FOUND);
    }
    logger.debug(
      `MongoDB: Lottery signup ${lotterySignup.programItemId} saved for user ${username}`,
    );
 
    const result = UserSchemaDb.safeParse(response);
    Iif (!result.success) {
      logger.error(
        new Error(
          `Error validating saveLotterySignup DB value: ${JSON.stringify(result.error)}`,
        ),
      );
      return makeErrorResult(MongoDbError.UNKNOWN_ERROR);
    }
 
    return makeSuccessResult(result.data);
  } catch (error) {
    logger.error(
      new Error(
        `MongoDB: Error saving lottery signup ${lotterySignup.programItemId} for user ${username}`,
        { cause: error },
      ),
    );
    return makeErrorResult(MongoDbError.UNKNOWN_ERROR);
  }
};
 
export interface DeleteLotterySignupsParams {
  lotterySignupProgramItemIds: string[];
  username: string;
}
 
export const delLotterySignups = async (
  usersToUpdate: DeleteLotterySignupsParams[],
): Promise<Result<void, MongoDbError>> => {
  try {
    const bulkOps = usersToUpdate.map(
      ({ username, lotterySignupProgramItemIds }) => ({
        updateOne: {
          filter: { username },
          update: {
            $pull: {
              lotterySignups: {
                programItemId: { $in: lotterySignupProgramItemIds },
              },
            },
          },
        },
      }),
    );
 
    await UserModel.bulkWrite(bulkOps);
 
    usersToUpdate.map((e) => {
      logger.debug(
        `MongoDB: Deleted lottery signups ${e.lotterySignupProgramItemIds.join(", ")} from users: ${e.username}`,
      );
    });
 
    return makeSuccessResult();
  } catch (error) {
    logger.error(
      new Error("MongoDB: Error deleting lottery signups from multiple users", {
        cause: error,
      }),
    );
    return makeErrorResult(MongoDbError.UNKNOWN_ERROR);
  }
};
 
export const removeLotterySignups = async (): Promise<
  Result<void, MongoDbError>
> => {
  logger.info("MongoDB: remove ALL signups from db");
  try {
    await UserModel.updateMany({}, { lotterySignups: [] });
    return makeSuccessResult();
  } catch (error) {
    logger.error(
      new Error("MongoDB: Error removing lottery signups", { cause: error }),
    );
    return makeErrorResult(MongoDbError.UNKNOWN_ERROR);
  }
};