import RSVP from 'rsvp';
import { computed } from '@ember/object';
import { inject as service } from '@ember/service';
import { later, cancel, scheduleOnce, next } from '@ember/runloop';
import { isPresent, typeOf, isEmpty, isNone } from '@ember/utils';
import { A } from '@ember/array';
import { getOwner } from '@ember/application';
import { warn } from '@ember/debug';
import BaseStore from './base';
import objectsAreEqual from '../utils/objects-are-equal';
import { isTesting } from '@embroider/macros';
const persistingProperty = function (beforeSet = function () {}) {
return computed({
get(key) {
return this.get(`_${key}`);
},
set(key, value) {
beforeSet.apply(this, [key, value]);
this.set(`_${key}`, value);
scheduleOnce('actions', this, this.rewriteCookie);
return value;
},
});
};
/**
Session store that persists data in a cookie.
By default the cookie session store uses a session cookie that expires and is
deleted when the browser is closed. The cookie expiration period can be
configured by setting the
{@linkplain CookieStore.cookieExpirationTime}
property. This can be used to implement "remember me" functionality that will
either store the session persistently or in a session cookie depending on
whether the user opted in or not:
```js
// app/controllers/login.js
import Controller from '@ember/controller';
import { inject as service } from '@ember/service';
export default class LoginController extends Controller {
@service session;
_rememberMe = false;
get rememberMe() {
return this._rememberMe;
}
set rememberMe(value) {
let expirationTime = value ? (14 * 24 * 60 * 60) : null;
this.set('session.store.cookieExpirationTime', expirationTime);
this._rememberMe = value;
}
}
```
__Applications that use FastBoot must use this session store by defining the
application session store like this:__
```js
// app/session-stores/application.js
import CookieStore from 'ember-simple-auth/session-stores/cookie';
export default class ApplicationSessionStore extends CookieStore {}
```
@class CookieStore
@extends BaseStore
@public
*/
export default BaseStore.extend({
_syncDataTimeout: null,
_renewExpirationTimeout: null,
/**
The domain to use for the cookie, e.g., "example.com", ".example.com"
(which includes all subdomains) or "subdomain.example.com". If not
explicitly set, the cookie domain defaults to the domain the session was
authenticated on.
@memberof CookieStore
@property cookieDomain
@type String
@default null
@public
*/
_cookieDomain: null,
cookieDomain: persistingProperty(),
/**
Allows servers to assert that a cookie ought not to be sent along with cross-site requests,
which provides some protection against cross-site request forgery attacks (CSRF).
Available options:
- "Strict"
- "Lax"
@memberof CookieStore
@property sameSite
@type String
@default null
@public
*/
_sameSite: null,
sameSite: persistingProperty(),
/**
The name of the cookie.
@memberof CookieStore
@property cookieName
@type String
@default ember_simple_auth-session
@public
*/
_cookieName: 'ember_simple_auth-session',
cookieName: persistingProperty(function () {
this._oldCookieName = this._cookieName;
}),
/**
The path to use for the cookie, e.g., "/", "/something".
@memberof CookieStore
@property cookiePath
@type String
@default '/'
@public
*/
_cookiePath: '/',
cookiePath: persistingProperty(),
/**
The expiration time for the cookie in seconds. A value of `null` will make
the cookie a session cookie that expires and gets deleted when the browser
is closed.
The recommended minimum value is 90 seconds. If your value is less than
that, the cookie may expire before its expiration time is extended
(expiration time is extended every 60 seconds).
@memberof CookieStore
@property cookieExpirationTime
@default null
@type Integer
@public
*/
_cookieExpirationTime: null,
cookieExpirationTime: persistingProperty(function (key, value) {
// When nulling expiry time on purpose, we need to clear the cached value.
// Otherwise, `_calculateExpirationTime` will reuse it.
if (isNone(value)) {
this.get('_cookies').clear(`${this.get('cookieName')}-expiration_time`);
} else if (value < 90) {
warn(
'The recommended minimum value for `cookieExpirationTime` is 90 seconds. If your value is less than that, the cookie may expire before its expiration time is extended (expiration time is extended every 60 seconds).',
false,
{ id: 'ember-simple-auth.cookieExpirationTime' }
);
}
}),
_cookies: service('cookies'),
_secureCookies() {
if (this.get('_fastboot.isFastBoot')) {
return this.get('_fastboot.request.protocol') === 'https:';
}
return window.location.protocol === 'https:';
},
_isPageVisible() {
if (this.get('_fastboot.isFastBoot')) {
return false;
} else {
const visibilityState =
typeof document !== 'undefined' ? document.visibilityState || 'visible' : false;
return visibilityState === 'visible';
}
},
init() {
this._super(...arguments);
let owner = getOwner(this);
if (owner && !this.hasOwnProperty('_fastboot')) {
this._fastboot = owner.lookup('service:fastboot');
}
let cachedExpirationTime = this._read(`${this.get('cookieName')}-expiration_time`);
if (cachedExpirationTime) {
this.set('cookieExpirationTime', parseInt(cachedExpirationTime, 10));
}
if (!this.get('_fastboot.isFastBoot')) {
next(() => {
this._syncData().then(() => {
this._renewExpiration();
});
});
} else {
this._renew();
}
},
/**
Persists the `data` in the cookie.
@memberof CookieStore
@method persist
@param {Object} data The data to persist
@return {RSVP.Promise} A promise that resolves when the data has successfully been persisted and rejects otherwise.
@public
*/
persist(data) {
this._lastData = data;
data = JSON.stringify(data || {});
let expiration = this._calculateExpirationTime();
this._write(data, expiration);
return RSVP.resolve();
},
/**
Returns all data currently stored in the cookie as a plain object.
@memberof CookieStore
@method restore
@return {RSVP.Promise} A promise that resolves with the data currently persisted in the store when the data has been restored successfully and rejects otherwise.
@public
*/
restore() {
let data = this._read(this.get('cookieName'));
if (isEmpty(data)) {
return RSVP.resolve({});
} else {
return RSVP.resolve(JSON.parse(data));
}
},
/**
Clears the store by deleting the cookie.
@memberof CookieStore
@method clear
@return {RSVP.Promise} A promise that resolves when the store has been cleared successfully and rejects otherwise.
@public
*/
clear() {
this._write('', 0);
this._lastData = {};
return RSVP.resolve();
},
_read(name) {
return this.get('_cookies').read(name) || '';
},
_calculateExpirationTime() {
let cachedExpirationTime = this._read(`${this.get('cookieName')}-expiration_time`);
cachedExpirationTime = cachedExpirationTime
? new Date().getTime() + cachedExpirationTime * 1000
: null;
return this.get('cookieExpirationTime')
? new Date().getTime() + this.get('cookieExpirationTime') * 1000
: cachedExpirationTime;
},
_write(value, expiration) {
let cookieOptions = {
domain: this.get('cookieDomain'),
expires: isEmpty(expiration) ? null : new Date(expiration),
path: this.get('cookiePath'),
secure: this._secureCookies(),
sameSite: this.get('sameSite'),
};
if (this._oldCookieName) {
A([this._oldCookieName, `${this._oldCookieName}-expiration_time`]).forEach(oldCookie => {
this.get('_cookies').clear(oldCookie);
});
delete this._oldCookieName;
}
this.get('_cookies').write(this.get('cookieName'), value, cookieOptions);
if (!isEmpty(expiration)) {
let expirationCookieName = `${this.get('cookieName')}-expiration_time`;
let cachedExpirationTime = this.get('_cookies').read(expirationCookieName);
this.get('_cookies').write(
expirationCookieName,
this.get('cookieExpirationTime') || cachedExpirationTime,
cookieOptions
);
}
},
_syncData() {
return this.restore().then(data => {
if (!objectsAreEqual(data, this._lastData)) {
this._lastData = data;
this.trigger('sessionDataUpdated', data);
}
if (!isTesting()) {
cancel(this._syncDataTimeout);
this._syncDataTimeout = later(this, this._syncData, 500);
}
});
},
_renew() {
return this.restore().then(data => {
if (!isEmpty(data) && !(data.constructor === Object && Object.keys(data).length === 0)) {
data = typeOf(data) === 'string' ? data : JSON.stringify(data || {});
let expiration = this._calculateExpirationTime();
this._write(data, expiration);
}
});
},
_renewExpiration() {
if (!isTesting()) {
cancel(this._renewExpirationTimeout);
this._renewExpirationTimeout = later(this, this._renewExpiration, 60000);
}
if (this._isPageVisible()) {
return this._renew();
} else {
return RSVP.resolve();
}
},
rewriteCookie() {
// if `cookieName` has not been renamed, `oldCookieName` will be nil
const cookieName = this._oldCookieName || this._cookieName;
const data = this._read(cookieName);
if (isPresent(data)) {
const expiration = this._calculateExpirationTime();
this._write(data, expiration);
}
},
});
Source