bestool-psql 1.7.0

use std::ops::ControlFlow;

use bestool_postgres::error::format_error;
use comfy_table::Table;

use crate::repl::state::ReplContext;

use super::output::OutputWriter;

pub(super) async fn handle_describe_view(
	ctx: &mut ReplContext<'_>,
	schema: &str,
	view_name: &str,
	detail: bool,
	sameconn: bool,
	writer: &OutputWriter,
) -> ControlFlow<()> {
	let columns_query = r#"
		SELECT
			a.attname AS column_name,
			pg_catalog.format_type(a.atttypid, a.atttypmod) AS data_type,
			CASE
				WHEN a.attstorage = 'p' THEN 'plain'
				WHEN a.attstorage = 'e' THEN 'external'
				WHEN a.attstorage = 'm' THEN 'main'
				WHEN a.attstorage = 'x' THEN 'extended'
				ELSE ''
			END AS storage
		FROM pg_catalog.pg_class c
		LEFT JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace
		LEFT JOIN pg_catalog.pg_attribute a ON a.attrelid = c.oid
		WHERE n.nspname = $1
			AND c.relname = $2
			AND c.relkind IN ('v', 'm')
			AND a.attnum > 0
			AND NOT a.attisdropped
		ORDER BY a.attnum
	"#;

	let view_info_query = r#"
		SELECT
			c.relkind::text AS view_kind,
			pg_catalog.pg_get_viewdef(c.oid, true) AS view_definition,
			pg_size_pretty(pg_total_relation_size(c.oid)) AS size,
			obj_description(c.oid, 'pg_class') AS view_comment
		FROM pg_catalog.pg_class c
		LEFT JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace
		WHERE n.nspname = $1
			AND c.relname = $2
			AND c.relkind IN ('v', 'm')
	"#;

	let columns_result = if sameconn {
		ctx.client
			.query(columns_query, &[&schema, &view_name])
			.await
	} else {
		match ctx.pool.get().await {
			Ok(client) => client.query(columns_query, &[&schema, &view_name]).await,
			Err(e) => {
				eprintln!("Error getting connection from pool: {}", format_error(&e));
				return ControlFlow::Continue(());
			}
		}
	};

	match columns_result {
		Ok(rows) => {
			if rows.is_empty() {
				eprintln!("No columns found for view \"{}.{}\".", schema, view_name);
				return ControlFlow::Continue(());
			}

			let view_info_result = if sameconn {
				ctx.client
					.query(view_info_query, &[&schema, &view_name])
					.await
			} else {
				match ctx.pool.get().await {
					Ok(client) => client.query(view_info_query, &[&schema, &view_name]).await,
					Err(_) => {
						return ControlFlow::Continue(());
					}
				}
			};

			let (view_kind, view_definition, size, view_comment) =
				if let Ok(info_rows) = view_info_result {
					if let Some(row) = info_rows.first() {
						let kind: String = row.get(0);
						let def: String = row.get(1);
						let sz: String = row.get(2);
						let cmt: Option<String> = row.get(3);
						(Some(kind), Some(def), Some(sz), cmt)
					} else {
						(None, None, None, None)
					}
				} else {
					(None, None, None, None)
				};

			let view_type = match view_kind.as_deref() {
				Some("m") => "Materialized View",
				Some("v") => "View",
				_ => "View",
			};

			writer
				.writeln(&format!("{} \"{}.{}\"", view_type, schema, view_name))
				.await;

			let mut table = Table::new();
			crate::table::configure(&mut table);

			table.set_header(vec!["Column", "Type", "Storage"]);

			for row in rows {
				let column_name: String = row.get(0);
				let data_type: String = row.get(1);
				let storage: String = row.get(2);

				table.add_row(vec![column_name, data_type, storage]);
			}

			crate::table::style_header(&mut table);
			writer.writeln(&format!("{table}")).await;

			if detail && let Some(definition) = view_definition {
				writer.writeln("\nView definition:").await;
				writer.writeln(&definition).await;
			}

			if let Some(sz) = &size {
				writer.writeln(&format!("\nSize: {}", sz)).await;
			}

			if detail
				&& let Some(comment) = view_comment
				&& !comment.is_empty()
			{
				writer.writeln(&format!("Comment: {}", comment)).await;
			}

			writer.writeln("").await;
			ControlFlow::Continue(())
		}
		Err(e) => {
			eprintln!(
				"Error describing view \"{}.{}\": {}",
				schema,
				view_name,
				format_error(&e)
			);
			ControlFlow::Continue(())
		}
	}
}