sprinklers3/client/state/TokenStore.ts

import { action, observable } from "mobx";

import { Token } from "@client/state/Token";
import { AccessToken, RefreshToken } from "@common/TokenClaims";

const LOCAL_STORAGE_KEY = "TokenStore";

export class TokenStore {
    @observable accessToken: Token<AccessToken> = new Token();
    @observable refreshToken: Token<RefreshToken> = new Token();

    @action
    clearAccessToken() {
        this.accessToken.token = null;
        this.saveLocalStorage();
    }

    @action
    clearAll() {
        this.accessToken.token = null;
        this.refreshToken.token = null;
        this.saveLocalStorage();
    }

    @action
    saveLocalStorage() {
        window.localStorage.setItem(LOCAL_STORAGE_KEY, JSON.stringify(this.toJSON()));
    }

    @action
    loadLocalStorage() {
        const data = window.localStorage.getItem(LOCAL_STORAGE_KEY);
        if (data) {
            const data2 = JSON.parse(data);
            this.updateFromJson(data2);
        }
    }

    toJSON() {
        return { accessToken: this.accessToken.toJSON(), refreshToken: this.refreshToken.toJSON() };
    }

    @action
    updateFromJson(json: any) {
        this.accessToken.token = json.accessToken;
        this.refreshToken.token = json.refreshToken;
    }
}