const Apartment = require('./Apartment');
const Room = require('./Room');
const Rental = require('./Rental');
const WaterBill = require('./WaterBill');
const ElectricityBill = require('./ElectricityBill');
const User = require('./User');
const Role = require('./Role');
const Menu = require('./Menu');
const RoleMenu = require('./RoleMenu');
const OperationLog = require('./OperationLog');
const LoginLog = require('./LoginLog');
const Expense = require('./Expense');

// 关联关系
User.belongsTo(Role, { foreignKey: 'roleId', as: 'role' });
Role.hasMany(User, { foreignKey: 'roleId', as: 'users' });

// 菜单自关联（父子菜单）
Menu.belongsTo(Menu, { foreignKey: 'parentId', as: 'parent' });
Menu.hasMany(Menu, { foreignKey: 'parentId', as: 'children' });

// 角色与菜单多对多关联
Role.belongsToMany(Menu, { through: RoleMenu, foreignKey: 'roleId', otherKey: 'menuId', as: 'menus' });
Menu.belongsToMany(Role, { through: RoleMenu, foreignKey: 'menuId', otherKey: 'roleId', as: 'roles' });

module.exports = {
  Apartment,
  Room,
  Rental,
  WaterBill,
  ElectricityBill,
  User,
  Role,
  Menu,
  RoleMenu,
  OperationLog,
  LoginLog,
  Expense
};