add gpu_name and ip

Author: bleedingfight

README.md

@@ -4,12 +4,13 @@ Enables displaying CPU and GPU information in Tmux `status-right` and `status-le
 Configurable percentage and icon display.
 
 ## Installation
+
 ### Installation with [Tmux Plugin Manager](https://github.com/tmux-plugins/tpm) (recommended)
 
 Add plugin to the list of TPM plugins in `.tmux.conf`:
 
 ```shell
-set -g @plugin 'tmux-plugins/tmux-cpu'
+set -g @plugin 'tmux-plugins/tmux-hardware'
 ```
 
 Hit `prefix + I` to fetch the plugin and source it.
@@ -21,7 +22,7 @@ If format strings are added to `status-right`, they should now be visible.
 Clone the repo:
 
 ```shell
-$ git clone https://github.com/tmux-plugins/tmux-cpu ~/clone/path
+$ git clone https://github.com/tmux-plugins/tmux-hardware ~/clone/path
 ```
 
 Add this line to the bottom of `.tmux.conf`:
@@ -42,13 +43,13 @@ If format strings are added to `status-right`, they should now be visible.
 ### Optional requirements (Linux, BSD, OSX)
 
 - `iostat` or `sar` are the best way to get an accurate CPU percentage.
-A fallback is included using `ps -aux` but could be inaccurate.
+  A fallback is included using `ps -aux` but could be inaccurate.
 - `free` is used for obtaining system RAM status.
 - `lm-sensors` is used for CPU temperature.
 - `nvidia-smi` is required for GPU information.
-For OSX, `cuda-smi` is required instead (but only shows GPU memory use rather than load).
-If "No GPU" is displayed, it means the script was not able to find `nvidia-smi`/`cuda-smi`.
-Please make sure the appropriate command is installed and in the `$PATH`.
+  For OSX, `cuda-smi` is required instead (but only shows GPU memory use rather than load).
+  If "No GPU" is displayed, it means the script was not able to find `nvidia-smi`/`cuda-smi`.
+  Please make sure the appropriate command is installed and in the `$PATH`.
 
 ## Usage
 
@@ -77,6 +78,7 @@ This is done by introducing 12 new format strings that can be added to
 - `#{cpu_temp}` - will show CPU temperature (averaged across cores)
 - `#{cpu_temp_bg_color}` - will change the background color based on the CPU temperature
 - `#{cpu_temp_fg_color}` - will change the foreground color based on the CPU temperature
+- `#{host_ip}` - will change the ip of hostname
 
 GPU equivalents also exist:
 
@@ -92,6 +94,7 @@ GPU equivalents also exist:
 - `#{gpu_temp}` - will show GPU temperature (average across devices)
 - `#{gpu_temp_bg_color}` - will change the background color based on the GPU temperature
 - `#{gpu_temp_fg_color}` - will change the foreground color based on the GPU temperature
+- `#{gpu_name}` - will change the name of GPU
 
 ## Examples
 

cpu.tmux

@@ -29,6 +29,8 @@ cpu_interpolation=(
   "\#{gpu_temp_icon}"
   "\#{gpu_temp_bg_color}"
   "\#{gpu_temp_fg_color}"
+  "\#{gpu_name}"
+  "\#{host_ip}"
 )
 cpu_commands=(
   "#($CURRENT_DIR/scripts/cpu_percentage.sh)"
@@ -55,6 +57,8 @@ cpu_commands=(
   "#($CURRENT_DIR/scripts/gpu_temp_icon.sh)"
   "#($CURRENT_DIR/scripts/gpu_temp_bg_color.sh)"
   "#($CURRENT_DIR/scripts/gpu_temp_fg_color.sh)"
+  "#($CURRENT_DIR/scripts/gpu_name.sh)"
+  "#($CURRENT_DIR/scripts/host_ip.sh)"
 )
 
 set_tmux_option() {

scripts/gpu_name.sh

@@ -0,0 +1,30 @@
+#!/usr/bin/env bash
+
+CURRENT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+
+# shellcheck source=scripts/helpers.sh
+source "$CURRENT_DIR/helpers.sh"
+
+print_gpu_name() {
+
+  if command_exists "nvidia-smi"; then
+    loads=$(cached_eval nvidia-smi)
+  elif command_exists "cuda-smi"; then
+    loads=$(cached_eval cuda-smi)
+  else
+    echo "No GPU"
+    return
+  fi
+  gpu_name=$(echo "$loads" | nvidia-smi | grep -Eo "NVIDIA\s+\w+\s+\w+\s[0-9]+")
+  gpu_nums=$(nvidia-smi --query-gpu=count --format=csv,noheader)
+  if [ ${gpu_nums} -eq 1 ]; then
+    echo ${gpu_name}
+  else
+    echo "${gpu_name}${gpu_nums}"
+  fi
+}
+
+main() {
+  print_gpu_name
+}
+main "$@"

scripts/host_ip.sh

@@ -0,0 +1,23 @@
+#!/usr/bin/env bash
+
+CURRENT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+
+# shellcheck source=scripts/helpers.sh
+source "$CURRENT_DIR/helpers.sh"
+
+print_host_ip() {
+  # 判断系统类型
+  if [[ "$(uname)" == "Linux" ]]; then
+    host_ip=$(ip addr show | grep 'inet ' | grep -v '127.0.0.1' | awk '{print $2}' | cut -d'/' -f1 | head -n 1)
+  elif [[ "$(uname)" == "Darwin" ]]; then
+    host_ip=$(ifconfig | grep 'inet ' | grep -v '127.0.0.1' | awk '{print $2}' | head -n 1)
+  else
+    echo "Unsupported operating system"
+  fi
+  echo ${host_ip}
+}
+
+main() {
+  print_host_ip
+}
+main "$@"