zippa-db 0.1.0

//! The connection manager: empty state, cards, and the editor's round trips.

use super::*;

use gpui_kit::component::{Theme, ThemeRegistry};

#[gpui_kit::test]
fn the_empty_state_offers_new_connection_and_open_file(cx: &mut TestAppContext) {
    let handle = workspace(cx);
    let welcome = handle
        .update(cx, |workspace, _, _| workspace.active_welcome_for_test())
        .unwrap()
        .expect("the active tab should be showing the connection manager");
    welcome.update(cx, |welcome, cx| {
        welcome.set_connections_for_test(vec![], cx)
    });

    cx.update_window(handle.window.into(), |_, window, cx| {
        window.draw(cx).clear(cx);
        assert!(window.find("empty-new-connection").visible());
        assert!(window.find("open-sqlite-file").visible());
    })
    .unwrap();
}

#[gpui_kit::test]
fn saving_a_connection_persists_its_colour(cx: &mut TestAppContext) {
    let dir = ScratchDir::new();
    crate::db::store::set_config_dir_for_test(dir.path.clone());

    let handle = workspace(cx);
    let welcome = handle
        .update(cx, |workspace, _, _| workspace.active_welcome_for_test())
        .unwrap()
        .expect("the active tab should be showing the connection manager");

    let config = ConnectionConfig {
        name: "Prod DB".into(),
        color: Some(TagColor::Red),
        ..ConnectionConfig::new(Engine::Postgres)
    };
    let id = config.id;

    handle
        .update(cx, |_, window, cx| {
            welcome.update(cx, |welcome, cx| {
                welcome.save_for_test(config.clone(), None, window, cx)
            })
        })
        .unwrap();

    // The write goes to the background executor; let it land before reading it
    // back from the scratch dir.
    cx.run_until_parked();

    // The launcher holds it, and the store wrote it to the scratch dir.
    let color = welcome.update(cx, |welcome, _| {
        let saved = welcome
            .connections_for_test()
            .iter()
            .find(|c| c.id == id)
            .cloned()
            .expect("the connection should be in the launcher");
        saved.color
    });
    assert_eq!(color, Some(TagColor::Red));

    let loaded = crate::db::store::load().expect("the connection should be on disk");
    assert!(
        loaded
            .iter()
            .any(|c| c.id == id && c.color == Some(TagColor::Red))
    );
}

/// The footgun IDEAS.md's safety nudge exists to catch: a card for a
/// production-marked, auto-applying connection asks before connecting rather
/// than opening straight away, the way an ordinary card does.
#[gpui_kit::test]
fn connecting_to_a_production_auto_apply_connection_asks_first(cx: &mut TestAppContext) {
    let handle = workspace(cx);
    let database = runtime::block_on(TempDatabase::new());
    let config = ConnectionConfig {
        color: Some(TagColor::Red),
        safety: SafetyMode::AutoApply,
        ..database.config()
    };
    let id = config.id;

    let welcome = handle
        .update(cx, |workspace, _, _| workspace.active_welcome_for_test())
        .unwrap()
        .expect("the active tab should be showing the connection manager");
    welcome.update(cx, |welcome, cx| {
        welcome.set_connections_for_test(vec![config], cx)
    });

    click_workspace(
        cx,
        &handle,
        gpui_kit::SharedString::from(format!("connect-{id}")),
    );

    assert!(
        workspace_dialog_open(cx, &handle),
        "a production connection set to auto-apply should ask before connecting"
    );
    assert!(
        handle
            .update(cx, |workspace, _, _| workspace.active_session_for_test())
            .unwrap()
            .is_none(),
        "connecting should wait for confirmation"
    );

    click_workspace(cx, &handle, "ok");

    assert!(
        handle
            .update(cx, |workspace, _, _| workspace.active_session_for_test())
            .unwrap()
            .is_some(),
        "confirming should connect"
    );
}

