1 files changed, 24 insertions, 14 deletions

diff --git a/src/main.c b/src/main.c
index 1955e35..c96a9b7 100644
--- a/src/main.c
+++ b/src/main.c
@@ -4,7 +4,6 @@
 
 #include "editor.h"
 #include "raylib.h"
-#include "screens.h"
 #include "xd.h"
 
 int main() {
@@ -33,21 +32,32 @@ int main() {
   camera.rotation = 0.0f;
   camera.zoom = 4.0f;
 
-  GameScreen currentScreen = SCREEN_MENU;
-
   while (!WindowShouldClose()) {
-    switch (currentScreen) {
-      case SCREEN_MENU: {
-        SE_DrawMenuScreen(&currentScreen, &editor, data, datas);
-        break;
-      }
-      case SCREEN_EDITOR: {
-        SE_DrawEditorScreen(&currentScreen, &editor, &camera);
-        break;
-      }
-      default:
-        break;
+    if (GetMouseWheelMove() != 0.0) {
+      camera.zoom += (int)GetMouseWheelMove();
+
+      if (camera.zoom > 6.0f)
+        camera.zoom = 6.0f;
+      else if (camera.zoom < 4.0f)
+        camera.zoom = 4.0f;
+    }
+
+    if (IsMouseButtonDown(MOUSE_BUTTON_RIGHT)) {
+      Vector2 mousePos = GetMouseDelta();
+      camera.target.x -= mousePos.x / 5.0f;
+      camera.target.y -= mousePos.y / 5.0f;
     }
+
+    BeginDrawing();
+    ClearBackground(RAYWHITE);
+
+    BeginMode2D(camera);
+    SE_DrawEditor(&editor, &camera);
+    EndMode2D();
+
+    SE_DrawEditorToolkit(&editor);
+
+    EndDrawing();
   }
 
   return 0;