first commit

wangzhibo

2026-05-09 53418655c061ac26f66058253f39200aced1ab1d

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
/*
 * Copyright (c) Microsoft Corporation. All rights reserved.
 * Licensed under the MIT License.
 */
 
import { RESOURCE_DELIM } from "./Constants.js";
import { ICrypto } from "../crypto/ICrypto.js";
import {
    ClientAuthErrorCodes,
    createClientAuthError,
} from "../error/ClientAuthError.js";
import { LibraryStateObject, RequestStateObject } from "./StateTypes.js";
 
/**
 * Appends user state with random guid, or returns random guid.
 * @param cryptoObj
 * @param userState
 * @param meta
 */
export function setRequestState(
    cryptoObj: ICrypto,
    userState?: string,
    meta?: Record<string, string>
): string {
    const libraryState = generateLibraryState(cryptoObj, meta);
    return userState
        ? `${libraryState}${RESOURCE_DELIM}${userState}`
        : libraryState;
}
 
/**
 * Generates the state value used by the common library.
 * @param cryptoObj
 * @param meta
 */
export function generateLibraryState(
    cryptoObj: ICrypto,
    meta?: Record<string, string>
): string {
    if (!cryptoObj) {
        throw createClientAuthError(ClientAuthErrorCodes.noCryptoObject);
    }
 
    // Create a state object containing a unique id and the timestamp of the request creation
    const stateObj: LibraryStateObject = {
        id: cryptoObj.createNewGuid(),
    };
 
    if (meta) {
        stateObj.meta = meta;
    }
 
    const stateString = JSON.stringify(stateObj);
 
    return cryptoObj.base64Encode(stateString);
}
 
/**
 * Parses the state into the RequestStateObject, which contains the LibraryState info and the state passed by the user.
 * @param base64Decode
 * @param state
 */
export function parseRequestState(
    base64Decode: (input: string) => string,
    state: string
): RequestStateObject {
    if (!base64Decode) {
        throw createClientAuthError(ClientAuthErrorCodes.noCryptoObject);
    }
 
    if (!state) {
        throw createClientAuthError(ClientAuthErrorCodes.invalidState);
    }
 
    try {
        // Split the state between library state and user passed state and decode them separately
        const splitState = state.split(RESOURCE_DELIM);
        const libraryState = splitState[0];
        const userState =
            splitState.length > 1
                ? splitState.slice(1).join(RESOURCE_DELIM)
                : "";
        const libraryStateString = base64Decode(libraryState);
        const libraryStateObj = JSON.parse(
            libraryStateString
        ) as LibraryStateObject;
        return {
            userRequestState: userState || "",
            libraryState: libraryStateObj,
        };
    } catch (e) {
        throw createClientAuthError(ClientAuthErrorCodes.invalidState);
    }
}