injector function 分享

[maslow]

Injector Function 是利用云函数扩展 Database Proxy Policy 的一种手段，编写一个 injector function 可以向 Policy 注入/暴露 自定义逻辑（变量、函数等）。

分享一个我在 app 的后端管理端常用的 injector function:

import cloud from '@/cloud-sdk'
import * as crypto from 'crypto'

/**
 * 本函数为 policy injector， 当用户请求 proxy/:policy 时，会调用本函数返回的函数获取该策略的 injections。
 * 返回的 injections 会注入到该策略规则中执行。
 * 例如，本例中返回了 $has 和 $is 函数，则在规则中可以这样使用：
 * ```json
 * {
 *   "add": "$has('article.create')",
 *   "remove": "$is('admin')"
 * }
 * ```
 */

exports.main = async function (ctx) {

  return async function (payload: any, params: any) {
    const auth = payload || {}
    const { permissions, roles } = await getPermissions(auth.uid)
    return {
      ...auth,
      $has: (permissionName: string) => {
        return permissions.includes(permissionName)
      },
      $is: (roleName: string) => {
        return roles.includes(roleName)
      }
    }
  }
}


/**
 * 通过 user id 获取权限列表
 * @param role_ids 
 * @returns 
 */
async function getPermissions(uid: string) {
  const db = cloud.database()
  // 查用户
  const { data: admin } = await db.collection('admins')
    .where({ _id: uid })
    .getOne()


  // 查角色
  const { data: roles } = await db.collection('roles')
    .where({
      name: {
        $in: admin.roles ?? []
      }
    })
    .get()

  if (!roles) {
    return { permissions: [], roles: [], user: admin }
  }

  const permissions = []
  for (const role of roles) {
    const perms = role.permissions ?? []
    permissions.push(...perms)
  }

  return {
    permissions,
    roles: roles.map(role => role.name),
    user: admin
  }
}


/**
 * 判断用户是否有权限
 * @param uid 用户ID
 * @param permission 权限名
 * @returns 0 表示用户有权限， 401 表示用户未登录， 403 表示用户未授权
 */
async function checkPermission(uid: string, permission: string): Promise<number> {
  if (!uid) {
    return 401
  }
  const { permissions } = await getPermissions(uid)

  if (!permissions.includes(permission)) {
    return 403
  }
  return 0
}

/**
 * @param {string} content
 * @return {string}
 */
function hashPassword(content: string): string {
  return crypto
    .createHash('sha256')
    .update(content)
    .digest('hex')
}

[yenche123]

原来 injector 是这样用的，学到了！
在新建访问策略的时候，看到 Injector 选项就很疑惑，现在懂了！

如上图，在这个地方可以选对应的云函数作为 injector

[youmengme]

emmm 我想请教一下新建完策略之后怎么使用策略呢？

[yenche123]

有个"发布策略"的按钮，点下去后，使用 laf-client-sdk 请求时，指定 dbProxyUrl 就会直接使用啦。