• Index
• Search by APIs
• Search by Words
• Project Search

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• CSZ-Bot-master
  • .github
    • ISSUE_TEMPLATE
      • Fehler_Report.md
      • Funktions_Wunsch.md
    • workflows
      • CI.yml
      • CD.yml
      • codestyle.yml
  • src
    • app.ts
    • utils
      • typeUtils.ts
      • stringUtils.ts
      • userUtils.ts
      • configHandler.ts
      • logger.ts
    • storage
      • storage.ts
      • model
        • AdditionalMessageData.ts
        • Penis.ts
        • Reminder.ts
        • Birthday.ts
        • FadingMessage.ts
        • Ban.ts
        • Boob.ts
        • Nickname.ts
        • Stempel.ts
        • GuildRagequit.ts
    • commands
      • info.ts
      • saufen.ts
      • extend.ts
      • ficktabelle.ts
      • boobs.ts
      • roll.ts
      • command.ts
      • stempelgraph.ts
      • vote.js
      • bonk.ts
      • nickname.ts
      • erinnerung.ts
      • poll.js
      • erleuchtung.ts
      • min.ts
      • metafrage.ts
      • sdm.ts
      • never.ts
      • invite.ts
      • hilfe.ts
      • geburtstag.ts
      • special
        • where.ts
        • tiktok.ts
        • dadJoke.ts
        • nixos.ts
        • wat.ts
        • emoteSender.ts
        • nischdaaa.ts
      • stempelkarte.ts
      • mock.ts
      • google.ts
      • stempeln.ts
      • woisping.ts
      • selfban.js
      • penis.ts
      • modcommands
        • ban.ts
        • unban.ts
        • listroles.ts
        • hilfe.ts
        • assigner.ts
    • context.ts
    • types.d.ts
    • handler
      • fadingMessageHandler.ts
      • cmdHandler.ts
      • quoteHandler.ts
      • aprilFoolsHandler.ts
      • aocHandler.ts
      • nicknameHandler.ts
      • commandHandler.ts
      • bdayHandler.ts
      • messageDeleteHandler.ts
      • voiceHandler.ts
      • messageHandler.ts
      • reactionHandler.ts
  • shell.nix
  • .nvmrc
  • .eslintrc.json
  • LICENSE
  • .devcontainer
    • devcontainer.json
  • CONTRIBUTING.md
  • .editorconfig
  • .gitattributes
  • config.template.json
  • flake.nix
  • README.md
  • package.json
  • Dockerfile
  • tsconfig.json
  • .gitignore
  • .dockerignore
  • .vscode
    • settings.json
    • extensions.json
  • docker-compose.yml
  • assets

/* Disabled due to sequelize's DataTypes */
/* eslint-disable new-cap */

import { User } from "discord.js";
import moment from "moment";
import { Model, DataTypes, Sequelize, Optional, Op } from "sequelize";
import { v4 as uuidv4 } from "uuid";
import { Radius } from "../../commands/penis";
import log from "../../utils/logger";

export interface PenisAttributes {
    id: string;
    userId: string;
    measuredAt: Date;
    size: number;
    diameter: Radius;
}

export interface PenisCreationAttributes extends Optional<PenisAttributes, "id"> { }

export default class Penis extends Model<PenisAttributes, PenisCreationAttributes> implements PenisAttributes {
    declare id: string;
    declare userId: string;
    declare measuredAt: Date;
    declare size: number;
    declare diameter: Radius;

    declare readonly createdAt: Date;
    declare readonly updatedAt: Date;

    static getAveragePenisSizes = async(): Promise<Record<string, number>> => {
        // Everything hacky, but I just want to implement it.
        const averageObj: Record<string, number> = {};
        // @ts-ignore
        const result: ({ id: string, avgSize: number })[] = (await Penis.findAll({
            attributes: [
                "id",
                [Sequelize.fn("AVG", Sequelize.col("size")), "avgSize"]
            ],
            group: [ "id" ]
        }));

        for(const res of result) {
            averageObj[res.id] = res.avgSize;
        }

        return averageObj;
    };

    static longestRecentMeasurement = (): Promise<Penis | null> => {
        const startToday = moment().startOf("days");
        const startTomorrow = moment().add(1, "days").startOf("days");

        return Penis.max("size", {
            where: {
                measuredAt: {
                    [Op.and]: {
                        [Op.gte]: startToday.toDate(),
                        [Op.lt]: startTomorrow.toDate()
                    }
                }
            }
        });
    };

    static fetchRecentMeasurement = (user: User): Promise<Penis | null> => {
        const startToday = moment().startOf("days");
        const startTomorrow = moment().add(1, "days").startOf("days");

        return Penis.findOne({
            where: {
                userId: user.id,
                measuredAt: {
                    [Op.and]: {
                        [Op.gte]: startToday.toDate(),
                        [Op.lt]: startTomorrow.toDate()
                    }
                }
            }
        });
    };

    static insertMeasurement = (user: User, size: number, diameter: Radius, measuredAt: Date = new Date()): Promise<Penis> => {
        log.debug(`Saving Penis Measurement for user ${user.id} with size ${size} from ${measuredAt}`);
        return Penis.create({
            userId: user.id,
            measuredAt,
            size,
            diameter
        });
    };

    static initialize(sequelize: Sequelize) {
        this.init({
            id: {
                type: DataTypes.STRING(36),
                defaultValue: () => uuidv4(),
                primaryKey: true
            },
            userId: {
                type: DataTypes.STRING(32),
                allowNull: false
            },
            measuredAt: {
                type: DataTypes.DATE,
                allowNull: true
            },
            size: {
                type: DataTypes.INTEGER,
                allowNull: false
            },
            diameter: {
                type: DataTypes.INTEGER,
                allowNull: false
            }
        }, {
            sequelize,
            indexes: [
                {
                    using: "BTREE",
                    fields: [
                        {
                            name: "measuredAt",
                            order: "ASC"
                        }
                    ]
                }
            ]
        });
    }
}