feat(show): 添加完整显示指定元数据功能，并支持不显示张量信息

Signed-off-by: YdrMaster <ydrml@hotmail.com>

.cargo/config.toml

@@ -2,5 +2,6 @@
 xtask = "run --package xtask --release --"
 debug = "run --package xtask --"
 show = "xtask show"
+show-meta = "xtask show-meta"
 split = "xtask split"
 merge = "xtask merge"

xtask/src/main.rs

@@ -13,6 +13,7 @@ fn main() {
     use Commands::*;
     match Cli::parse().command {
         Show(args) => args.show(),
+        ShowMeta(args) => args.show(),
         Split(args) => args.split(),
         Merge(args) => args.merge(),
     }
@@ -29,6 +30,7 @@ struct Cli {
 #[derive(Subcommand)]
 enum Commands {
     Show(show::ShowArgs),
+    ShowMeta(show::ShowMetaArgs),
     Split(split::SplitArgs),
     Merge(merge::MergeArgs),
 }

xtask/src/show.rs

@@ -12,6 +12,9 @@ pub struct ShowArgs {
     /// If set, show all shards in the directory
     #[clap(long)]
     shards: bool,
+    /// If set, do not show tensor info
+    #[clap(long)]
+    no_tensor: bool,
 }
 
 struct Failed;
@@ -64,7 +67,7 @@ impl ShowArgs {
                     continue;
                 }
             };
-            if let Err(Failed) = show_meta_kvs(&kvs) {
+            if show_meta_kvs(&kvs).is_err() || self.no_tensor {
                 println!();
                 continue;
             }
@@ -146,7 +149,7 @@ fn show_meta_kv(kv: GGufMetaKV, width: usize) -> Result<(), Failed> {
     let ty = kv.ty();
     let mut reader = kv.value_reader();
     let mut buf = String::new();
-    match fmt_meta_val(&mut reader, ty, 1, &mut buf) {
+    match fmt_meta_val(&mut reader, ty, 1, 8, &mut buf) {
         Ok(()) => {
             println!("{YES}{key:·<width$} {buf}");
             Ok(())
@@ -162,6 +165,7 @@ fn fmt_meta_val<'a>(
     reader: &mut GGufReader<'a>,
     ty: GGufMetaDataValueType,
     len: usize,
+    ele_arr_to_show: usize,
     buf: &mut String,
 ) -> Result<(), GGufReadError<'a>> {
     struct MultiLines<'a>(&'a str);
@@ -212,24 +216,24 @@ fn fmt_meta_val<'a>(
                 }
                 T::Array => {
                     let (ty, len) = reader.read_arr_header()?;
-                    fmt_meta_val(reader, ty, len, buf)?;
+                    fmt_meta_val(reader, ty, len, ele_arr_to_show, buf)?;
                 }
             }
         }
-        _ if len <= 8 => {
+        _ if len <= ele_arr_to_show => {
             buf.push('[');
             for i in 0..len {
                 if i > 0 {
                     buf.push_str(", ");
                 }
-                fmt_meta_val(reader, ty, 1, buf)?;
+                fmt_meta_val(reader, ty, 1, ele_arr_to_show, buf)?;
             }
             buf.push(']');
         }
         _ => {
             buf.push('[');
-            for _ in 0..8 {
-                fmt_meta_val(reader, ty, 1, buf)?;
+            for _ in 0..ele_arr_to_show {
+                fmt_meta_val(reader, ty, 1, ele_arr_to_show, buf)?;
                 buf.push_str(", ");
             }
             buf.push_str(&format!("...({} more)]", len - 8));
@@ -256,3 +260,51 @@ fn show_tensors(tensors: &GGufTensors) -> Result<(), Failed> {
 
     Ok(())
 }
+
+#[derive(Args, Default)]
+pub struct ShowMetaArgs {
+    /// The file to show
+    file: PathBuf,
+    /// Meta data key to show
+    key: String,
+}
+
+impl ShowMetaArgs {
+    pub fn show(self) {
+        let Self { file, key } = self;
+
+        let file = match File::open(&file) {
+            Ok(file) => file,
+            Err(e) => {
+                eprintln!("Failed to open");
+                eprintln!("  file: {}", file.display());
+                eprintln!("  cause: {e}");
+                return;
+            }
+        };
+        let file = unsafe { memmap2::Mmap::map(&file) }.unwrap();
+        let header = unsafe { file.as_ptr().cast::<GGufFileHeader>().read() };
+        if let Err(Failed) = show_header(&header) {
+            return;
+        }
+
+        let cursor = header.nbytes();
+        let kvs = match GGufMetaKVPairs::scan(header.metadata_kv_count, &file[cursor..]) {
+            Ok(kvs) => kvs,
+            Err(e) => {
+                eprintln!("{ERR}{e:?}");
+                return;
+            }
+        };
+
+        let Some(kv) = kvs.get(&key) else {
+            eprintln!("{ERR}Key \"{key}\" not found");
+            return;
+        };
+        let mut buf = String::new();
+        match fmt_meta_val(&mut kv.value_reader(), kv.ty(), 1, usize::MAX, &mut buf) {
+            Ok(()) => println!("{YES}{key}: {buf}"),
+            Err(e) => eprintln!("{ERR}{key} {e:?}"),
+        }
+    }
+}