sprinklers3/client/state/TokenStore.ts

import { action, observable } from "mobx";

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

const LOCAL_STORAGE_KEY = "TokenStore";

export class TokenStore {
  @observable
  accessToken: Token<AccessToken & BaseClaims> = new Token();
  @observable
  refreshToken: Token<RefreshToken & BaseClaims> = 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;
  }
}