WIP

2 files changed, 34 insertions, 0 deletions

diff --git a/app_expo/models/extensions/with-environment.ts b/app_expo/models/extensions/with-environment.ts
new file mode 100644
index 0000000..1fc190e
--- /dev/null
+++ b/app_expo/models/extensions/with-environment.ts
@@ -0,0 +1,17 @@
+import { getEnv, IStateTreeNode } from 'mobx-state-tree'
+import { Environment } from '../environment'
+
+/**
+ * Adds a environment property to the node for accessing our
+ * Environment in strongly typed.
+ */
+export const withEnvironment = (self: IStateTreeNode) => ({
+  views: {
+    /**
+     * The environment.
+     */
+    get environment() {
+      return getEnv<Environment>(self)
+    },
+  },
+})
diff --git a/app_expo/models/extensions/with-root-store.ts b/app_expo/models/extensions/with-root-store.ts
new file mode 100644
index 0000000..72ab478
--- /dev/null
+++ b/app_expo/models/extensions/with-root-store.ts
@@ -0,0 +1,17 @@
+import { getRoot, IStateTreeNode } from 'mobx-state-tree'
+import { RootStoreModel } from '../root-store/root-store'
+
+/**
+ * Adds a rootStore property to the node for a convenient
+ * and strongly typed way for stores to access other stores.
+ */
+export const withRootStore = (self: IStateTreeNode) => ({
+  views: {
+    /**
+     * The root store.
+     */
+    get rootStore() {
+      return getRoot<typeof RootStoreModel>(self)
+    },
+  },
+})