@rnaga/wp-node
Version:
👉 **[View Full Documentation at rnaga.github.io/wp-node →](https://rnaga.github.io/wp-node/)**
190 lines (189 loc) • 7.63 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
return c > 3 && r && Object.defineProperty(target, key, r), r;
};
var __importStar = (this && this.__importStar) || (function () {
var ownKeys = function(o) {
ownKeys = Object.getOwnPropertyNames || function (o) {
var ar = [];
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
return ar;
};
return ownKeys(o);
};
return function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
__setModuleDefault(result, mod);
return result;
};
})();
var __metadata = (this && this.__metadata) || function (k, v) {
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.RolesUtil = void 0;
const config_1 = require("../../config");
const constants_1 = require("../../constants");
const component_1 = require("../../decorators/component");
const defaults = __importStar(require("../../defaults"));
const val = __importStar(require("../../validators"));
const components_1 = require("../components");
const logger_1 = require("../logger");
const options_1 = require("../options");
const tables_1 = require("../tables");
const meta_util_1 = require("./meta.util");
const query_util_1 = require("./query.util");
let RolesUtil = class RolesUtil {
config;
logger;
components;
constructor(config, logger, components) {
this.config = config;
this.logger = logger;
this.components = components;
}
/**
* Reformat user roles to be stored in DB
*
* @param userRoles - User roles
* @returns
*/
reformatInDB(userRoles) {
return Object.entries(userRoles).reduce((a, b) => ({
...a,
[b[0]]: {
name: b[1].name,
capabilities: b[1].capabilities.reduce((a, b) => ({ ...a, [b]: 1 }), {}),
},
}), {});
}
// Get roles of blog
async get(blogId = 1) {
const tables = this.components.get(tables_1.Tables);
tables.index = blogId;
const optionsKey = `${tables.prefix}user_roles`;
const options = this.components.get(options_1.Options);
const roleRecord = (await options
.usingBlog(blogId)
.get(optionsKey)) ?? {};
const roles = Object.entries(roleRecord)
.map(([roleName, role]) => ({
[roleName]: {
name: role.name,
// 1. The format of capabilities is either Record<string, true> or string[]
// 2. Reformat it to string[]
capabilities: Array.isArray(role.capabilities)
? role.capabilities
: Object.entries(role.capabilities).map(([c]) => c),
},
}))
.reduce((a, b) => ({ ...a, ...b }), {});
if (!roles) {
this.logger.warn("User roles are not defined in DB. Use default roles");
return defaults.roles;
}
if (this.config.isMultiSite()) {
const mergeCapabilities = (roles) => {
const capabilities = new Set();
for (const role of roles) {
if (!role || !Array.isArray(role["capabilities"])) {
continue;
}
role["capabilities"].map((capability) => capabilities.add(capability));
}
return Array.from(capabilities);
};
roles["superadmin"] = {
...this.config.config.roles["superadmin"],
capabilities: mergeCapabilities([
this.config.config.roles["superadmin"],
roles["administrator"],
]),
};
}
// Set anonymous
roles["anonymous"] = this.config.config.roles["anonymous"];
return roles;
}
// Return number of roles
async count(blogId = 1) {
const roles = await this.get(blogId);
const roleNames = Object.keys(roles).filter((roleName) => !["superadmin", "anonymous"].includes(roleName));
const queryUtil = this.components.get(query_util_1.QueryUtil);
const superAdmins = await this.getSuperAdmins({ blogId });
const counts = {};
for (const roleName of roleNames) {
const count = await queryUtil.usingBlog(blogId).users((query) => {
query.withRoles([roleName]).select(["ID"]);
query.count("users", "ID");
}, val.query.resultCount);
counts[roleName] = count?.count ?? 0;
}
// Anonymous users
const countAnonymous = await queryUtil.usingBlog(blogId).users((query) => {
query
.select(["ID"])
.hasNoRole()
.andWhereNot((query) => {
query.whereIn("user_login", superAdmins);
});
query.count("users", "ID");
}, val.query.resultCount);
counts["anonymous"] = countAnonymous?.count ?? 0;
return counts;
}
async getSuperAdmins(args) {
if (!this.config.isMultiSite()) {
return [];
}
const queryUtil = this.components.get(query_util_1.QueryUtil);
const metaUtil = this.components.get(meta_util_1.MetaUtil);
let siteId;
if (args.siteId) {
siteId = args.siteId;
}
else if (args.blogId) {
const blogId = args.blogId;
const blogs = await queryUtil.blogs((query) => {
query.where("blog_id", blogId);
});
if (!blogs) {
return [];
}
siteId = blogs[0].site_id;
}
else {
return [];
}
const superAdmins = await metaUtil.getValue("site", siteId, "site_admins");
return superAdmins ?? [];
}
};
exports.RolesUtil = RolesUtil;
exports.RolesUtil = RolesUtil = __decorate([
(0, component_1.component)({ scope: constants_1.Scope.Transient }),
__metadata("design:paramtypes", [config_1.Config,
logger_1.Logger,
components_1.Components])
], RolesUtil);