#[gpui_kit::test]
fn duplicating_a_connection_uses_a_new_id(cx: &mut TestAppContext) {
    let dir = ScratchDir::new();
    crate::db::store::set_config_dir_for_test(dir.path.clone());

    let handle = workspace(cx);
    let welcome = handle
        .update(cx, |workspace, _, _| workspace.active_welcome_for_test())
        .unwrap()
        .expect("the active tab should be showing the connection manager");

    let config = ConnectionConfig {
        name: "Original".into(),
        ..ConnectionConfig::new(Engine::Postgres)
    };
    let id = config.id;
    welcome.update(cx, |welcome, cx| {
        welcome.set_connections_for_test(vec![config], cx)
    });

    handle
        .update(cx, |_, window, cx| {
            welcome.update(cx, |welcome, cx| welcome.duplicate_for_test(id, window, cx))
        })
        .unwrap();

    let ids = welcome.update(cx, |welcome, _| {
        welcome
            .connections_for_test()
            .iter()
            .map(|c| c.id)
            .collect::<Vec<_>>()
    });
    assert_eq!(ids.len(), 2, "duplicating should add a second connection");
    assert!(ids.contains(&id), "the original should stay");
    assert_eq!(
        ids.iter().filter(|other| **other == id).count(),
        1,
        "the duplicate should have its own id"
    );
}

#[gpui_kit::test]
fn deleting_a_connection_removes_it_and_persists(cx: &mut TestAppContext) {
    let dir = ScratchDir::new();
    crate::db::store::set_config_dir_for_test(dir.path.clone());

    let handle = workspace(cx);
    let welcome = handle
        .update(cx, |workspace, _, _| workspace.active_welcome_for_test())
        .unwrap()
        .expect("the active tab should be showing the connection manager");

    let config = ConnectionConfig {
        name: "Doomed".into(),
        ..ConnectionConfig::new(Engine::Postgres)
    };
    let id = config.id;
    welcome.update(cx, |welcome, cx| {
        welcome.set_connections_for_test(vec![config], cx)
    });

    // The card's `…` menu asks first via `open_alert_dialog` (covered by the
    // workspace close-confirmation test); this is what its Delete button runs.
    handle
        .update(cx, |_, window, cx| {
            welcome.update(cx, |welcome, cx| {
                welcome.delete_confirmed_for_test(id, window, cx)
            })
        })
        .unwrap();

    assert_eq!(
        welcome.update(cx, |welcome, _| welcome.connections_for_test().len()),
        0,
        "deleting should remove the connection"
    );
    cx.run_until_parked();
    assert!(
        crate::db::store::load()
            .expect("the store should be readable")
            .iter()
            .all(|c| c.id != id),
        "the deleted connection should not be on disk"
    );
}

#[gpui_kit::test]
fn editing_a_connection_leaves_its_stored_password_alone(cx: &mut TestAppContext) {
    let dir = ScratchDir::new();
    crate::db::store::set_config_dir_for_test(dir.path.clone());

    let handle = workspace(cx);
    let welcome = handle
        .update(cx, |workspace, _, _| workspace.active_welcome_for_test())
        .unwrap()
        .expect("the active tab should be showing the connection manager");

    let config = ConnectionConfig {
        name: "Prod DB".into(),
        ..ConnectionConfig::new(Engine::Postgres)
    };
    let id = config.id;
    welcome.update(cx, |welcome, cx| {
        welcome.set_connections_for_test(vec![config], cx)
    });

    // Open the editor on the saved connection, the way its card's `…` menu
    // does. `update_window` rather than `handle.update`: opening a dialog
    // updates the window's `Root`, which the latter already holds.
    cx.update_window(handle.window.into(), |_, window, cx| {
        welcome.update(cx, |welcome, cx| welcome.edit_for_test(id, window, cx));
    })
    .unwrap();
    assert!(
        welcome.update(cx, |welcome, _| welcome.editor_open_for_test()),
        "the editor should be open on the saved connection"
    );

    // The stored password is never loaded into the form, so an edit that was
    // not about the password has to say "leave it alone" rather than "write an
    // empty one", which would delete it.
    assert_eq!(
        welcome.update(cx, |welcome, cx| welcome
            .editor_password_intent_for_test(cx)),
        None,
        "an untouched password box must not claim a password to write"
    );

    // Typing one still reports it, so the box keeps working.
    draw_workspace(cx, &handle);
    welcome
        .downgrade()
        .update_in(cx, |welcome, window, cx| {
            welcome.focus_editor_password_for_test(window, cx)
        })
        .unwrap();
    cx.update_window(handle.window.into(), |_, window, cx| {
        window.input("hunter2", cx);
    })
    .unwrap();

    assert_eq!(
        welcome.update(cx, |welcome, cx| welcome
            .editor_password_intent_for_test(cx)),
        Some("hunter2".to_string()),
        "typing in the box reports the new password"
    );
}

