set_writemostly.sh: add functions print_model, print_size and use it

hdd/functions.sh

@@ -21,3 +21,16 @@ is_hdd()
 # sdb     1
 # sdc     0
 # sdd     0
+
+print_model()
+{
+    hdparm -i $dev | grep "Model=" | sed -e "s|.*Model=\(.*\), F.*|\1|"
+}
+
+print_size()
+{
+    local sectors
+    sectors="$(hdparm -i $dev | grep "LBAsects=" | sed -e "s|.*LBAsects=\(.*\)|\1|")" #"
+    echo "$(($sectors/2/1024/1024)) MB"
+}
+

hdd/set_writemostly.sh

 #!/bin/sh
 
-is_hdd()
-{
-    hdparm -i "$1" | grep -q "Model=WDC"
-}
+PROGDIR=$(dirname "$0")
+[ "$PROGDIR" = "." ] && PROGDIR=$(pwd)
+. $PROGDIR/functions.sh
 
 use_in_md()
 {
@@ -13,12 +12,13 @@ use_in_md()
 for dev in /dev/sd? ; do
     echo
     if is_hdd $dev ; then
-        echo "HDD:$dev"
+        DRIVETYPE="HDD"
         FLAG=writemostly
     else
-        echo "SSD:$dev"
+        DRIVETYPE="SSD"
         FLAG=-writemostly
     fi
+    echo "$DRIVETYPE:$dev (Model $(print_model $dev), size $(print_size $dev))"
     for md in /dev/md?* ; do
         use_in_md "$md" "$dev" || continue
         devpart="$(mdadm -D "$md" | grep "$dev" | sed -e "s|.*$dev|$dev|")"

hdd/tests.sh

 #!/bin/sh
 
-. ./functions.sh
+PROGDIR=$(dirname "$0")
+[ "$PROGDIR" = "." ] && PROGDIR=$(pwd)
+. $PROGDIR/functions.sh
 
 is_hdd /dev/sda && echo HDD || echo SSD
+dev=/dev/sda
+echo "$dev (Model $(print_model $dev), size $(print_size $dev)"
+
+dev=/dev/sdc
 is_hdd /dev/sdc && echo HDD || echo SSD
+echo "$dev (Model $(print_model $dev), size $(print_size $dev)"