re_redap_tests 0.31.2

use std::collections::HashMap;

use arrow::datatypes::{DataType, Field, Schema};
use tonic::Code;

use re_protos::cloud::v1alpha1::GetTableSchemaRequest;
use re_protos::cloud::v1alpha1::ext::{
    CreateTableEntryRequest, EntryDetails, LanceTable, ProviderDetails,
};
use re_protos::cloud::v1alpha1::rerun_cloud_service_server::RerunCloudService;

use super::common::entry_name;
use crate::SchemaTestExt as _;

pub async fn create_table_entry(service: impl RerunCloudService) {
    let tmp_dir = tempfile::tempdir().expect("create temp dir");

    let table_name = "created_table";

    let schema = Schema::new(vec![
        Field::new("column_a", DataType::Utf8, false),
        Field::new("column_b", DataType::Int64, true),
        Field::new("column_c", DataType::Float64, false),
        Field::new("column_d", DataType::Boolean, true),
    ]);

    let table_url =
        url::Url::from_directory_path(tmp_dir.path()).expect("create url from tmp directory");
    let provider_details = ProviderDetails::LanceTable(LanceTable { table_url });

    let create_table_request = CreateTableEntryRequest {
        name: entry_name(table_name),
        schema: schema.clone(),
        provider_details: Some(provider_details),
    }
    .try_into()
    .expect("Unable to create table request");

    let response = service
        .create_table_entry(tonic::Request::new(create_table_request))
        .await
        .expect("create table entry");

    let response = response
        .into_inner()
        .table
        .expect("table missing in create_table response")
        .details
        .expect("table entry details missing");
    let entry: EntryDetails = response.try_into().expect("convert into entry details");

    assert_eq!(entry.name, entry_name(table_name));

    let schema_response = service
        .get_table_schema(tonic::Request::new(GetTableSchemaRequest {
            table_id: Some(entry.id.into()),
        }))
        .await
        .expect("get table schema")
        .into_inner();

    let returned_schema: Schema = schema_response
        .schema
        .expect("schema is not set in response")
        .try_into()
        .expect("Unable to convert into schema");

    // Strip the metadata that gets added during creation
    let returned_schema = returned_schema.with_metadata(HashMap::new());

    assert_eq!(schema, returned_schema);

    insta::assert_snapshot!("create_table_data", returned_schema.format_snapshot());
}

pub async fn create_table_entry_duplicate_url(service: impl RerunCloudService) {
    let tmp_dir = tempfile::tempdir().expect("create temp dir");

    let schema = Schema::new(vec![Field::new("column_a", DataType::Utf8, false)]);

    let table_url =
        url::Url::from_directory_path(tmp_dir.path()).expect("create url from tmp directory");
    let provider_details = ProviderDetails::LanceTable(LanceTable {
        table_url: table_url.clone(),
    });

    let create_table_request = CreateTableEntryRequest {
        name: entry_name("table_1"),
        schema: schema.clone(),
        provider_details: Some(provider_details.clone()),
    }
    .try_into()
    .expect("Unable to create table request");

    service
        .create_table_entry(tonic::Request::new(create_table_request))
        .await
        .expect("first create_table_entry should succeed");

    // Second call with the same URL but a different name should fail with AlreadyExists.
    let create_table_request_2 = CreateTableEntryRequest {
        name: entry_name("table_2"),
        schema,
        provider_details: Some(provider_details),
    }
    .try_into()
    .expect("Unable to create table request");

    let err = service
        .create_table_entry(tonic::Request::new(create_table_request_2))
        .await
        .expect_err("second create_table_entry with same URL should fail");

    assert_eq!(err.code(), Code::AlreadyExists);
    assert!(err.message().contains(&table_url.to_string()));
}

pub async fn create_table_entry_failed_does_not_leak_name(service: impl RerunCloudService) {
    // Regression test for https://linear.app/rerun/issue/RR-3644/create-table-failure-leads-to-unlisted-existing-table
    let schema = Schema::new(vec![Field::new("column_a", DataType::Utf8, false)]);

    let table_name = "should_not_leak";

    // First attempt: use an unsupported URL scheme, which should fail.
    let bad_provider = ProviderDetails::LanceTable(LanceTable {
        table_url: url::Url::parse("surprise://bad").expect("parse url"),
    });

    let create_table_request = CreateTableEntryRequest {
        name: entry_name(table_name),
        schema: schema.clone(),
        provider_details: Some(bad_provider),
    }
    .try_into()
    .expect("Unable to create table request");

    service
        .create_table_entry(tonic::Request::new(create_table_request))
        .await
        .expect_err("create_table_entry with unsupported URL scheme should fail");

    // Second attempt: same name but a valid URL — should succeed because the
    // failed first attempt must not have leaked the name into the store.
    let tmp_dir = tempfile::tempdir().expect("create temp dir");
    let good_provider = ProviderDetails::LanceTable(LanceTable {
        table_url: url::Url::from_directory_path(tmp_dir.path())
            .expect("create url from tmp directory"),
    });

    let create_table_request = CreateTableEntryRequest {
        name: entry_name(table_name),
        schema,
        provider_details: Some(good_provider),
    }
    .try_into()
    .expect("Unable to create table request");

    service
        .create_table_entry(tonic::Request::new(create_table_request))
        .await
        .expect("create_table_entry with valid URL should succeed after prior failure");
}