use async_trait::async_trait;
use chrono::Utc;
use k8s_openapi::{api::apps::v1::Deployment, apimachinery::pkg::util::intstr::IntOrString};
use ratatui::{
  layout::Rect,
  widgets::{Cell, Row},
  Frame,
};

use super::{
  models::{self, AppResource, KubeResource, Named},
  utils, ActiveBlock, App,
};
use crate::{
  app::key_binding::DEFAULT_KEYBINDING,
  draw_resource_tab,
  network::Network,
  ui::utils::{
    action_hint, describe_yaml_and_logs_hint, draw_describe_block, draw_resource_block,
    draw_yaml_block, get_describe_active, get_resource_title, help_bold_line, responsive_columns,
    style_text, title_with_dual_style, wide_hint, ColumnDef, ResourceTableProps, ViewTier,
  },
};

#[derive(Clone, Debug, PartialEq)]
pub struct KubeDeployment {
  pub name: String,
  pub namespace: String,
  pub ready: String,
  pub updated: i32,
  pub available: i32,
  pub strategy: String,
  pub max_surge: String,
  pub max_unavailable: String,
  pub age: String,
  k8s_obj: Deployment,
}

fn format_int_or_string(v: &IntOrString) -> String {
  match v {
    IntOrString::Int(i) => i.to_string(),
    IntOrString::String(s) => s.clone(),
  }
}

impl From<Deployment> for KubeDeployment {
  fn from(deployment: Deployment) -> Self {
    let (ready, available, updated) = match &deployment.status {
      Some(s) => (
        format!(
          "{}/{}",
          s.available_replicas.unwrap_or_default(),
          s.replicas.unwrap_or_default()
        ),
        s.available_replicas.unwrap_or_default(),
        s.updated_replicas.unwrap_or_default(),
      ),
      _ => ("".into(), 0, 0),
    };

    let (strategy, max_surge, max_unavailable) =
      deployment
        .spec
        .as_ref()
        .map_or((String::new(), String::new(), String::new()), |spec| {
          let strat = spec.strategy.as_ref();
          let type_name = strat.and_then(|s| s.type_.clone()).unwrap_or_default();
          let (surge, unavail) = strat.and_then(|s| s.rolling_update.as_ref()).map_or(
            (String::new(), String::new()),
            |ru| {
              (
                ru.max_surge
                  .as_ref()
                  .map_or(String::new(), format_int_or_string),
                ru.max_unavailable
                  .as_ref()
                  .map_or(String::new(), format_int_or_string),
              )
            },
          );
          (type_name, surge, unavail)
        });

    Self {
      name: deployment.metadata.name.clone().unwrap_or_default(),
      namespace: deployment.metadata.namespace.clone().unwrap_or_default(),
      age: utils::to_age(deployment.metadata.creation_timestamp.as_ref(), Utc::now()),
      available,
      updated,
      ready,
      strategy,
      max_surge,
      max_unavailable,
      k8s_obj: utils::sanitize_obj(deployment),
    }
  }
}

impl Named for KubeDeployment {
  fn get_name(&self) -> &String {
    &self.name
  }
}

impl KubeResource<Deployment> for KubeDeployment {
  fn get_k8s_obj(&self) -> &Deployment {
    &self.k8s_obj
  }
}

impl models::HasPodSelector for KubeDeployment {
  fn pod_label_selector(&self) -> Option<String> {
    self
      .k8s_obj
      .spec
      .as_ref()
      .and_then(|s| s.selector.match_labels.as_ref())
      .filter(|labels| !labels.is_empty())
      .map(models::labels_to_selector)
  }
}

static DEPLOYMENTS_TITLE: &str = "Deployments";

pub struct DeploymentResource {}

#[async_trait]
impl AppResource for DeploymentResource {
  fn render(block: ActiveBlock, f: &mut Frame<'_>, app: &mut App, area: Rect) {
    draw_resource_tab!(
      DEPLOYMENTS_TITLE,
      block,
      f,
      app,
      area,
      Self::render,
      draw_block,
      app.data.deployments
    );
  }

