Skip to content
This repository has been archived by the owner on Aug 21, 2024. It is now read-only.

Commit

Permalink
Added new check-scope hook which will return true or false (#10279)
Browse files Browse the repository at this point in the history
* Updated verify scope hook to conditionally not throw exception

* Updated tests

* Updated test label

* Added await

* Added new check-scope hook

* Updated tests
  • Loading branch information
hanzlamateen authored May 30, 2024
1 parent 3b192f4 commit 43c7dfa
Show file tree
Hide file tree
Showing 2 changed files with 248 additions and 0 deletions.
207 changes: 207 additions & 0 deletions packages/server-core/src/hooks/check-scope.test.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,207 @@
/*
CPAL-1.0 License
The contents of this file are subject to the Common Public Attribution License
Version 1.0. (the "License"); you may not use this file except in compliance
with the License. You may obtain a copy of the License at
https://github.com/EtherealEngine/etherealengine/blob/dev/LICENSE.
The License is based on the Mozilla Public License Version 1.1, but Sections 14
and 15 have been added to cover use of software over a computer network and
provide for limited attribution for the Original Developer. In addition,
Exhibit A has been modified to be consistent with Exhibit B.
Software distributed under the License is distributed on an "AS IS" basis,
WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License for the
specific language governing rights and limitations under the License.
The Original Code is Ethereal Engine.
The Original Developer is the Initial Developer. The Initial Developer of the
Original Code is the Ethereal Engine team.
All portions of the code written by the Ethereal Engine team are Copyright © 2021-2023
Ethereal Engine. All Rights Reserved.
*/

import { HookContext, Paginated } from '@feathersjs/feathers/lib'
import assert from 'assert'

import { scopePath, ScopeType } from '@etherealengine/common/src/schemas/scope/scope.schema'
import { AvatarID } from '@etherealengine/common/src/schemas/user/avatar.schema'
import { userApiKeyPath, UserApiKeyType } from '@etherealengine/common/src/schemas/user/user-api-key.schema'
import { InviteCode, UserName, userPath, UserType } from '@etherealengine/common/src/schemas/user/user.schema'
import { destroyEngine } from '@etherealengine/ecs/src/Engine'

import { Application } from '../../declarations'
import { createFeathersKoaApp } from '../createApp'
import checkScope from './check-scope'

const mockUserHookContext = (user: UserType, app: Application) => {
return {
app,
params: {
user
}
} as unknown as HookContext<Application>
}

describe('check-scope', () => {
let app: Application
before(async () => {
app = createFeathersKoaApp()
await app.setup()
})

after(() => {
return destroyEngine()
})

it('should return false if user does not have scope', async () => {
const name = `Test #${Math.random()}` as UserName
const isGuest = true

let user = await app.service(userPath).create({
name,
isGuest,
avatarId: '' as AvatarID,
inviteCode: '' as InviteCode,
scopes: []
})

user = await app.service(userPath).get(user.id, { user })

const user1ApiKeys = (await app.service(userApiKeyPath).find({
query: {
userId: user.id
}
})) as Paginated<UserApiKeyType>

user.apiKey = user1ApiKeys.data.length > 0 ? user1ApiKeys.data[0] : user.apiKey

const checkLocationReadScope = checkScope('location', 'read')
const hookContext = mockUserHookContext(user, app)

const hasScope = await checkLocationReadScope(hookContext)
assert.equal(hasScope, false)

// cleanup
await app.service(userPath).remove(user.id!)
})

it('should return true if guest has scope', async () => {
const name = `Test #${Math.random()}` as UserName
const isGuest = true

let user = await app.service(userPath).create({
name,
isGuest,
avatarId: '' as AvatarID,
inviteCode: '' as InviteCode,
scopes: []
})

await app.service(scopePath).create({
type: 'location:read' as ScopeType,
userId: user.id
})

user = await app.service(userPath).get(user.id, { user })

const checkLocationReadScope = checkScope('location', 'read')
const hookContext = mockUserHookContext(user, app)

const hasScope = await checkLocationReadScope(hookContext)
assert.equal(hasScope, true)

// cleanup
await app.service(userPath).remove(user.id!)
})

it('should return true if user has scope', async () => {
const name = `Test #${Math.random()}` as UserName
const isGuest = false

let user = await app.service(userPath).create({
name,
isGuest,
avatarId: '' as AvatarID,
inviteCode: '' as InviteCode,
scopes: []
})

await app.service(scopePath).create({
type: 'location:read' as ScopeType,
userId: user.id
})

user = await app.service(userPath).get(user.id, { user })

const user1ApiKeys = (await app.service(userApiKeyPath).find({
query: {
userId: user.id
}
})) as Paginated<UserApiKeyType>

user.apiKey = user1ApiKeys.data.length > 0 ? user1ApiKeys.data[0] : user.apiKey

const checkLocationReadScope = checkScope('location', 'read')
const hookContext = mockUserHookContext(user, app)

const hasScope = await checkLocationReadScope(hookContext)
assert.equal(hasScope, true)

// cleanup
await app.service(userPath).remove(user.id!)
})

it('should return true if admin', async () => {
const name = `Test #${Math.random()}` as UserName
const isGuest = false

let user = await app.service(userPath).create({
name,
isGuest,
avatarId: '' as AvatarID,
inviteCode: '' as InviteCode,
scopes: []
})

await app.service(scopePath).create({
type: 'location:read' as ScopeType,
userId: user.id
})

await app.service(scopePath).create({
type: 'admin:admin' as ScopeType,
userId: user.id
})

user = await app.service(userPath).get(user.id, { user })

const user1ApiKeys = (await app.service(userApiKeyPath).find({
query: {
userId: user.id
}
})) as Paginated<UserApiKeyType>

user.apiKey = user1ApiKeys.data.length > 0 ? user1ApiKeys.data[0] : user.apiKey

const checkLocationReadScope = checkScope('location', 'read')
const hookContext = mockUserHookContext(user, app)

const hasScope = await checkLocationReadScope(hookContext)
assert.equal(hasScope, true)

// cleanup
await app.service(userPath).remove(user.id!)
})

it('should return true if isInternal', async () => {
const checkLocationReadScope = checkScope('location', 'read')
const hookContext = mockUserHookContext(null!, app)
hookContext.params.isInternal = true

const hasScope = await checkLocationReadScope(hookContext)
assert.equal(hasScope, true)
})
})
41 changes: 41 additions & 0 deletions packages/server-core/src/hooks/check-scope.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
/*
CPAL-1.0 License
The contents of this file are subject to the Common Public Attribution License
Version 1.0. (the "License"); you may not use this file except in compliance
with the License. You may obtain a copy of the License at
https://github.com/EtherealEngine/etherealengine/blob/dev/LICENSE.
The License is based on the Mozilla Public License Version 1.1, but Sections 14
and 15 have been added to cover use of software over a computer network and
provide for limited attribution for the Original Developer. In addition,
Exhibit A has been modified to be consistent with Exhibit B.
Software distributed under the License is distributed on an "AS IS" basis,
WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License for the
specific language governing rights and limitations under the License.
The Original Code is Ethereal Engine.
The Original Developer is the Initial Developer. The Initial Developer of the
Original Code is the Ethereal Engine team.
All portions of the code written by the Ethereal Engine team are Copyright © 2021-2023
Ethereal Engine. All Rights Reserved.
*/

import { HookContext } from '@feathersjs/feathers'

import { Application } from '../../declarations'
import verifyScope from './verify-scope'

export default (currentType: string, scopeToVerify: string) => {
return async (context: HookContext<Application>) => {
try {
await verifyScope(currentType, scopeToVerify)(context)

return true
} catch {
return false
}
}
}

0 comments on commit 43c7dfa

Please sign in to comment.