juice-shop

1
/*
2
 * Copyright (c) 2014-2024 Bjoern Kimminich & the OWASP Juice Shop contributors.
3
 * SPDX-License-Identifier: MIT
4
 */
5

6
/* jslint node: true */
7
import {
8
  Model,
9
  type InferAttributes,
10
  type InferCreationAttributes,
11
  DataTypes,
12
  type CreationOptional,
13
  type Sequelize
14
} from 'sequelize'
15

16
class Wallet extends Model<
17
InferAttributes<Wallet>,
18
InferCreationAttributes<Wallet>
19
> {
20
  declare UserId: number
21
  declare id: CreationOptional<number>
22
  declare balance: CreationOptional<number>
23
}
24

25
const WalletModelInit = (sequelize: Sequelize) => {
26
  Wallet.init(
27
    {
28
      UserId: {
29
        type: DataTypes.INTEGER
30
      },
31
      id: {
32
        type: DataTypes.INTEGER,
33
        primaryKey: true,
34
        autoIncrement: true
35
      },
36
      balance: {
37
        type: DataTypes.INTEGER,
38
        validate: {
39
          isInt: true
40
        },
41
        defaultValue: 0
42
      }
43
    },
44
    {
45
      tableName: 'Wallets',
46
      sequelize
47
    }
48
  )
49
}
50

51
export { Wallet as WalletModel, WalletModelInit }
52