  async fn get_resource(nw: &Network<'_>) {
    let items: Vec<KubeDeployment> = nw.get_namespaced_resources(Deployment::into).await;

    let mut app = nw.app.lock().await;
    app.data.deployments.set_items(items);
  }
}

const DEPLOY_COLUMNS: [ColumnDef; 9] = [
  ColumnDef::all("Namespace", 25, 20, 15),
  ColumnDef::all("Name", 35, 30, 20),
  ColumnDef::all("Ready", 10, 10, 8),
  ColumnDef::all("Up-to-date", 10, 10, 10),
  ColumnDef::all("Available", 10, 10, 8),
  ColumnDef::standard("Strategy", 12, 12),
  ColumnDef::wide("Max Surge", 9),
  ColumnDef::wide("Max Unavail", 9),
  ColumnDef::all("Age", 10, 8, 9),
];

fn draw_block(f: &mut Frame<'_>, app: &mut App, area: Rect) {
  let is_loading = app.is_loading();
  let title = get_resource_title(app, DEPLOYMENTS_TITLE, "", app.data.deployments.items.len());

  let tier = ViewTier::from_width(area.width, app.wide_columns);
  let (headers, widths) = responsive_columns(&DEPLOY_COLUMNS, tier);

  draw_resource_block(
    f,
    area,
    ResourceTableProps {
      title,
      inline_help: help_bold_line(
        format!(
          "{} · {} · {}",
          action_hint("pods", DEFAULT_KEYBINDING.submit.key),
          describe_yaml_and_logs_hint(),
          wide_hint()
        ),
        app.palette,
      ),
      resource: &mut app.data.deployments,
      table_headers: headers,
      column_widths: widths,
    },
    |c| {
      let mut cells = vec![
        Cell::from(c.namespace.to_owned()),
        Cell::from(c.name.to_owned()),
        Cell::from(c.ready.to_owned()),
        Cell::from(c.updated.to_string()),
        Cell::from(c.available.to_string()),
      ];
      if tier >= ViewTier::Standard {
        cells.push(Cell::from(c.strategy.to_owned()));
      }
      if tier >= ViewTier::Wide {
        cells.push(Cell::from(c.max_surge.to_owned()));
        cells.push(Cell::from(c.max_unavailable.to_owned()));
      }
      cells.push(Cell::from(c.age.to_owned()));
      Row::new(cells).style(style_text(app.palette))
    },
    app.palette,
    is_loading,
  );
}

#[cfg(test)]
mod tests {
  use super::*;
  use crate::app::models::HasPodSelector;
  use crate::app::test_utils::*;

  #[test]
  fn test_deployments_from_api() {
    let (deployments, deployment_list): (Vec<KubeDeployment>, Vec<_>) =
      convert_resource_from_file("deployments");

    assert_eq!(deployments.len(), 4);
    assert_eq!(
      deployments[0],
      KubeDeployment {
        name: "metrics-server".into(),
        namespace: "kube-system".into(),
        age: utils::to_age(Some(&get_time("2021-05-10T21:48:06Z")), Utc::now()),
        k8s_obj: deployment_list[0].clone(),
        available: 1,
        updated: 1,
        ready: "1/1".into(),
        strategy: "RollingUpdate".into(),
        max_surge: "25%".into(),
        max_unavailable: "25%".into(),
      }
    );
  }

  #[test]
  fn test_deployment_pod_label_selector() {
    let (deployments, _): (Vec<KubeDeployment>, Vec<_>) = convert_resource_from_file("deployments");

    // metrics-server has matchLabels: {k8s-app: metrics-server}
    let selector = deployments[0].pod_label_selector();
    assert!(selector.is_some());
    assert_eq!(selector.unwrap(), "k8s-app=metrics-server");
  }
}
