2 files changed, 34 insertions, 0 deletions
diff --git a/app/models/extensions/with-environment.ts b/app/models/extensions/with-environment.ts
new file mode 100644
index 0000000..9fe5fd5
--- /dev/null
+++ b/app/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/models/extensions/with-root-store.ts b/app/models/extensions/with-root-store.ts
new file mode 100644
index 0000000..eff769c
--- /dev/null
+++ b/app/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)
+    },
+  },
+})