Examples with Users com.infiniteautomation.mango.db.tables.Users used on opensource projects

Example 1 with Users

use of com.infiniteautomation.mango.db.tables.Users in project ma-core-public by infiniteautomation.

the class DatabaseProxy method initializeCoreDatabase.

/**
 * Inserts and updates data for a new installation
 */
default void initializeCoreDatabase(DSLContext context) {
    SystemSettings ss = SystemSettings.SYSTEM_SETTINGS;
    Roles r = Roles.ROLES;
    Users u = Users.USERS;
    UserRoleMappings urm = UserRoleMappings.USER_ROLE_MAPPINGS;
    RoleInheritance ri = RoleInheritance.ROLE_INHERITANCE;
    Permissions permissions = Permissions.PERMISSIONS;
    context.insertInto(ss, ss.settingName, ss.settingValue).values(SystemSettingsDao.NEW_INSTANCE, BaseDao.boolToChar(true)).values(SystemSettingsDao.DATABASE_SCHEMA_VERSION, Integer.toString(Common.getDatabaseSchemaVersion())).execute();
    Translations translations = Common.getTranslations();
    context.insertInto(r, r.id, r.xid, r.name).values(PermissionHolder.SUPERADMIN_ROLE.getId(), PermissionHolder.SUPERADMIN_ROLE.getXid(), translations.translate("roles.superadmin")).values(PermissionHolder.USER_ROLE.getId(), PermissionHolder.USER_ROLE.getXid(), translations.translate("roles.user")).values(PermissionHolder.ANONYMOUS_ROLE.getId(), PermissionHolder.ANONYMOUS_ROLE.getXid(), translations.translate("roles.anonymous")).execute();
    // Fix next sequence value for postgres
    if (getType().getDialect() == SQLDialect.POSTGRES) {
        String sequence = r.getName() + "_" + r.id.getName() + "_seq";
        context.alterSequence(sequence).restartWith(4).execute();
    }
    context.insertInto(ri, ri.roleId, ri.inheritedRoleId).values(PermissionHolder.SUPERADMIN_ROLE.getId(), PermissionHolder.USER_ROLE.getId()).values(PermissionHolder.USER_ROLE.getId(), PermissionHolder.ANONYMOUS_ROLE.getId()).execute();
    // create superadmin only permission, with no minterm mappings
    int adminOnlyPermissionId = context.insertInto(permissions).defaultValues().returningResult(permissions.id).fetchOptional().orElseThrow(IllegalStateException::new).get(permissions.id);
    if (Common.envProps.getBoolean("initialize.admin.create")) {
        long createdTs = System.currentTimeMillis();
        String defaultPassword = Common.envProps.getProperty("initialize.admin.password");
        long passwordChangeTs = defaultPassword.equals("admin") ? createdTs : createdTs + 1;
        int adminId = context.insertInto(u).set(u.name, translations.translate("users.defaultAdministratorName")).set(u.username, Common.envProps.getProperty("initialize.admin.username")).set(u.password, Common.encrypt(defaultPassword)).set(u.email, Common.envProps.getProperty("initialize.admin.email")).set(u.phone, "").set(u.disabled, BaseDao.boolToChar(false)).set(u.lastLogin, 0L).set(u.homeUrl, "/ui/administration/home").set(u.receiveAlarmEmails, AlarmLevels.IGNORE.value()).set(u.receiveOwnAuditEvents, BaseDao.boolToChar(false)).set(u.muted, BaseDao.boolToChar(true)).set(u.tokenVersion, 1).set(u.passwordVersion, 1).set(u.passwordChangeTimestamp, passwordChangeTs).set(u.sessionExpirationOverride, BaseDao.boolToChar(false)).set(u.createdTs, createdTs).set(u.readPermissionId, adminOnlyPermissionId).set(u.editPermissionId, adminOnlyPermissionId).returningResult(u.id).fetchOptional().orElseThrow(IllegalStateException::new).get(u.id);
        context.insertInto(urm, urm.userId, urm.roleId).values(adminId, PermissionHolder.SUPERADMIN_ROLE.getId()).values(adminId, PermissionHolder.USER_ROLE.getId()).execute();
    }
}

Example 2 with Users

use of com.infiniteautomation.mango.db.tables.Users in project ma-core-public by infiniteautomation.

the class Upgrade38 method upgrade.

@Override
protected void upgrade() throws Exception {
    Users users = Users.USERS;
    Permissions permissions = Permissions.PERMISSIONS;
    create.batch(// allow null values for now
    DSL.alterTable(users).addColumn(users.readPermissionId.getName(), users.readPermissionId.getDataType().nullable(true)), DSL.alterTable(users).addColumn(users.editPermissionId.getName(), users.editPermissionId.getDataType().nullable(true))).execute();
    doInTransaction(txStatus -> {
        MangoPermission adminOnlyPermission = getOrCreatePermissionNoCache(MangoPermission.superadminOnly());
        create.batch(DSL.update(users).set(users.readPermissionId, adminOnlyPermission.getId()), DSL.update(users).set(users.editPermissionId, adminOnlyPermission.getId())).execute();
    });
    create.batch(// change columns to non-null
    DSL.alterTable(users).alterColumn(users.readPermissionId).set(users.readPermissionId.getDataType()), DSL.alterTable(users).alterColumn(users.editPermissionId).set(users.editPermissionId.getDataType()), DSL.alterTable(users).add(DSL.constraint("usersFk1").foreignKey(users.readPermissionId).references(permissions, permissions.id).onDeleteRestrict()), DSL.alterTable(users).add(DSL.constraint("usersFk2").foreignKey(users.editPermissionId).references(permissions, permissions.id).onDeleteRestrict())).execute();
}