#[gpui_kit::test]
fn editing_a_connection_keeps_its_last_connected_stamp(cx: &mut TestAppContext) {
    let dir = ScratchDir::new();
    crate::db::store::set_config_dir_for_test(dir.path.clone());

    let handle = workspace(cx);
    let welcome = handle
        .update(cx, |workspace, _, _| workspace.active_welcome_for_test())
        .unwrap()
        .expect("the active tab should be showing the connection manager");

    let when = "2026-01-02T03:04:05Z"
        .parse::<chrono::DateTime<chrono::Utc>>()
        .expect("a timestamp");
    let config = ConnectionConfig {
        name: "Prod DB".into(),
        last_connected: Some(when),
        ..ConnectionConfig::new(Engine::Postgres)
    };
    let id = config.id;
    welcome.update(cx, |welcome, cx| {
        welcome.set_connections_for_test(vec![config], cx)
    });

    // The editor builds its config fresh from the form, so it carries no stamp;
    // saving must not let that wipe the stored one and drop the card out of
    // most-recent-first order.
    let mut edited = ConnectionConfig::new(Engine::Postgres);
    edited.id = id;
    edited.name = "Prod DB (edited)".into();
    assert!(edited.last_connected.is_none());
    cx.update_window(handle.window.into(), |_, window, cx| {
        welcome.update(cx, |welcome, cx| {
            welcome.save_for_test(edited, None, window, cx)
        });
    })
    .unwrap();

    let saved = welcome.read_with(cx, |welcome, _| welcome.connections_for_test().to_vec());
    let saved = saved
        .iter()
        .find(|saved| saved.id == id)
        .expect("the edited connection should be saved");
    assert_eq!(saved.name, "Prod DB (edited)");
    assert_eq!(
        saved.last_connected,
        Some(when),
        "editing a connection is not connecting to it, so the stamp survives"
    );
}

#[gpui_kit::test]
fn tag_colors_contrast_in_every_bundled_theme(cx: &mut TestAppContext) {
    cx.update(|cx| {
        init_ui(cx);
        crate::settings::load_builtin_themes(cx);
    });

    cx.update(|cx| {
        let themes = ThemeRegistry::global(cx)
            .themes()
            .values()
            .cloned()
            .collect::<Vec<_>>();

        for config in themes {
            let mode = config.mode;
            {
                let theme = Theme::global_mut(cx);
                theme.mode = mode;
                theme.apply_config(&config);
            }
            let theme = Theme::global(cx);

            for color in TagColor::ALL {
                let background = color.hsla_on(theme);
                let foreground = color.on_color_for(theme);
                let ratio = crate::ui::contrast(background, foreground);
                assert!(
                    ratio >= 4.5,
                    "{} {} text ({:.2}:1) falls below 4.5:1",
                    theme.theme_name(),
                    color.label(),
                    ratio
                );
            }
        }
    });
}

#[gpui_kit::test]
fn engine_marks_read_in_every_bundled_theme(cx: &mut TestAppContext) {
    cx.update(|cx| {
        init_ui(cx);
        crate::settings::load_builtin_themes(cx);
    });

    cx.update(|cx| {
        let themes = ThemeRegistry::global(cx)
            .themes()
            .values()
            .cloned()
            .collect::<Vec<_>>();

        for config in themes {
            let mode = config.mode;
            {
                let theme = Theme::global_mut(cx);
                theme.mode = mode;
                theme.apply_config(&config);
            }
            let theme = Theme::global(cx);

            // The official colours are picked for light surfaces; a dark theme
            // is allowed to move one in lightness, but never below the floor
            // for a mark that is only a shape.
            for engine in Engine::ALL {
                let mark = crate::ui::engine_color(engine, cx);
                let ratio = crate::ui::contrast(mark, theme.background);
                assert!(
                    ratio >= 3.0,
                    "{}'s mark on {} reads at {:.2}:1, below 3:1",
                    engine.label(),
                    theme.theme_name(),
                    ratio
                );
            }
        }
    });
}

