From deafe43cdf67d4d17d4be09d0b46bdd07b10fd8e Mon Sep 17 00:00:00 2001
From: Olivier Tassinari <olivier.tassinari@gmail.com>
Date: Sun, 16 Aug 2020 19:54:10 +0200
Subject: [PATCH] [docs] Improve codesandbox generation logic (#22221)

---
 docs/src/modules/utils/helpers.js      | 20 +++++++++++++++++++-
 docs/src/modules/utils/helpers.test.js |  1 +
 2 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/docs/src/modules/utils/helpers.js b/docs/src/modules/utils/helpers.js
index c4ef4d8af5f619..ef945aa116193d 100644
--- a/docs/src/modules/utils/helpers.js
+++ b/docs/src/modules/utils/helpers.js
@@ -79,9 +79,27 @@ function includePeerDependencies(deps, versions) {
     react: versions.react,
   });
 
-  if (deps['@material-ui/lab'] && !deps['@material-ui/core']) {
+  if (
+    deps['@material-ui/lab'] ||
+    deps['@material-ui/pickers'] ||
+    deps['@material-ui/x'] ||
+    deps['@material-ui/x-grid'] ||
+    deps['@material-ui/x-pickers'] ||
+    deps['@material-ui/x-tree-view'] ||
+    deps['@material-ui/data-grid']
+  ) {
     deps['@material-ui/core'] = versions['@material-ui/core'];
   }
+
+  if (deps['@material-ui/x-data-grid-generator']) {
+    deps['@material-ui/core'] = versions['@material-ui/core'];
+    deps['@material-ui/icons'] = versions['@material-ui/icons'];
+    deps['@material-ui/lab'] = versions['@material-ui/lab'];
+  }
+
+  if (deps['@material-ui/pickers']) {
+    deps['date-fns'] = 'latest';
+  }
 }
 
 /**
diff --git a/docs/src/modules/utils/helpers.test.js b/docs/src/modules/utils/helpers.test.js
index ebb96db8864ebb..13c39b09561626 100644
--- a/docs/src/modules/utils/helpers.test.js
+++ b/docs/src/modules/utils/helpers.test.js
@@ -116,6 +116,7 @@ import {
 
     expect(getDependencies(source)).to.deep.equal({
       'date-fns': 'latest',
+      '@material-ui/core': 'latest',
       '@material-ui/pickers': 'latest',
       react: 'latest',
       'react-dom': 'latest',