-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathPayload.ts
More file actions
161 lines (131 loc) · 4.8 KB
/
Payload.ts
File metadata and controls
161 lines (131 loc) · 4.8 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
import { PayloadDto } from "./model/PayloadDto";
import { Utils } from "./QueueITHelpers";
import { TokenSerializationException } from "./Token";
import { Base64 } from "./helpers/Base64";
import { ModeOfOperationCBC } from "./helpers/Aes";
import { TokenOrigin } from "./model/TokenOrigin";
export class Payload {
public static Enqueue(): EnqueueTokenPayloadGenerator {
return new EnqueueTokenPayloadGenerator();
}
}
export class EnqueueTokenPayloadGenerator {
private _payload: EnqueueTokenPayload
constructor() {
this._payload = new EnqueueTokenPayload();
}
public WithKey(key: string): EnqueueTokenPayloadGenerator {
this._payload = EnqueueTokenPayload.create(this._payload, key);
return this;
}
public WithRelativeQuality(relativeQuality: number): EnqueueTokenPayloadGenerator {
this._payload = EnqueueTokenPayload.create(this._payload, null, relativeQuality);
return this;
}
public WithCustomData(key: string, value: string): EnqueueTokenPayloadGenerator {
this._payload = EnqueueTokenPayload.create(this._payload, null);
this._payload.AddCustomData(key, value);
return this;
}
public WithOrigin(origin: TokenOrigin): EnqueueTokenPayloadGenerator {
this._payload = EnqueueTokenPayload.create(this._payload, null);
this._payload.AddTokenOrigin(origin);
return this;
}
public Generate(): IEnqueueTokenPayload {
return this._payload;
}
}
export interface IEnqueueTokenPayload {
readonly Key: string
readonly RelativeQuality?: number
readonly CustomData: Object
readonly TokenOrigin: TokenOrigin
EncryptAndEncode(secretKey: string, tokenIdentifier: string): string
Serialize(): Uint8Array;
}
export class EnqueueTokenPayload implements IEnqueueTokenPayload {
private _customData: Object;
private _key: string;
private _relativeQuality: number
private _origin: TokenOrigin
public get Key(): string {
return this._key;
}
private set Key(value: string) {
this._key = value;
}
public get CustomData(): Object {
return this._customData;
}
public get RelativeQuality(): number {
return this._relativeQuality;
}
private set RelativeQuality(value: number) {
this._relativeQuality = value;
}
public get TokenOrigin(): TokenOrigin {
return this._origin;
}
constructor() {
this._customData = {};
this._origin = TokenOrigin.Connector;
}
static create(payload?: EnqueueTokenPayload, key?: string, relativeQuality?: number, customData?: object, origin?: TokenOrigin): EnqueueTokenPayload {
const newPayload = new EnqueueTokenPayload();
newPayload.Key = key;
if (payload) {
newPayload.RelativeQuality = payload.RelativeQuality;
newPayload._customData = payload._customData;
newPayload._origin = payload._origin;
if (!key || key.length == 0) {
newPayload.Key = payload.Key;
}
}
if (relativeQuality != null) {
newPayload.RelativeQuality = relativeQuality
}
if (customData) {
newPayload._customData = customData;
}
if (origin) {
newPayload._origin = origin;
}
return newPayload;
}
public AddCustomData(key: string, value: string): EnqueueTokenPayload {
if (!this._customData) {
this._customData = {};
}
this._customData[key] = value;
return this;
}
public AddTokenOrigin(origin: TokenOrigin): EnqueueTokenPayload {
this._origin = origin ?? TokenOrigin.Connector;
return this;
}
public Serialize(): Uint8Array {
const dto = new PayloadDto();
dto.Key = this.Key;
dto.RelativeQuality = this.RelativeQuality;
dto.CustomData = this.CustomData;
dto.Origin = this.TokenOrigin;
return dto.Serialize();
}
static Deserialize(input: string, secretKey: string, tokenIdentifier: string): EnqueueTokenPayload {
const dto = PayloadDto.DeserializePayload(input, secretKey, tokenIdentifier);
return EnqueueTokenPayload.create(null, dto.Key, dto.RelativeQuality, dto.CustomData, dto.Origin);
}
EncryptAndEncode(secretKey: string, tokenIdentifier: string): string {
try {
const serializedPayload = this.Serialize();
const key = Utils.generateKey(secretKey);
const iv = Utils.generateIV(tokenIdentifier);
const aesCBC = new ModeOfOperationCBC(key, iv)
const encrypted: Uint8Array = aesCBC.encrypt(serializedPayload);
return Base64.encode(encrypted);
} catch (ex) {
throw new TokenSerializationException(ex);
}
}
}