changes for pro UI

src/features/GridScaling.cpp

@@ -6,6 +6,7 @@
 #include <Geode/binding/LevelEditorLayer.hpp>
 #include <Geode/ui/TextInput.hpp>
 #include <utils/Editor.hpp>
+#include <utils/Pro.hpp>
 
 using namespace geode::prelude;
 
@@ -35,6 +36,9 @@ class $modify(GridUI, EditorUI) {
         if (!Mod::get()->template getSettingValue<bool>("grid-size-controls")) {
             return true;
         }
+        if (be::isProUIEnabled()) {
+            return true;
+        }
 
         // Move position slider a bit out of the way to make space
 

src/utils/Editor.cpp

@@ -215,7 +215,7 @@ class $modify(HideUI, EditorUI) {
         m_currentLayerLabel->setVisible(show);
 
         auto lockBtn = static_cast<CCMenuItemSpriteExtra*>(
-            this->getChildByID("layer-menu")->getChildByID("lock-layer"_spr)
+            this->querySelector("layer-menu hjfod.betteredit/lock-layer")
         );
         if (lockBtn) {
             lockBtn->setVisible(show);

src/utils/Pro.cpp

@@ -0,0 +1,11 @@
+#include "Pro.hpp"
+
+#ifdef BETTEREDIT_PRO
+bool be::isProUIEnabled() {
+    return pro::isNewUIEnabled();
+}
+#else
+bool be::isProUIEnabled() {
+    return false;
+}
+#endif

src/utils/Pro.hpp

@@ -1,5 +1,9 @@
 #pragma once
 
+namespace be {
+    bool isProUIEnabled();
+}
+
 #ifdef BETTEREDIT_PRO
 #include <pro/Pro.hpp>
 #endif