#12 | Missing database name in EE

alan-null · May 19, 2016 · a1982e9 · a1982e9
1 parent ada8eac
commit a1982e9
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/app/scripts.ts/sc_ext.ts b/app/scripts.ts/sc_ext.ts
@@ -802,6 +802,19 @@ namespace SitecoreExtensions {
             if (pageMode == Location.Desktop) {
                 return (document.querySelector('.scDatabaseName') as HTMLDivElement).innerText;
             }
+            if (pageMode == Location.ExperienceEditor) {
+                var webEditRibbonIFrame = (document.querySelector('#scWebEditRibbon') as HTMLIFrameElement)
+                if (webEditRibbonIFrame != null) {
+                    var src = webEditRibbonIFrame.src
+                    var start = src.indexOf("database=");
+                    var end = src.indexOf("&", start);
+                    return src.slice(start + 9, end)
+                }
+                var peBar = document.querySelector('[data-sc-id=PageEditBar]');
+                if (peBar != null) {
+                    return peBar.attributes['data-sc-database'].value
+                }
+            }
             else {
                 var contendDb = <HTMLMetaElement>document.querySelector('[data-sc-name=sitecoreContentDatabase]')
                 if (contendDb != null) {
@@ -828,7 +841,7 @@ namespace SitecoreExtensions {
             if (document.querySelector('input#__FRAMENAME') !== null) {
                 return Location.Desktop;
             }
-            if (document.querySelector('#scWebEditRibbon') !== null) {
+            if (document.querySelector('#scWebEditRibbon') !== null || document.querySelector('[data-sc-id=PageEditBar]') != null) {
                 return Location.ExperienceEditor;
             }
             return Location.Unknown;