feat: table of content

fromtheeast710 · Dec 3, 2024 · fdf6d71 · fdf6d71
1 parent 4990dd2
commit fdf6d71
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 11 deletions.
diff --git a/src/cli.rs b/src/cli.rs
@@ -14,7 +14,7 @@ pub struct CLI {
   #[arg(short, long, default_value_t = false)]
   pub preview: bool,
 
-  #[arg(short = 'F', long, default_value = "+ **[$name/$owner]($url)** `$star`")]
+  #[arg(short = 'F', long, default_value = "+ **[{owner}/{name}]({url})** `{star}`")]
   pub format: Option<String>,
 
   #[arg(short, long, default_value_t = false)]

diff --git a/src/repo.rs b/src/repo.rs
@@ -28,6 +28,7 @@ impl Repo {
       .send()
       .await?;
     let repos = ocrab.all_pages(pages).await?;
+    // let repo = repos.clone().into_iter().next().expect(&DATA_ERR);
 
     Ok(Self { repos })
   }
@@ -42,15 +43,15 @@ impl Repo {
         .map_or("N/A".to_string(), |l| l.to_string())
         .replace('\"', "")
         .replace(" ", "");
+      let owner = &repo.owner.as_ref().expect(&DATA_ERR).login;
+      let name = &repo.name;
+      let star = &repo.stargazers_count.expect(&DATA_ERR);
+      let url = repo.html_url.as_ref().expect(&DATA_ERR);
+
       // TODO: custom formatting
-      let form = format!(
-        r#"+ **[{}/{}]({})** `{}`"#,
-        repo.owner.as_ref().expect(&DATA_ERR).login,
-        repo.name,
-        repo.html_url.as_ref().expect(&DATA_ERR),
-        repo.stargazers_count.expect(&DATA_ERR),
-      );
+      let form = format!(r#"+ **[{owner}/{name}]({url})** `{star}`"#);
 
+      // TODO: move N/A to the end
       col
         .entry(lang)
         .and_modify(|c| c.push(form.clone()))
@@ -60,16 +61,30 @@ impl Repo {
     col.into_iter().collect()
   }
 
-  // TODO: compress into a table
   pub fn format_table(&self) -> String {
     let langs: Vec<String> = self.iter_repo().keys().cloned().collect();
 
     let mut table = String::new();
 
     table.push_str("## Table of Contents\n");
 
-    for (i, lang) in langs.iter().enumerate() {
-      table.push_str(&format!("  {i:>2}. [{lang}](#{lang})\n"));
+    if langs.iter().count() > 9 {
+      for (i, lang) in langs.iter().enumerate() {
+        if i == 4 {
+          table.push_str("|\n|-|-|-|-|\n")
+        }
+
+        if i % 4 == 0 && i != 0 && i != 4 {
+          table.push_str("|\n");
+        }
+
+        table.push_str(&format!("|**{}. [{lang}](#{lang})**", i + 1));
+      }
+      table.push_str("\n");
+    } else {
+      for (i, lang) in langs.iter().enumerate() {
+        table.push_str(&format!("**{}. [{lang}](#{lang})**\n", i + 1))
+      }
     }
 
     table