mirror of https://github.com/TaKO8Ki/gobang
create properties component
Takayuki Maeda
3 years ago
parent
4bcd4802fc
commit
89a1b8ec7c
@ -0,0 +1,190 @@
|
||||
use super::{Component, EventState, StatefulDrawableComponent};
|
||||
use crate::components::command::{self, CommandInfo};
|
||||
use crate::components::TableComponent;
|
||||
use crate::config::KeyConfig;
|
||||
use crate::database::Pool;
|
||||
use crate::event::Key;
|
||||
use anyhow::Result;
|
||||
use async_trait::async_trait;
|
||||
use database_tree::{Database, Table as DTable};
|
||||
use strum_macros::EnumIter;
|
||||
use tui::{
|
||||
backend::Backend,
|
||||
layout::{Constraint, Direction, Layout, Rect},
|
||||
style::{Color, Style},
|
||||
widgets::{Block, Borders, List, ListItem},
|
||||
Frame,
|
||||
};
|
||||
|
||||
#[derive(Debug, EnumIter)]
|
||||
pub enum Focus {
|
||||
Column,
|
||||
Constraint,
|
||||
ForeignKey,
|
||||
Index,
|
||||
}
|
||||
|
||||
impl std::fmt::Display for Focus {
|
||||
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
|
||||
write!(f, "{:?}", self)
|
||||
}
|
||||
}
|
||||
|
||||
pub struct PropertiesComponent {
|
||||
table: Option<(Database, DTable)>,
|
||||
column_table: TableComponent,
|
||||
constraint_table: TableComponent,
|
||||
foreign_key_table: TableComponent,
|
||||
index_table: TableComponent,
|
||||
focus: Focus,
|
||||
key_config: KeyConfig,
|
||||
}
|
||||
|
||||
impl PropertiesComponent {
|
||||
pub fn new(key_config: KeyConfig) -> Self {
|
||||
Self {
|
||||
table: None,
|
||||
column_table: TableComponent::new(key_config.clone()),
|
||||
constraint_table: TableComponent::new(key_config.clone()),
|
||||
foreign_key_table: TableComponent::new(key_config.clone()),
|
||||
index_table: TableComponent::new(key_config.clone()),
|
||||
focus: Focus::Column,
|
||||
key_config,
|
||||
}
|
||||
}
|
||||
|
||||
fn focused_component(&mut self) -> &mut TableComponent {
|
||||
match self.focus {
|
||||
Focus::Column => &mut self.column_table,
|
||||
Focus::Constraint => &mut self.constraint_table,
|
||||
Focus::ForeignKey => &mut self.foreign_key_table,
|
||||
Focus::Index => &mut self.index_table,
|
||||
}
|
||||
}
|
||||
|
||||
pub async fn update(
|
||||
&mut self,
|
||||
database: Database,
|
||||
table: DTable,
|
||||
pool: &Box<dyn Pool>,
|
||||
) -> Result<()> {
|
||||
self.column_table.reset();
|
||||
let columns = pool.get_columns(&database, &table).await?;
|
||||
if !columns.is_empty() {
|
||||
self.column_table.update(
|
||||
columns
|
||||
.iter()
|
||||
.map(|c| c.columns())
|
||||
.collect::<Vec<Vec<String>>>(),
|
||||
columns.get(0).unwrap().fields(),
|
||||
database.clone(),
|
||||
table.clone(),
|
||||
);
|
||||
}
|
||||
self.constraint_table.reset();
|
||||
let constraints = pool.get_constraints(&database, &table).await?;
|
||||
if !constraints.is_empty() {
|
||||
self.constraint_table.update(
|
||||
constraints
|
||||
.iter()
|
||||
.map(|c| c.columns())
|
||||
.collect::<Vec<Vec<String>>>(),
|
||||
constraints.get(0).unwrap().fields(),
|
||||
database.clone(),
|
||||
table.clone(),
|
||||
);
|
||||
}
|
||||
self.foreign_key_table.reset();
|
||||
let foreign_keys = pool.get_foreign_keys(&database, &table).await?;
|
||||
if !foreign_keys.is_empty() {
|
||||
self.foreign_key_table.update(
|
||||
foreign_keys
|
||||
.iter()
|
||||
.map(|c| c.columns())
|
||||
.collect::<Vec<Vec<String>>>(),
|
||||
foreign_keys.get(0).unwrap().fields(),
|
||||
database.clone(),
|
||||
table.clone(),
|
||||
);
|
||||
}
|
||||
self.index_table.reset();
|
||||
let indexes = pool.get_indexes(&database, &table).await?;
|
||||
if !indexes.is_empty() {
|
||||
self.index_table.update(
|
||||
indexes
|
||||
.iter()
|
||||
.map(|c| c.columns())
|
||||
.collect::<Vec<Vec<String>>>(),
|
||||
indexes.get(0).unwrap().fields(),
|
||||
database.clone(),
|
||||
table.clone(),
|
||||
);
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub fn reset(&mut self) {
|
||||
// self.table.reset();
|
||||
// self.filter.reset();
|
||||
}
|
||||
|
||||
fn tab_names(&self) -> Vec<String> {
|
||||
vec![
|
||||
command::tab_columns(&self.key_config).name,
|
||||
command::tab_constraints(&self.key_config).name,
|
||||
command::tab_foreign_keys(&self.key_config).name,
|
||||
command::tab_indexes(&self.key_config).name,
|
||||
command::tab_sql_editor(&self.key_config).name,
|
||||
command::tab_properties(&self.key_config).name,
|
||||
]
|
||||
}
|
||||
}
|
||||
|
||||
impl StatefulDrawableComponent for PropertiesComponent {
|
||||
fn draw<B: Backend>(&mut self, f: &mut Frame<B>, area: Rect, focused: bool) -> Result<()> {
|
||||
let layout = Layout::default()
|
||||
.direction(Direction::Horizontal)
|
||||
.constraints(vec![Constraint::Length(20), Constraint::Min(1)])
|
||||
.split(area);
|
||||
|
||||
let tab_names = self
|
||||
.tab_names()
|
||||
.iter()
|
||||
.enumerate()
|
||||
.map(|(i, c)| {
|
||||
ListItem::new(c.to_string()).style(if i == 0 {
|
||||
Style::default().bg(Color::Blue)
|
||||
} else {
|
||||
Style::default()
|
||||
})
|
||||
})
|
||||
.collect::<Vec<ListItem>>();
|
||||
|
||||
let tab_list = List::new(tab_names)
|
||||
.block(Block::default().borders(Borders::ALL).style(if focused {
|
||||
Style::default()
|
||||
} else {
|
||||
Style::default().fg(Color::DarkGray)
|
||||
}))
|
||||
.style(Style::default());
|
||||
|
||||
f.render_widget(tab_list, layout[0]);
|
||||
|
||||
self.focused_component().draw(f, layout[1], focused)?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
#[async_trait]
|
||||
impl Component for PropertiesComponent {
|
||||
fn commands(&self, out: &mut Vec<CommandInfo>) {}
|
||||
|
||||
fn event(&mut self, key: Key) -> Result<EventState> {
|
||||
self.focused_component().event(key)?;
|
||||
Ok(EventState::NotConsumed)
|
||||
}
|
||||
|
||||
async fn async_event(&mut self, _key: Key, pool: &Box<dyn Pool>) -> Result<EventState> {
|
||||
Ok(EventState::NotConsumed)
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue