discourse/test/javascripts/helpers/create-store.js

60 lines
1.9 KiB
JavaScript

import Store from "discourse/models/store";
import RestAdapter from "discourse/adapters/rest";
import KeyValueStore from "discourse/lib/key-value-store";
import TopicListAdapter from "discourse/adapters/topic-list";
import TopicTrackingState from "discourse/models/topic-tracking-state";
import { buildResolver } from "discourse-common/resolver";
import { currentSettings } from "helpers/site-settings";
const CatAdapter = RestAdapter.extend({
primaryKey: "cat_id",
});
export default function (customLookup = () => {}) {
const resolver = buildResolver("discourse").create();
return Store.create({
register: {
lookup(type) {
if (type === "adapter:cat") {
this._catAdapter =
this._catAdapter || CatAdapter.create({ owner: this });
return this._catAdapter;
}
if (type === "adapter:rest") {
if (!this._restAdapter) {
this._restAdapter = RestAdapter.create({ owner: this });
}
return this._restAdapter;
}
if (type === "adapter:topicList") {
this._topicListAdapter =
this._topicListAdapter || TopicListAdapter.create({ owner: this });
return this._topicListAdapter;
}
if (type === "key-value-store:main") {
this._kvs = this._kvs || new KeyValueStore();
return this._kvs;
}
if (type === "topic-tracking-state:main") {
this._tracker = this._tracker || TopicTrackingState.create();
return this._tracker;
}
if (type === "site-settings:main") {
this._settings = this._settings || currentSettings();
return this._settings;
}
return customLookup(type);
},
lookupFactory(type) {
const split = type.split(":");
return resolver.customResolve({
type: split[0],
fullNameWithoutType: split[1],
});
},
},
});
}