we compiled

client/src/modes/canvas/common.rs

@@ -4,6 +4,7 @@ use crate::tui::terminal::core::TerminalCore;
 use crate::state::pages::{form::FormState, auth::AuthState};
 use crate::state::state::AppState;
 use crate::services::grpc_client::GrpcClient;
+use crate::services::auth::AuthClient;
 use crate::tui::functions::common::{
     form::{save as form_save, revert},
     login::{save as login_save, cancel}
@@ -14,6 +15,7 @@ pub async fn handle_core_action(
     form_state: &mut FormState,
     auth_state: &mut AuthState,
     grpc_client: &mut GrpcClient,
+    auth_client: &mut AuthClient,
     terminal: &mut TerminalCore,
     app_state: &mut AppState,
     current_position: &mut u64,
@@ -22,7 +24,7 @@ pub async fn handle_core_action(
     match action {
         "save" => {
             if app_state.ui.show_login {
-                let message = login_save(auth_state, grpc_client).await?;
+                let message = login_save(auth_state, auth_client, app_state).await?;
                 Ok((false, message))
             } else {
                 let message = form_save(
@@ -41,7 +43,7 @@ pub async fn handle_core_action(
         },
         "save_and_quit" => {
             let message = if app_state.ui.show_login {
-                login_save(auth_state, grpc_client).await?
+                login_save(auth_state, auth_client, app_state).await?
             } else {
                 form_save(
                     form_state,
@@ -56,7 +58,7 @@ pub async fn handle_core_action(
         },
         "revert" => {
             if app_state.ui.show_login {
-                let message = cancel(auth_state).await;
+                let message = cancel(auth_state, app_state).await;
                 Ok((false, message))
             } else {
                 let message = revert(

client/src/modes/canvas/edit.rs

@@ -1,16 +1,19 @@
 // src/modes/canvas/edit.rs
 
 // TODO THIS is freaking bloated with functions it never uses REFACTOR 200 LOC can be gone
+
+use std::any::Any;
 use crossterm::event::{KeyEvent, KeyCode, KeyModifiers};
 use crate::config::binds::config::Config;
+use crate::state::canvas_state::CanvasState;
 use crate::state::pages::form::FormState;
 use crate::services::grpc_client::GrpcClient;
 use crate::tui::functions::common::form::{save, revert};
 
-pub async fn handle_edit_event_internal(
+pub async fn handle_edit_event_internal<S: CanvasState + Any>(
     key: KeyEvent,
     config: &Config,
-    form_state: &mut FormState,
+    state: &mut S,
     ideal_cursor_column: &mut usize,
     command_message: &mut String,
     is_saved: &mut bool,
@@ -19,16 +22,14 @@ pub async fn handle_edit_event_internal(
     grpc_client: &mut GrpcClient,
 ) -> Result<String, Box<dyn std::error::Error>> {
     if let Some("enter_command_mode") = config.get_action_for_key_in_mode(&config.keybindings.global, key.code, key.modifiers) {
-        // Ignore in edit mode and process as normal input
-        handle_edit_specific_input(key, form_state, ideal_cursor_column);
+        handle_edit_specific_input(key, state, ideal_cursor_column);
         return Ok(command_message.clone());
     }
 
-    // Check common actions first
     if let Some(action) = config.get_action_for_key_in_mode(&config.keybindings.common, key.code, key.modifiers) {
         return execute_common_action(
             action,
-            form_state,
+            state,
             grpc_client,
             is_saved,
             current_position,
@@ -39,302 +40,340 @@ pub async fn handle_edit_event_internal(
     if let Some(action) = config.get_edit_action_for_key(key.code, key.modifiers) {
         return execute_edit_action(
             action,
-            form_state,
+            state,
             ideal_cursor_column,
-            grpc_client,  // Changed from AppTerminal
+            grpc_client,
             is_saved,
             current_position,
             total_count,
         ).await;
     }
 
-    handle_edit_specific_input(
-        key,
-        form_state,
-        ideal_cursor_column,
-    );
-
+    handle_edit_specific_input(key, state, ideal_cursor_column);
     Ok(command_message.clone())
 }
 
-async fn execute_common_action(
+async fn execute_common_action<S: CanvasState + Any>(
     action: &str,
-    form_state: &mut FormState,
+    state: &mut S,
     grpc_client: &mut GrpcClient,
     is_saved: &mut bool,
     current_position: &mut u64,
     total_count: u64,
 ) -> Result<String, Box<dyn std::error::Error>> {
     match action {
-        "save" => {
-            save(
-                form_state,
-                grpc_client,
-                is_saved,
-                current_position,
-                total_count,
-            ).await
-        },
-        "revert" => {
-            revert(
-                form_state,
-                grpc_client,
-                current_position,
-                total_count,
-            ).await
-        },
+        "save" | "revert" if state.has_unsaved_changes() => {
+            if let Some(form_state) = (state as &mut dyn Any).downcast_mut::<FormState>() {
+                return match action {
+                    "save" => save(form_state, grpc_client, is_saved, current_position, total_count).await,
+                    "revert" => revert(form_state, grpc_client, current_position, total_count).await,
+                    _ => unreachable!(),
+                };
+            }
+            Ok("Action not available in this context".to_string())
+        }
         "move_up" | "move_down" => {
-            // Reuse edit mode's existing logic
             execute_edit_action(
                 action,
-                form_state,
-                &mut 0, // Dummy ideal_cursor_column (not used here)
+                state,
+                &mut 0,
                 grpc_client,
                 is_saved,
                 current_position,
                 total_count,
             ).await
-        },
+        }
         _ => Ok(format!("Common action not handled: {}", action)),
     }
 }
 
-fn handle_edit_specific_input(
+fn handle_edit_specific_input<S: CanvasState>( // No Any needed here
     key: KeyEvent,
-    form_state: &mut FormState,
+    state: &mut S,
     ideal_cursor_column: &mut usize,
 ) {
     match key.code {
         KeyCode::Char(c) => {
-            let cursor_pos = form_state.current_cursor_pos;
-            let field_value = form_state.get_current_input_mut();
+            let cursor_pos = state.current_cursor_pos();
+            let field_value = state.get_current_input_mut();
             let mut chars: Vec<char> = field_value.chars().collect();
             if cursor_pos <= chars.len() {
                 chars.insert(cursor_pos, c);
                 *field_value = chars.into_iter().collect();
-                form_state.current_cursor_pos = cursor_pos + 1;
-                *ideal_cursor_column = form_state.current_cursor_pos;
-                form_state.has_unsaved_changes = true;
+                // Use trait setters
+                state.set_current_cursor_pos(cursor_pos + 1);
+                state.set_has_unsaved_changes(true);
+                *ideal_cursor_column = state.current_cursor_pos(); // Update ideal column
             }
         }
         KeyCode::Backspace => {
-            if form_state.current_cursor_pos > 0 {
-                let cursor_pos = form_state.current_cursor_pos;
-                let field_value = form_state.get_current_input_mut();
+            if state.current_cursor_pos() > 0 {
+                let cursor_pos = state.current_cursor_pos();
+                let field_value = state.get_current_input_mut();
                 let mut chars: Vec<char> = field_value.chars().collect();
                 if cursor_pos <= chars.len() && cursor_pos > 0 {
                     chars.remove(cursor_pos - 1);
                     *field_value = chars.into_iter().collect();
-                    form_state.current_cursor_pos = cursor_pos - 1;
-                    *ideal_cursor_column = form_state.current_cursor_pos;
-                    form_state.has_unsaved_changes = true;
+                    // Use trait setters
+                    state.set_current_cursor_pos(cursor_pos - 1);
+                    state.set_has_unsaved_changes(true);
+                    *ideal_cursor_column = state.current_cursor_pos(); // Update ideal column
                 }
             }
         }
         KeyCode::Delete => {
-            let cursor_pos = form_state.current_cursor_pos;
-            let field_value = form_state.get_current_input_mut();
+            let cursor_pos = state.current_cursor_pos();
+            let field_value = state.get_current_input_mut();
             let chars: Vec<char> = field_value.chars().collect();
             if cursor_pos < chars.len() {
                 let mut new_chars = chars.clone();
                 new_chars.remove(cursor_pos);
                 *field_value = new_chars.into_iter().collect();
-                form_state.has_unsaved_changes = true;
+                // Use trait setter
+                state.set_has_unsaved_changes(true);
+                // Cursor position doesn't change, but ideal might if text changed
+                *ideal_cursor_column = state.current_cursor_pos();
             }
         }
         KeyCode::Tab => {
-            if key.modifiers.contains(KeyModifiers::SHIFT) {
-                if form_state.current_field == 0 {
-                    form_state.current_field = form_state.fields.len() - 1;
+            let num_fields = state.fields().len();
+            if num_fields > 0 { // Avoid panic on empty fields
+                let current_field = state.current_field();
+                let new_field = if key.modifiers.contains(KeyModifiers::SHIFT) {
+                    if current_field == 0 { num_fields - 1 } else { current_field - 1 }
                 } else {
-                    form_state.current_field = form_state.current_field.saturating_sub(1);
-                }
-            } else {
-                form_state.current_field = (form_state.current_field + 1) % form_state.fields.len();
+                    (current_field + 1) % num_fields
+                };
+                // Use trait setter
+                state.set_current_field(new_field);
+                let current_input = state.get_current_input();
+                let max_cursor_pos = current_input.len();
+                // Use trait setter
+                state.set_current_cursor_pos((*ideal_cursor_column).min(max_cursor_pos));
             }
-            let current_input = form_state.get_current_input();
-            let max_cursor_pos = current_input.len();
-            form_state.current_cursor_pos = (*ideal_cursor_column).min(max_cursor_pos);
         }
         KeyCode::Enter => {
-            form_state.current_field = (form_state.current_field + 1) % form_state.fields.len();
-            let current_input = form_state.get_current_input();
-            let max_cursor_pos = current_input.len();
-            form_state.current_cursor_pos = (*ideal_cursor_column).min(max_cursor_pos);
+             let num_fields = state.fields().len();
+            if num_fields > 0 { // Avoid panic on empty fields
+                let new_field = (state.current_field() + 1) % num_fields;
+                // Use trait setter
+                state.set_current_field(new_field);
+                let current_input = state.get_current_input();
+                let max_cursor_pos = current_input.len();
+                // Use trait setter
+                state.set_current_cursor_pos((*ideal_cursor_column).min(max_cursor_pos));
+            }
         }
         _ => {}
     }
 }
 
-async fn execute_edit_action(
+async fn execute_edit_action<S: CanvasState>( // No Any needed here
     action: &str,
-    form_state: &mut FormState,
+    state: &mut S,
     ideal_cursor_column: &mut usize,
-    grpc_client: &mut GrpcClient,  // Changed from AppTerminal
-    is_saved: &mut bool,
-    current_position: &mut u64,
-    total_count: u64,
+    // These parameters are unused if save/revert aren't handled here,
+    // but keep them for now if execute_common_action calls this
+    _grpc_client: &mut GrpcClient,
+    _is_saved: &mut bool,
+    _current_position: &mut u64,
+    _total_count: u64,
 ) -> Result<String, Box<dyn std::error::Error>> {
     match action {
-        "save" => {
-            save(
-                form_state,
-                grpc_client,  // Changed from AppTerminal
-                is_saved,
-                current_position,
-                total_count,
-            ).await
-        },
         "move_left" => {
-            form_state.current_cursor_pos = form_state.current_cursor_pos.saturating_sub(1);
-            *ideal_cursor_column = form_state.current_cursor_pos;
+            let new_pos = state.current_cursor_pos().saturating_sub(1);
+            // Use trait setter
+            state.set_current_cursor_pos(new_pos);
+            *ideal_cursor_column = new_pos;
             Ok("".to_string())
         }
         "move_right" => {
-            let current_input = form_state.get_current_input();
-            if form_state.current_cursor_pos < current_input.len() {
-                form_state.current_cursor_pos += 1;
-                *ideal_cursor_column = form_state.current_cursor_pos;
+            let current_input = state.get_current_input();
+            let current_pos = state.current_cursor_pos();
+            // Allow moving cursor to position *after* last character
+            if current_pos < current_input.len() {
+                let new_pos = current_pos + 1;
+                 // Use trait setter
+                state.set_current_cursor_pos(new_pos);
+                *ideal_cursor_column = new_pos;
             }
             Ok("".to_string())
         }
         "move_up" => {
-            if form_state.current_field == 0 {
-                form_state.current_field = form_state.fields.len() - 1;
-            } else {
-                form_state.current_field = form_state.current_field.saturating_sub(1);
+            let num_fields = state.fields().len();
+            if num_fields > 0 {
+                let current_field = state.current_field();
+                let new_field = if current_field == 0 { num_fields - 1 } else { current_field - 1 };
+                // Use trait setter
+                state.set_current_field(new_field);
+                let current_input = state.get_current_input();
+                let max_pos = current_input.len();
+                // Use trait setter
+                state.set_current_cursor_pos((*ideal_cursor_column).min(max_pos));
             }
-            let current_input = form_state.get_current_input();
-            let max_cursor_pos = current_input.len();
-            form_state.current_cursor_pos = (*ideal_cursor_column).min(max_cursor_pos);
             Ok("".to_string())
         }
         "move_down" => {
-            form_state.current_field = (form_state.current_field + 1) % form_state.fields.len();
-            let current_input = form_state.get_current_input();
-            let max_cursor_pos = current_input.len();
-            form_state.current_cursor_pos = (*ideal_cursor_column).min(max_cursor_pos);
+            let num_fields = state.fields().len();
+             if num_fields > 0 {
+                let new_field = (state.current_field() + 1) % num_fields;
+                // Use trait setter
+                state.set_current_field(new_field);
+                let current_input = state.get_current_input();
+                let max_pos = current_input.len();
+                // Use trait setter
+                state.set_current_cursor_pos((*ideal_cursor_column).min(max_pos));
+            }
             Ok("".to_string())
         }
         "move_line_start" => {
-            form_state.current_cursor_pos = 0;
-            *ideal_cursor_column = form_state.current_cursor_pos;
+            // Use trait setter
+            state.set_current_cursor_pos(0);
+            *ideal_cursor_column = 0;
             Ok("".to_string())
         }
         "move_line_end" => {
-            let current_input = form_state.get_current_input();
-            form_state.current_cursor_pos = current_input.len();
-            *ideal_cursor_column = form_state.current_cursor_pos;
+            let current_input = state.get_current_input();
+            let new_pos = current_input.len();
+            // Use trait setter
+            state.set_current_cursor_pos(new_pos);
+            *ideal_cursor_column = new_pos;
             Ok("".to_string())
         }
         "move_first_line" => {
-            form_state.current_field = 0;
-            let current_input = form_state.get_current_input();
-            let max_cursor_pos = current_input.len();
-            form_state.current_cursor_pos = (*ideal_cursor_column).min(max_cursor_pos);
+             let num_fields = state.fields().len();
+             if num_fields > 0 {
+                // Use trait setter
+                state.set_current_field(0);
+                let current_input = state.get_current_input();
+                let max_pos = current_input.len();
+                // Use trait setter
+                state.set_current_cursor_pos((*ideal_cursor_column).min(max_pos));
+            }
             Ok("Moved to first line".to_string())
         }
         "move_last_line" => {
-            form_state.current_field = form_state.fields.len() - 1;
-            let current_input = form_state.get_current_input();
-            let max_cursor_pos = current_input.len();
-            form_state.current_cursor_pos = (*ideal_cursor_column).min(max_cursor_pos);
+            let num_fields = state.fields().len();
+             if num_fields > 0 {
+                let new_field = num_fields - 1;
+                // Use trait setter
+                state.set_current_field(new_field);
+                let current_input = state.get_current_input();
+                let max_pos = current_input.len();
+                // Use trait setter
+                state.set_current_cursor_pos((*ideal_cursor_column).min(max_pos));
+            }
             Ok("Moved to last line".to_string())
         }
-        // Word movement actions
         "move_word_next" => {
-            let current_input = form_state.get_current_input();
+            let current_input = state.get_current_input();
             if !current_input.is_empty() {
-                let new_pos = find_next_word_start(current_input, form_state.current_cursor_pos);
-                form_state.current_cursor_pos = new_pos.min(current_input.len());
-                *ideal_cursor_column = form_state.current_cursor_pos;
+                let new_pos = find_next_word_start(current_input, state.current_cursor_pos());
+                let final_pos = new_pos.min(current_input.len());
+                // Use trait setter
+                state.set_current_cursor_pos(final_pos);
+                *ideal_cursor_column = final_pos;
             }
             Ok("".to_string())
         }
         "move_word_end" => {
-            let current_input = form_state.get_current_input();
+            let current_input = state.get_current_input();
             if !current_input.is_empty() {
-                let new_pos = find_word_end(current_input, form_state.current_cursor_pos);
-                form_state.current_cursor_pos = new_pos.min(current_input.len());
-                *ideal_cursor_column = form_state.current_cursor_pos;
+                let new_pos = find_word_end(current_input, state.current_cursor_pos());
+                let final_pos = new_pos.min(current_input.len());
+                 // Use trait setter
+                state.set_current_cursor_pos(final_pos);
+                *ideal_cursor_column = final_pos;
             }
             Ok("".to_string())
         }
         "move_word_prev" => {
-            let current_input = form_state.get_current_input();
+            let current_input = state.get_current_input();
             if !current_input.is_empty() {
-                let new_pos = find_prev_word_start(current_input, form_state.current_cursor_pos);
-                form_state.current_cursor_pos = new_pos;
-                *ideal_cursor_column = form_state.current_cursor_pos;
+                let new_pos = find_prev_word_start(current_input, state.current_cursor_pos());
+                // Use trait setter
+                state.set_current_cursor_pos(new_pos);
+                *ideal_cursor_column = new_pos;
             }
             Ok("".to_string())
         }
         "move_word_end_prev" => {
-            let current_input = form_state.get_current_input();
+            let current_input = state.get_current_input();
             if !current_input.is_empty() {
-                let new_pos = find_prev_word_end(current_input, form_state.current_cursor_pos);
-                form_state.current_cursor_pos = new_pos;
-                *ideal_cursor_column = form_state.current_cursor_pos;
+                let new_pos = find_prev_word_end(current_input, state.current_cursor_pos());
+                // Use trait setter
+                state.set_current_cursor_pos(new_pos);
+                *ideal_cursor_column = new_pos;
             }
-            Ok("".to_string())
+            Ok("Moved to previous word end".to_string())
         }
-        // Edit-specific actions that can be bound to keys
         "delete_char_forward" => {
-            let cursor_pos = form_state.current_cursor_pos;
-            let field_value = form_state.get_current_input_mut();
-            let chars: Vec<char> = field_value.chars().collect();
+            let cursor_pos = state.current_cursor_pos();
+            let field_value = state.get_current_input_mut();
+            let mut chars: Vec<char> = field_value.chars().collect(); // Use mut for modification
             if cursor_pos < chars.len() {
-                let mut new_chars = chars.clone();
-                new_chars.remove(cursor_pos);
-                *field_value = new_chars.into_iter().collect();
-                form_state.has_unsaved_changes = true;
+                chars.remove(cursor_pos);
+                *field_value = chars.into_iter().collect();
+                // Use trait setter
+                state.set_has_unsaved_changes(true);
+                // Cursor position doesn't change
+                *ideal_cursor_column = cursor_pos;
             }
             Ok("".to_string())
         }
         "delete_char_backward" => {
-            if form_state.current_cursor_pos > 0 {
-                let cursor_pos = form_state.current_cursor_pos;
-                let field_value = form_state.get_current_input_mut();
-                let mut chars: Vec<char> = field_value.chars().collect();
+            if state.current_cursor_pos() > 0 {
+                let cursor_pos = state.current_cursor_pos();
+                let field_value = state.get_current_input_mut();
+                let mut chars: Vec<char> = field_value.chars().collect(); // Use mut
                 if cursor_pos <= chars.len() && cursor_pos > 0 {
                     chars.remove(cursor_pos - 1);
                     *field_value = chars.into_iter().collect();
-                    form_state.current_cursor_pos = cursor_pos - 1;
-                    *ideal_cursor_column = form_state.current_cursor_pos;
-                    form_state.has_unsaved_changes = true;
+                    let new_pos = cursor_pos - 1;
+                    // Use trait setters
+                    state.set_current_cursor_pos(new_pos);
+                    state.set_has_unsaved_changes(true);
+                    *ideal_cursor_column = new_pos;
                 }
             }
             Ok("".to_string())
         }
         "insert_char" => {
-            // This could be expanded to allow configurable character insertion
-            // For now, it's a placeholder that would need additional parameters
-            Ok("Character insertion requires configuration".to_string())
+            // This action doesn't make sense here as char insertion is handled
+            // directly in handle_edit_specific_input
+            Ok("Character insertion handled directly".to_string())
         }
         "next_field" => {
-            form_state.current_field = (form_state.current_field + 1) % form_state.fields.len();
-            let current_input = form_state.get_current_input();
-            let max_cursor_pos = current_input.len();
-            form_state.current_cursor_pos = (*ideal_cursor_column).min(max_cursor_pos);
+            let num_fields = state.fields().len();
+             if num_fields > 0 {
+                let new_field = (state.current_field() + 1) % num_fields;
+                // Use trait setter
+                state.set_current_field(new_field);
+                let current_input = state.get_current_input();
+                let max_pos = current_input.len();
+                // Use trait setter
+                state.set_current_cursor_pos((*ideal_cursor_column).min(max_pos));
+            }
             Ok("".to_string())
         }
         "prev_field" => {
-            if form_state.current_field == 0 {
-                form_state.current_field = form_state.fields.len() - 1;
-            } else {
-                form_state.current_field = form_state.current_field.saturating_sub(1);
+            let num_fields = state.fields().len();
+             if num_fields > 0 {
+                let current_field = state.current_field();
+                let new_field = if current_field == 0 { num_fields - 1 } else { current_field - 1 };
+                // Use trait setter
+                state.set_current_field(new_field);
+                let current_input = state.get_current_input();
+                let max_pos = current_input.len();
+                // Use trait setter
+                state.set_current_cursor_pos((*ideal_cursor_column).min(max_pos));
             }
-            let current_input = form_state.get_current_input();
-            let max_cursor_pos = current_input.len();
-            form_state.current_cursor_pos = (*ideal_cursor_column).min(max_cursor_pos);
             Ok("".to_string())
         }
-        // Fallback for unrecognized actions
-        _ => Ok(format!("Unknown action: {}", action)),
+        _ => Ok(format!("Unknown edit action: {}", action)),
     }
 }
 
-// Reuse these character and word navigation helper functions
 #[derive(PartialEq)]
 enum CharType {
     Whitespace,

client/src/modes/handlers/event.rs

@@ -5,6 +5,7 @@ use crate::tui::terminal::{
     core::TerminalCore,
 };
 use crate::services::grpc_client::GrpcClient;
+use crate::services::auth::AuthClient;
 use crate::tui::functions::common::commands::CommandHandler;
 use crate::config::binds::config::Config;
 use crate::state::pages::form::FormState;
@@ -27,11 +28,12 @@ pub struct EventHandler {
     pub ideal_cursor_column: usize,
     pub key_sequence_tracker: KeySequenceTracker,
     pub auth_state: AuthState,
+    pub auth_client: AuthClient,
 }
 
 impl EventHandler {
-    pub fn new() -> Self {
-        EventHandler {
+    pub async fn new() -> Result<Self, Box<dyn std::error::Error>> {
+        Ok(EventHandler {
             command_mode: false,
             command_input: String::new(),
             command_message: String::new(),
@@ -40,7 +42,8 @@ impl EventHandler {
             ideal_cursor_column: 0,
             key_sequence_tracker: KeySequenceTracker::new(800),
             auth_state: AuthState::new(),
-        }
+            auth_client: AuthClient::new().await?,
+        })
     }
 
     pub async fn handle_event(
@@ -133,6 +136,7 @@ impl EventHandler {
                                     form_state,
                                     &mut self.auth_state,
                                     grpc_client,
+                                    &mut self.auth_client,
                                     terminal,
                                     app_state,
                                     current_position,
@@ -194,6 +198,7 @@ impl EventHandler {
                                     form_state,
                                     &mut self.auth_state,
                                     grpc_client,
+                                    &mut self.auth_client,
                                     terminal,
                                     app_state,
                                     current_position,
@@ -205,17 +210,31 @@ impl EventHandler {
                     }
 
                     // Let edit mode handle its own actions (including navigation from common bindings)
-                    let result = edit::handle_edit_event_internal(
-                        key,
-                        config,
-                        form_state,
-                        &mut self.ideal_cursor_column,
-                        &mut self.command_message,
-                        &mut app_state.ui.is_saved,
-                        current_position,
-                        total_count,
-                        grpc_client,
-                    ).await?;
+                    let result = if app_state.ui.show_login {
+                        edit::handle_edit_event_internal(
+                            key,
+                            config,
+                            &mut self.auth_state, // Use auth_state instead of form_state
+                            &mut self.ideal_cursor_column,
+                            &mut self.command_message,
+                            &mut app_state.ui.is_saved,
+                            current_position,
+                            total_count,
+                            grpc_client,
+                        ).await?
+                    } else {
+                        edit::handle_edit_event_internal(
+                            key,
+                            config,
+                            form_state,
+                            &mut self.ideal_cursor_column,
+                            &mut self.command_message,
+                            &mut app_state.ui.is_saved,
+                            current_position,
+                            total_count,
+                            grpc_client,
+                        ).await?
+                    };
 
                     self.key_sequence_tracker.reset();
                     return Ok((false, result));

client/src/state/canvas_state.rs

@@ -10,40 +10,8 @@ pub trait CanvasState {
     fn get_current_input(&self) -> &str;
     fn get_current_input_mut(&mut self) -> &mut String;
     fn fields(&self) -> Vec<&str>;
-}
-
-// Implement for FormState (keep existing form.rs code and add this)
-impl CanvasState for FormState {
-    fn current_field(&self) -> usize {
-        self.current_field
-    }
-
-    fn current_cursor_pos(&self) -> usize {
-        self.current_cursor_pos
-    }
-
-    fn has_unsaved_changes(&self) -> bool {
-        self.has_unsaved_changes
-    }
-
-    fn inputs(&self) -> Vec<&String> {
-        self.values.iter().collect()
-    }
-
-    fn get_current_input(&self) -> &str {
-        self.values
-            .get(self.current_field)
-            .map(|s| s.as_str())
-            .unwrap_or("")
-    }
-
-    fn get_current_input_mut(&mut self) -> &mut String {
-        self.values
-            .get_mut(self.current_field)
-            .expect("Invalid current_field index")
-    }
-
-    fn fields(&self) -> Vec<&str> {
-        self.fields.iter().map(|s| s.as_str()).collect()
-    }
+
+    fn set_current_field(&mut self, index: usize);
+    fn set_current_cursor_pos(&mut self, pos: usize);
+    fn set_has_unsaved_changes(&mut self, changed: bool);
 }

client/src/state/pages/auth.rs

@@ -67,4 +67,21 @@ impl CanvasState for AuthState {
     fn fields(&self) -> Vec<&str> {
         vec!["Username/Email", "Password"]
     }
+
+    // --- Implement the setter methods ---
+    fn set_current_field(&mut self, index: usize) {
+        // AuthState only has 2 fields (0 and 1)
+        if index < 2 {
+            self.current_field = index;
+        }
+    }
+
+    fn set_current_cursor_pos(&mut self, pos: usize) {
+        // Optional: Add validation based on current input length if needed
+        self.current_cursor_pos = pos;
+    }
+
+    fn set_has_unsaved_changes(&mut self, _changed: bool) {
+        // Auth form doesn't track unsaved changes, so do nothing
+    }
 }

client/src/state/pages/form.rs

@@ -2,6 +2,7 @@
 use crate::config::colors::themes::Theme;
 use ratatui::layout::Rect;
 use ratatui::Frame;
+use crate::state::canvas_state::CanvasState;
 
 pub struct FormState {
     pub id: i64,
@@ -82,3 +83,54 @@ impl FormState {
         ];
     }
 }
+
+impl CanvasState for FormState {
+    fn current_field(&self) -> usize {
+        self.current_field
+    }
+
+    fn current_cursor_pos(&self) -> usize {
+        self.current_cursor_pos
+    }
+
+    fn has_unsaved_changes(&self) -> bool {
+        self.has_unsaved_changes
+    }
+
+    fn inputs(&self) -> Vec<&String> {
+        self.values.iter().collect()
+    }
+
+    fn get_current_input(&self) -> &str {
+        self.values
+            .get(self.current_field)
+            .map(|s| s.as_str())
+            .unwrap_or("")
+    }
+
+    fn get_current_input_mut(&mut self) -> &mut String {
+        self.values
+            .get_mut(self.current_field)
+            .expect("Invalid current_field index")
+    }
+
+    fn fields(&self) -> Vec<&str> {
+        self.fields.iter().map(|s| s.as_str()).collect()
+    }
+
+    // --- Implement the setter methods ---
+    fn set_current_field(&mut self, index: usize) {
+        if index < self.fields.len() { // Basic bounds check
+            self.current_field = index;
+        }
+    }
+
+    fn set_current_cursor_pos(&mut self, pos: usize) {
+        // Optional: Add validation based on current input length if needed
+        self.current_cursor_pos = pos;
+    }
+
+    fn set_has_unsaved_changes(&mut self, changed: bool) {
+        self.has_unsaved_changes = changed;
+    }
+}

client/src/tui/functions/common/login.rs

@@ -1,18 +1,45 @@
 // src/tui/functions/common/login.rs
-
+use crate::services::auth::AuthClient;
 use crate::state::pages::auth::AuthState;
-use crate::services::grpc_client::GrpcClient;
+use crate::state::state::AppState;
 
 pub async fn save(
     auth_state: &mut AuthState,
-    grpc_client: &mut GrpcClient,
+    auth_client: &mut AuthClient,
+    app_state: &mut AppState,
 ) -> Result<String, Box<dyn std::error::Error>> {
-    // Implement your login-specific save logic here
-    Ok("Login credentials saved - not implemented yet".to_string())
+    let identifier = auth_state.username.clone();
+    let password = auth_state.password.clone();
+
+    match auth_client.login(identifier, password).await {
+        Ok(response) => {
+            auth_state.auth_token = Some(response.access_token);
+            auth_state.user_id = Some(response.user_id);
+            auth_state.role = Some(response.role);
+            auth_state.error_message = None;
+            
+            // Update app state to show main interface
+            app_state.ui.show_login = false;
+            app_state.ui.show_form = true;
+            
+            Ok("Login successful!".to_string())
+        }
+        Err(e) => {
+            let error_message = format!("Login failed: {}", e);
+            auth_state.error_message = Some(error_message.clone());
+            Ok(error_message)
+        }
+    }
 }
 
 pub async fn cancel(
     auth_state: &mut AuthState,
+    app_state: &mut AppState,
 ) -> String {
-    "Login credentials canceled - not implemented yet".to_string()
+    auth_state.username.clear();
+    auth_state.password.clear();
+    auth_state.error_message = None;
+    app_state.ui.show_login = false;
+    app_state.ui.show_intro = true;
+    "Login canceled".to_string()
 }

client/src/ui/handlers/ui.rs

@@ -2,6 +2,7 @@
 
 use crate::tui::terminal::TerminalCore;
 use crate::services::grpc_client::GrpcClient;
+use crate::services::auth::AuthClient;
 use crate::services::ui_service::UiService;  // Add this import
 use crate::tui::terminal::EventReader;
 use crate::tui::functions::common::CommandHandler;
@@ -17,6 +18,7 @@ pub async fn run_ui() -> Result<(), Box<dyn std::error::Error>> {
     let config = Config::load()?;
     let mut terminal = TerminalCore::new()?;
     let mut grpc_client = GrpcClient::new().await?;
+    let auth_client = AuthClient::new().await?;
     let mut command_handler = CommandHandler::new();
     let theme = Theme::from_str(&config.colors.theme);
     let mut auth_state = AuthState::default();
@@ -31,7 +33,7 @@ pub async fn run_ui() -> Result<(), Box<dyn std::error::Error>> {
     let mut form_state = FormState::new(column_names);
 
     // The rest of your UI initialization remains the same
-    let mut event_handler = EventHandler::new();
+    let mut event_handler = EventHandler::new().await?;
     let event_reader = EventReader::new();
 
     // Fetch the total count of Adresar entries