#[gpui_kit::test]
fn the_editor_suggests_values_for_the_chosen_engine(cx: &mut TestAppContext) {
    let handle = workspace(cx);
    let welcome = new_connection_editor(cx, &handle);
    let placeholders = |cx: &mut TestAppContext| {
        welcome.update(cx, |welcome, cx| {
            welcome.with_editor_for_test(cx, |editor, cx| editor.placeholders_for_test(cx))
        })
    };

    assert_eq!(
        placeholders(cx),
        ["Local PostgreSQL", "postgres", "postgres"].map(String::from)
    );

    fill_editor(cx, &handle, &welcome, Engine::MySql, "");
    assert_eq!(
        placeholders(cx),
        ["Local MySQL", "root", "Optional"].map(String::from),
        "switching engine should swap the suggestions for MySQL's"
    );

    fill_editor(cx, &handle, &welcome, Engine::Sqlite, "");
    let [name, _, database] = placeholders(cx);
    assert_eq!(name, "Local SQLite");
    assert!(
        database.ends_with("database.db"),
        "SQLite's database is a file, so it should suggest a path, not {database:?}"
    );
}

#[gpui_kit::test]
fn testing_a_connection_reports_in_the_dialog(cx: &mut TestAppContext) {
    let config_dir = ScratchDir::new();
    crate::db::store::set_config_dir_for_test(config_dir.path.clone());

    let handle = workspace(cx);
    let welcome = new_connection_editor(cx, &handle);

    // Nothing to connect to yet: the inputs are refused before any attempt.
    fill_editor(cx, &handle, &welcome, Engine::Sqlite, "");
    welcome.update(cx, |welcome, cx| {
        welcome.with_editor_for_test(cx, |editor, cx| editor.test_connection_for_test(cx))
    });
    cx.run_until_parked();
    assert_eq!(
        editor_status(cx, &welcome),
        "Error: Enter the path to the database file."
    );

    // A file that is not there is the driver's error, shown in the dialog.
    let dir = ScratchDir::new();
    let missing = dir.path.join("missing.db");
    fill_editor(
        cx,
        &handle,
        &welcome,
        Engine::Sqlite,
        &missing.to_string_lossy(),
    );
    cx.update_window(handle.window.into(), |_, window, cx| {
        welcome.update(cx, |welcome, cx| {
            welcome.with_editor_for_test(cx, |editor, cx| editor.focus_name_for_test(window, cx))
        });
        window.input("Scratch", cx);
    })
    .unwrap();
    assert_eq!(
        editor_status(cx, &welcome),
        "",
        "typing should clear the result that was about the old inputs"
    );
    welcome.update(cx, |welcome, cx| {
        welcome.with_editor_for_test(cx, |editor, cx| editor.test_connection_for_test(cx))
    });
    cx.run_until_parked();
    let status = editor_status(cx, &welcome);
    assert!(
        status.starts_with("Error: could not open"),
        "a failed test should say why, not {status:?}"
    );
    assert!(
        !missing.exists(),
        "testing must not create the database file"
    );

    // A real one succeeds, and either way the dialog stays open.
    let database = runtime::block_on(TempDatabase::new());
    fill_editor(
        cx,
        &handle,
        &welcome,
        Engine::Sqlite,
        &database.config().database,
    );
    welcome.update(cx, |welcome, cx| {
        welcome.with_editor_for_test(cx, |editor, cx| editor.test_connection_for_test(cx))
    });
    cx.run_until_parked();
    assert_eq!(editor_status(cx, &welcome), "succeeded");
    assert!(welcome.update(cx, |welcome, _| welcome.editor_open_for_test()));
    assert!(
        welcome.update(cx, |welcome, _| welcome.connections_for_test().is_empty()),
        "testing a connection should not save it"
    );
}

#[gpui_kit::test]
fn a_port_that_is_not_a_number_is_refused(cx: &mut TestAppContext) {
    let handle = workspace(cx);
    let welcome = new_connection_editor(cx, &handle);

    cx.update_window(handle.window.into(), |_, window, cx| {
        welcome.update(cx, |welcome, cx| {
            welcome.with_editor_for_test(cx, |editor, cx| {
                editor.set_port_for_test("54x2", window, cx)
            })
        });
    })
    .unwrap();
    welcome.update(cx, |welcome, cx| {
        welcome.with_editor_for_test(cx, |editor, cx| editor.connect_for_test(true, cx))
    });
    cx.run_until_parked();

    assert_eq!(
        editor_status(cx, &welcome),
        "Error: The port must be a number from 1 to 65535."
    );
    assert!(
        welcome.update(cx, |welcome, _| welcome.editor_open_for_test()),
        "a refused connect should leave the dialog open to fix the port"
    );
}

