-
Notifications
You must be signed in to change notification settings - Fork 9
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge remote-tracking branch 'origin/master' into mc_presenceEE_into_…
…subscribe_based_on_branch_mc_PresenceEE_effects
- Loading branch information
Showing
28 changed files
with
404 additions
and
140 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
120 changes: 106 additions & 14 deletions
120
src/main/kotlin/com/pubnub/api/managers/TokenParser.kt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,29 +1,121 @@ | ||
| package com.pubnub.api.managers | ||
|
|
||
| import com.fasterxml.jackson.annotation.JsonSetter | ||
| import com.fasterxml.jackson.annotation.Nulls | ||
| import com.fasterxml.jackson.databind.DeserializationFeature | ||
| import com.fasterxml.jackson.databind.ObjectMapper | ||
| import com.fasterxml.jackson.dataformat.cbor.CBORFactory | ||
| import co.nstant.`in`.cbor.CborDecoder | ||
| import co.nstant.`in`.cbor.model.ByteString | ||
| import co.nstant.`in`.cbor.model.NegativeInteger | ||
| import co.nstant.`in`.cbor.model.UnsignedInteger | ||
| import com.pubnub.api.PubNubError | ||
| import com.pubnub.api.PubNubException | ||
| import com.pubnub.api.models.consumer.access_manager.v3.PNToken | ||
| import com.pubnub.api.vendor.Base64 | ||
| import java.io.IOException | ||
| import java.math.BigInteger | ||
| import java.nio.charset.StandardCharsets | ||
| import kotlin.collections.Map | ||
| import kotlin.collections.component1 | ||
| import kotlin.collections.component2 | ||
| import kotlin.collections.set | ||
| import co.nstant.`in`.cbor.model.Map as CborMap | ||
|
|
||
| internal class TokenParser { | ||
| private val mapper = ObjectMapper(CBORFactory()).apply { | ||
| configOverride(Map::class.java).setterInfo = JsonSetter.Value.forValueNulls(Nulls.AS_EMPTY) | ||
| configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false) | ||
| } | ||
|
|
||
| fun unwrapToken(token: String): PNToken { | ||
| val byteArray = Base64.decode(token.toByteArray(StandardCharsets.UTF_8), Base64.URL_SAFE) | ||
| val firstElement = CborDecoder(byteArray.inputStream()).decode().firstOrNull() ?: throw PubNubException( | ||
| pubnubError = PubNubError.INVALID_ACCESS_TOKEN, errorMessage = "Empty token" | ||
| ) | ||
| val firstLevelMap = (firstElement as? CborMap)?.toJvmMap() ?: throw PubNubException( | ||
| pubnubError = PubNubError.INVALID_ACCESS_TOKEN, errorMessage = "First element is not a map" | ||
| ) | ||
| val version = firstLevelMap[VERSION_KEY]?.toString()?.toInt() ?: throw PubNubException( | ||
| pubnubError = PubNubError.INVALID_ACCESS_TOKEN, errorMessage = "Couldn't parse version" | ||
| ) | ||
| val timestamp = firstLevelMap[TIMESTAMP_KEY]?.toString()?.toLong() ?: throw PubNubException( | ||
| pubnubError = PubNubError.INVALID_ACCESS_TOKEN, errorMessage = "Couldn't parse timestamp" | ||
| ) | ||
| val ttl = firstLevelMap[TTL_KEY]?.toString()?.toLong() ?: throw PubNubException( | ||
| pubnubError = PubNubError.INVALID_ACCESS_TOKEN, errorMessage = "Couldn't parse ttl" | ||
| ) | ||
| val resourcesValue = firstLevelMap[RESOURCES_KEY] as? Map<*, *> ?: throw PubNubException( | ||
| pubnubError = PubNubError.INVALID_ACCESS_TOKEN, errorMessage = "Resources are not present or are not map" | ||
| ) | ||
| val patternsValue = firstLevelMap[PATTERNS_KEY] as? Map<*, *> ?: throw PubNubException( | ||
| pubnubError = PubNubError.INVALID_ACCESS_TOKEN, errorMessage = "Patterns are not present or are not map" | ||
| ) | ||
|
|
||
| return try { | ||
| val byteArray = Base64.decode(token.toByteArray(StandardCharsets.UTF_8), Base64.URL_SAFE) | ||
| mapper.readValue(byteArray, PNToken::class.java) | ||
| } catch (e: IOException) { | ||
| throw PubNubException(pubnubError = PubNubError.INVALID_ACCESS_TOKEN, errorMessage = e.message) | ||
|
|
||
| PNToken( | ||
| version = version, | ||
| timestamp = timestamp, | ||
| ttl = ttl, | ||
| authorizedUUID = firstLevelMap[AUTHORIZED_UUID_KEY]?.toString(), | ||
| resources = resourcesValue.toPNTokenResources(), | ||
| patterns = patternsValue.toPNTokenResources(), | ||
| meta = firstLevelMap[META_KEY] | ||
| ) | ||
| } catch (e: Exception) { | ||
| if (e is PubNubException) throw e | ||
| throw PubNubException( | ||
| pubnubError = PubNubError.INVALID_ACCESS_TOKEN, errorMessage = "Couldn't parse token: ${e.message}" | ||
| ) | ||
| } | ||
| } | ||
|
|
||
| private fun CborMap.toJvmMap(depth: Int = 0): MutableMap<String, Any> { | ||
| if (depth > 3) { | ||
| throw PubNubException(pubnubError = PubNubError.INVALID_ACCESS_TOKEN, errorMessage = "Token is too deep") | ||
| } | ||
| val result = mutableMapOf<String, Any>() | ||
| for (key in this.keys) { | ||
| val value = this.get(key) | ||
| val keyString = when (key) { | ||
| is ByteString -> key.bytes.toString(StandardCharsets.UTF_8) | ||
| else -> key.toString() | ||
| } | ||
|
|
||
| when (value) { | ||
| is CborMap -> result[keyString] = value.toJvmMap(depth + 1) | ||
| is ByteString -> result[keyString] = value.bytes | ||
| is List<*> -> result[keyString] = value.map { it.toString() } | ||
| is UnsignedInteger -> result[keyString] = value.value | ||
| is NegativeInteger -> result[keyString] = value.value | ||
| else -> result[keyString] = value.toString() | ||
| } | ||
| } | ||
| return result | ||
| } | ||
|
|
||
| private fun Map<*, *>.toMapOfStringToInt(): Map<String, Int> { | ||
| return mapNotNull { (k, v) -> | ||
| when (v) { | ||
| is BigInteger -> k.toString() to v.toInt() | ||
| else -> v.toString().toIntOrNull()?.let { k.toString() to it } | ||
| } | ||
| }.toMap() | ||
| } | ||
|
|
||
| private fun Map<*, *>.toPNTokenResources(): PNToken.PNTokenResources { | ||
| val channels = (this[CHANNELS_KEY] as? Map<*, *>)?.toMapOfStringToInt() ?: emptyMap() | ||
| val groups = (this[GROUPS_KEY] as? Map<*, *>)?.toMapOfStringToInt() ?: emptyMap() | ||
| val uuids = (this[UUIDS_KEY] as? Map<*, *>)?.toMapOfStringToInt() ?: emptyMap() | ||
|
|
||
| return PNToken.PNTokenResources( | ||
| channels = channels.mapValues { (_, v) -> PNToken.PNResourcePermissions(v) }, | ||
| channelGroups = groups.mapValues { (_, v) -> PNToken.PNResourcePermissions(v) }, | ||
| uuids = uuids.mapValues { (_, v) -> PNToken.PNResourcePermissions(v) } | ||
| ) | ||
| } | ||
|
|
||
| companion object { | ||
| private const val VERSION_KEY = "v" | ||
| private const val TIMESTAMP_KEY = "t" | ||
| private const val TTL_KEY = "ttl" | ||
| private const val AUTHORIZED_UUID_KEY = "uuid" | ||
| private const val RESOURCES_KEY = "res" | ||
| private const val PATTERNS_KEY = "pat" | ||
| private const val META_KEY = "meta" | ||
| private const val CHANNELS_KEY = "chan" | ||
| private const val GROUPS_KEY = "grp" | ||
| private const val UUIDS_KEY = "uuid" | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.