casbin
Version:
An authorization library that supports access control models like ACL, RBAC, ABAC in Node.JS
156 lines (155 loc) • 5.54 kB
JavaScript
// Copyright 2018 The Casbin Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
import { CoreEnforcer } from './coreEnforcer';
import { PolicyOp } from './model';
/**
* InternalEnforcer = CoreEnforcer + Internal API.
*/
export class InternalEnforcer extends CoreEnforcer {
/**
* addPolicyInternal adds a rule to the current policy.
*/
async addPolicyInternal(sec, ptype, rule) {
if (this.model.hasPolicy(sec, ptype, rule)) {
return false;
}
if (this.adapter && this.autoSave) {
try {
await this.adapter.addPolicy(sec, ptype, rule);
}
catch (e) {
if (e.message !== 'not implemented') {
throw e;
}
}
}
if (this.watcher && this.autoNotifyWatcher) {
// error intentionally ignored
this.watcher.update();
}
const ok = this.model.addPolicy(sec, ptype, rule);
if (sec === 'g' && ok) {
await this.buildIncrementalRoleLinks(PolicyOp.PolicyAdd, ptype, [rule]);
}
return ok;
}
// addPolicies adds rules to the current policy.
// removePolicies removes rules from the current policy.
async addPoliciesInternal(sec, ptype, rules) {
for (const rule of rules) {
if (this.model.hasPolicy(sec, ptype, rule)) {
return false;
}
}
const batchAdapter = this.adapter;
if (batchAdapter && this.autoSave) {
try {
await batchAdapter.addPolicies(sec, ptype, rules);
}
catch (e) {
if (e.message !== 'not implemented') {
throw e;
}
}
}
if (this.watcher && this.autoNotifyWatcher) {
// error intentionally ignored
this.watcher.update();
}
const [ok, effects] = await this.model.addPolicies(sec, ptype, rules);
if (sec === 'g' && ok && effects?.length) {
await this.buildIncrementalRoleLinks(PolicyOp.PolicyAdd, ptype, effects);
}
return ok;
}
/**
* removePolicyInternal removes a rule from the current policy.
*/
async removePolicyInternal(sec, ptype, rule) {
if (!this.model.hasPolicy(sec, ptype, rule)) {
return false;
}
if (this.adapter && this.autoSave) {
try {
await this.adapter.removePolicy(sec, ptype, rule);
}
catch (e) {
if (e.message !== 'not implemented') {
throw e;
}
}
}
if (this.watcher && this.autoNotifyWatcher) {
// error intentionally ignored
this.watcher.update();
}
const ok = await this.model.removePolicy(sec, ptype, rule);
if (sec === 'g' && ok) {
await this.buildIncrementalRoleLinks(PolicyOp.PolicyRemove, ptype, [rule]);
}
return ok;
}
// removePolicies removes rules from the current policy.
async removePoliciesInternal(sec, ptype, rules) {
for (const rule of rules) {
if (!this.model.hasPolicy(sec, ptype, rule)) {
return false;
}
}
const batchAdapter = this.adapter;
if (batchAdapter && this.autoSave) {
try {
await batchAdapter.removePolicies(sec, ptype, rules);
}
catch (e) {
if (e.message !== 'not implemented') {
throw e;
}
}
}
if (this.watcher && this.autoNotifyWatcher) {
// error intentionally ignored
this.watcher.update();
}
const [ok, effects] = this.model.removePolicies(sec, ptype, rules);
if (sec === 'g' && ok && effects?.length) {
await this.buildIncrementalRoleLinks(PolicyOp.PolicyRemove, ptype, effects);
}
return ok;
}
/**
* removeFilteredPolicyInternal removes rules based on field filters from the current policy.
*/
async removeFilteredPolicyInternal(sec, ptype, fieldIndex, fieldValues) {
if (this.adapter && this.autoSave) {
try {
await this.adapter.removeFilteredPolicy(sec, ptype, fieldIndex, ...fieldValues);
}
catch (e) {
if (e.message !== 'not implemented') {
throw e;
}
}
}
if (this.watcher && this.autoNotifyWatcher) {
// error intentionally ignored
this.watcher.update();
}
const [ok, effects] = this.model.removeFilteredPolicy(sec, ptype, fieldIndex, ...fieldValues);
if (sec === 'g' && ok && effects?.length) {
await this.buildIncrementalRoleLinks(PolicyOp.PolicyRemove, ptype, effects);
}
return ok;
}
}