UNPKG

@wener/console

Version:
382 lines (381 loc) 12.8 kB
function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } } function _async_to_generator(fn) { return function() { var self = this, args = arguments; return new Promise(function(resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; } function _define_property(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; } function _object_spread(target) { for(var i = 1; i < arguments.length; i++){ var source = arguments[i] != null ? arguments[i] : {}; var ownKeys = Object.keys(source); if (typeof Object.getOwnPropertySymbols === "function") { ownKeys = ownKeys.concat(Object.getOwnPropertySymbols(source).filter(function(sym) { return Object.getOwnPropertyDescriptor(source, sym).enumerable; })); } ownKeys.forEach(function(key) { _define_property(target, key, source[key]); }); } return target; } function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); if (enumerableOnly) { symbols = symbols.filter(function(sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; }); } keys.push.apply(keys, symbols); } return keys; } function _object_spread_props(target, source) { source = source != null ? source : {}; if (Object.getOwnPropertyDescriptors) { Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)); } else { ownKeys(Object(source)).forEach(function(key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } return target; } function _ts_generator(thisArg, body) { var f, y, t, _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype), d = Object.defineProperty; return d(g, "next", { value: verb(0) }), d(g, "throw", { value: verb(1) }), d(g, "return", { value: verb(2) }), typeof Symbol === "function" && d(g, Symbol.iterator, { value: function() { return this; } }), g; function verb(n) { return function(v) { return step([ n, v ]); }; } function step(op) { if (f) throw new TypeError("Generator is already executing."); while(g && (g = 0, op[0] && (_ = 0)), _)try { if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t; if (y = 0, t) op = [ op[0] & 2, t.value ]; switch(op[0]){ case 0: case 1: t = op; break; case 4: _.label++; return { value: op[1], done: false }; case 5: _.label++; y = op[1]; op = [ 0 ]; continue; case 7: op = _.ops.pop(); _.trys.pop(); continue; default: if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; } if (op[0] === 3 && (!t || op[1] > t[0] && op[1] < t[3])) { _.label = op[1]; break; } if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; } if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; } if (t[2]) _.ops.pop(); _.trys.pop(); continue; } op = body.call(thisArg, _); } catch (e) { op = [ 6, e ]; y = 0; } finally{ f = t = 0; } if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true }; } } import { use, useEffect, useRef } from 'react'; import { createReactContext } from '@wener/reaction'; import { useNetworkStatus } from '@wener/reaction/store'; import { createBoundedUseStore } from '@wener/reaction/zustand'; import { getGlobalStates } from '@wener/utils'; import { createStore } from 'zustand'; import { mutative } from 'zustand-mutative'; export var AuthStatus = { Init: 'Init', Authenticated: 'Authenticated', Unauthenticated: 'Unauthenticated', Expired: 'Expired', Loading: 'Loading', Error: 'Error', Locked: 'Locked' }; export function createAuthStore() { var init = arguments.length > 0 && arguments[0] !== void 0 ? arguments[0] : {}; return createStore(mutative(function(setState, getState, store) { return _object_spread_props(_object_spread({}, init), { status: AuthStatus.Init, setAuth: function setAuth(o) { setState(function(s) { Object.assign(s, _object_spread_props(_object_spread({ status: AuthStatus.Authenticated }, o), { expiresAt: o.expiresAt ? new Date(o.expiresAt) : undefined })); }); }, reset: function reset() { setState(function(s) { Object.assign(s, { status: AuthStatus.Unauthenticated, accessToken: undefined, refreshToken: undefined, expiresIn: undefined, expiresAt: undefined, error: undefined }); }); } }); })); } function useAuthSidecar(param) { var store = param.store, refresh = param.actions.refresh, _param_storage = param.storage, storage = _param_storage === void 0 ? localStorage : _param_storage; var online = useNetworkStatus().online; // todo watch storage ? var checkAuth = function() { return _async_to_generator(function() { var accessToken, _storage_getItem, refreshToken, out, e; return _ts_generator(this, function(_state) { switch(_state.label){ case 0: accessToken = storage.getItem('accessToken'); refreshToken = (_storage_getItem = storage.getItem('refreshToken')) !== null && _storage_getItem !== void 0 ? _storage_getItem : undefined; if (!accessToken) return [ 3, 4 ]; _state.label = 1; case 1: _state.trys.push([ 1, 3, , 4 ]); return [ 4, refresh({ accessToken: accessToken, refreshToken: refreshToken }) ]; case 2: out = _state.sent(); store.getState().setAuth(out); return [ 2, true ]; case 3: e = _state.sent(); console.error('Failed to refresh token', e); return [ 3, 4 ]; case 4: // not authenticated or server error store.getState().reset(); return [ 2, false ]; } }); })(); }; var authRef = useRef(undefined); var doAuthCheck = function() { var state = store.getState(); // only check for init and authenticated switch(state.status){ case AuthStatus.Init: case AuthStatus.Authenticated: break; default: return; } // avoid race var current = authRef.current; if (current) { return current.then(function(v) { // authed, skip for now, will check for next if (v) { authRef.current = undefined; } else { // check again return current = checkAuth(); } return v; }); } else { return authRef.current = checkAuth().then(function(v) { // done authRef.current = undefined; return v; }); } }; // auth check useEffect(function() { if (!online) return; doAuthCheck(); var timer = setInterval(doAuthCheck, 5 * 60 * 1000); return function() { clearInterval(timer); }; }, [ store, online ]); // useAuthTokenPersist(store, storage); } var AuthStoreStateKey = 'AuthStore'; export var AuthSidecar = function(props) { var store = useAuthStoreContext(); useAuthSidecar(_object_spread({ store: store }, props)); return null; }; function useAuthTokenPersist(store, storage) { var ref = useRef(storage); ref.current = storage; // persist token useEffect(function() { return store.subscribe(function(s) { if (s.status !== AuthStatus.Authenticated) { return; } var _s_accessToken = s.accessToken, accessToken = _s_accessToken === void 0 ? '' : _s_accessToken, refreshToken = s.refreshToken; var _$storage = ref.current; var _storage_getItem; if (((_storage_getItem = _$storage.getItem('accessToken')) !== null && _storage_getItem !== void 0 ? _storage_getItem : '') === accessToken) { return; } accessToken ? _$storage.setItem('accessToken', accessToken) : deleteItem(_$storage, 'accessToken'); refreshToken ? _$storage.setItem('refreshToken', refreshToken) : deleteItem(_$storage, 'refreshToken'); }); }, [ store ]); } export function getAuthStore() { return getGlobalStates(AuthStoreStateKey, function() { return createAuthStore(); }); } export function getAuthState() { return getAuthStore().getState(); } export function getAccessToken() { return getAuthState().accessToken; } export var AuthStoreContext = createReactContext('AuthStore', undefined); export function useAuthStoreContext() { return use(AuthStoreContext) || getAuthStore(); } function deleteItem(s, key) { if ('removeItem' in s) { s.removeItem(key); } else { delete s[key]; } } export var useAuthStore = createBoundedUseStore(useAuthStoreContext);