#[gpui_kit::test]
fn connect_saves_a_new_connection(cx: &mut TestAppContext) {
    let dir = ScratchDir::new();
    crate::db::store::set_config_dir_for_test(dir.path.clone());

    let handle = workspace(cx);
    let welcome = new_connection_editor(cx, &handle);
    let database = runtime::block_on(TempDatabase::new());
    let path = database.config().database;
    fill_editor(cx, &handle, &welcome, Engine::Sqlite, &path);

    welcome.update(cx, |welcome, cx| {
        welcome.with_editor_for_test(cx, |editor, cx| editor.connect_for_test(true, cx))
    });
    cx.run_until_parked();

    assert!(
        crate::db::store::load()
            .expect("the store should be readable")
            .iter()
            .any(|config| config.database == path),
        "Connect should have saved the new connection"
    );
    assert!(
        handle
            .update(cx, |workspace, _, _| workspace.active_session_for_test())
            .unwrap()
            .is_some(),
        "and then connected to it"
    );
}

#[gpui_kit::test]
fn connect_without_saving_leaves_the_list_alone(cx: &mut TestAppContext) {
    let dir = ScratchDir::new();
    crate::db::store::set_config_dir_for_test(dir.path.clone());

    let handle = workspace(cx);
    let welcome = new_connection_editor(cx, &handle);
    let database = runtime::block_on(TempDatabase::new());
    fill_editor(
        cx,
        &handle,
        &welcome,
        Engine::Sqlite,
        &database.config().database,
    );

    welcome.update(cx, |welcome, cx| {
        welcome.with_editor_for_test(cx, |editor, cx| editor.connect_for_test(false, cx))
    });
    cx.run_until_parked();

    assert!(
        handle
            .update(cx, |workspace, _, _| workspace.active_session_for_test())
            .unwrap()
            .is_some(),
        "the connection should open"
    );
    assert!(
        crate::db::store::load()
            .expect("the store should be readable")
            .is_empty(),
        "without being saved"
    );
}

#[gpui_kit::test]
fn enter_in_a_field_connects(cx: &mut TestAppContext) {
    let dir = ScratchDir::new();
    crate::db::store::set_config_dir_for_test(dir.path.clone());

    let handle = workspace(cx);
    let welcome = new_connection_editor(cx, &handle);
    let database = runtime::block_on(TempDatabase::new());
    fill_editor(
        cx,
        &handle,
        &welcome,
        Engine::Sqlite,
        &database.config().database,
    );
    cx.update_window(handle.window.into(), |_, window, cx| {
        welcome.update(cx, |welcome, cx| {
            welcome.with_editor_for_test(cx, |editor, cx| editor.focus_name_for_test(window, cx))
        });
    })
    .unwrap();

    press_workspace(cx, &handle, "enter");

    assert!(
        handle
            .update(cx, |workspace, _, _| workspace.active_session_for_test())
            .unwrap()
            .is_some(),
        "Enter in the name box should connect"
    );
}

#[gpui_kit::test]
fn the_editor_leaves_room_for_the_focus_ring(cx: &mut TestAppContext) {
    let handle = workspace(cx);
    new_connection_editor(cx, &handle);

    // The ring is drawn 3px outside an input's border, and the form's scroll
    // area clips whatever passes its edge.
    const RING: f32 = 3.;
    cx.update_window(handle.window.into(), |_, window, cx| {
        window.draw(cx).clear(cx);
        let area = window.find("editor-form").bounds();
        for field in ["field-Name", "field-Host", "field-Port", "field-Database"] {
            let input = window.find(field).bounds();
            assert!(
                input.left() - px(RING) >= area.left()
                    && input.right() + px(RING) <= area.right()
                    && input.top() - px(RING) >= area.top(),
                "{field}'s focus ring would be clipped: {input:?} in {area:?}"
            );
        }
    })
    .unwrap();
}