Help with AWK

Hi All,

I'm usually pretty good with this stuff, but my current shell script project has got me stumped. Essentially, I've created a dynamic shell menu based upon files in a single folder. It's tree-based and uses a numeric and standardised naming format:

#ls -l
total 0
-rw-r--r--  1 root eai 0 Oct  8 08:49 011_Functionality_Example.sh
-rw-r--r--  1 root eai 0 Oct  8 08:49 012_Another_Example.sh
-rw-r--r--  1 root eai 0 Oct  8 08:48 01_Menu_1
-rw-r--r--  1 root eai 0 Oct  8 08:49 021_More_Functionality.sh
-rw-r--r--  1 root eai 0 Oct  8 08:49 02_Menu_2
-rw-r--r--  1 root eai 0 Oct  8 08:48 0_Main_Menu

My shell script works well, but at the moment I can only provide the user with the following to assist in menu selection:

#ls -l | awk '{print $9}' | sort -n

0_Main_Menu
01_Menu_1
02_Menu_2
011_Functionality_Example.sh
012_Another_Example.sh
021_More_Functionality.sh

...but clearly this does not show files in a tree structure. Ideally, this is what I would like to see:

0)   Main Menu
01)  Menu 1
011) Functionality Example
012) Another Example
02)  Menu 2
021) More Functionality

In reality, I won't display the entire tree at any given time, but it is the following that I am looking to do for each particular ls using awk:

  • Drop the suffix '.sh', if any
  • Remove all underscores
  • Sort into a tree-like order*
  • Swap the first space for right bracket and tab

*Optional, since I will typically be displaying files from the same 'level', and hence a numeric sort will suffice.

Thanks in advance if you can help me out... :smiley:

Maybe using a multidimensional array will make it easier to achieve this objective? Or you can use one array to store the names as reference to the arrays containing the menu items. The sorting and positioning problem is solved then.

Arrays - The GNU Awk User's Guide

You can start with something like this:

> ll 
total 0
-rw-r--r--   1 oracle     dba              0 Oct  8 10:24 011_Functionality_Example.sh
-rw-r--r--   1 oracle     dba              0 Oct  8 10:24 012_Another_Example.sh
-rw-r--r--   1 oracle     dba              0 Oct  8 10:24 01_Menu_1
-rw-r--r--   1 oracle     dba              0 Oct  8 10:24 021_More_Functionality.sh
-rw-r--r--   1 oracle     dba              0 Oct  8 10:24 02_Menu_2
-rw-r--r--   1 oracle     dba              0 Oct  8 10:24 0_Main_Menu
> ls |awk -F\_ '{gsub(/\..+$/,"",$0);printf("\n%s)\t",$1);for (i=2;i<=NF-1;i++)printf("%s ",$i);printf("%s",$NF) }'|grep -vE "^$"|sort -k1
0)      Main Menu
01)     Menu 1
011)    Functionality Example
012)    Another Example
02)     Menu 2
021)    More Functionality

with GNU awk

ls ..... | awk '{ 
gsub(/\.sh?/,"",$9)
file=$9
n=split($9,a,"_")
temp[a[1]]=file    
}END{
     m = asorti(temp, newtemp)    
     for(i=1;i<=m;i++){
          gsub( newtemp"_",") ",temp[newtemp]) 
          print newtemp temp[newtemp]
     }
}
'

output:

# ./test.sh

0) Main_Menu
01) Menu_1
011) Functionality_Example
012) Another_Example
02) Menu_2
021) More_Functionality

Massive thanks to all three responses - I will be using a combination of your suggestions in the final version. Here's how it looks for now for anyone who may need similar:

#!/bin/ksh
# Script Name   : test.sh
# Purpose       : Basic User Interface
# Origin Date   : 02/10/07
# Version       : 1.0
# Author        : 
# Description   : Basic dynamic menu system for Unix scripts.

#===================================================
f_get_key()
#---------------------------------------------------
# Function to return the next keystroke.
#===================================================
{
    oldstty=`stty -g`
    stty -icanon -echo min 1 time 0
    key=`dd bs=1 count=1 2>/dev/null`
    stty "$oldstty"
}

#===================================================
f_head()
#---------------------------------------------------
# Function to clear the screen and draw the header.
#===================================================
{
    if [ "$nomenu" = "1" ]; then
        echo "===================================================="
        echo "$*"
        echo "===================================================="
    else
        clear
        echo "####################################################"
        echo "#                       _        _            _    #"
        echo "#   ___ _ __ ___   ___ | | _____| |_ ___  ___| |_  #"
        echo "#  / __| '_ \` _ \ / _ \| |/ / _ \ __/ _ \/ __| __| #"
        echo "#  \__ \ | | | | | (_) |   <  __/ ||  __/\__ \ |_  #"
        echo "#  |___/_| |_| |_|\___/|_|\_\___|\__\___||___/\__| #"
        echo "#                                                  #"
        echo "####################################################"
        echo `date '+%d/%m/%y'`"                                       "`date '+%H:%M'`
        echo "===================================================="
        echo
        c_width=52
        c_text="- $* - $current -"
        c_width=$(( ($c_width + ${#c_text}) / 2 )) 
        printf "%${c_width}.${c_width}s\n" "$c_text"
        echo
    fi
}

#===================================================
f_foot()
#---------------------------------------------------
# Function to add the footer and request an input.
#===================================================
{
    if [ "$nomenu" = "1" ]; then
        echo
        echo
    else
        echo
        echo "===================================================="
        echo "         MAIN MENU: 0 | UP: RETURN | EXIT: x        "
        echo "===================================================="
        f_get_key
    fi
}

#===================================================
#---------------------------------- END OF FUNCTIONS
#---------------------------------------------------
# START OF MAIN PROGRAM ----------------------------
#===================================================

current=""
parent=""
key=""
nomenu=""

if [ "$1" != "" ]; then
    current=$1
	key=unset
else
    current=0
fi

until [ "$key" = "x" ]
do
    if [ "$current" = "0" ]; then
        parent=0
    else
        parent=`echo $current | awk '{print substr($0,1,length($0)-1)}'`
    fi
    
	temp=$current$key"_*"
	if [ -e $temp ]; then
		current=$current$key
	else
		current=$current
	fi
	
	if [ "$key" = "" ]; then
		current=$parent
	elif [ "$key" = "0" ]; then
		current=0
	fi
	
	temp=$current"_*"
	title=`ls -l $temp | awk '{print $9}' | sed 's/^[0-9]*_//;s/_/ /g;s/.sh$//'`
	temp=$current"?_*"
	if [ -e $temp ]; then
		f_head $title
		ls -l $temp | awk '{print $9}' | sort -n | sed 's/^[0-9]*\([0-9]\)_/ \1) /;s/_/ /g;s/.sh$//'
		f_foot
	else
		temp=$current"_*.sh"
		f_head $title
		if [ -x $temp ]; then
			./$temp
		else
			echo "No associated executables ($temp) have been found!"
		fi
		f_foot
	fi
done
clear
echo "...THANKS FOR USING THE SMOKETEST!"
echo
exit 0

N.B. Naming convention is per the question.

I think this solution should be ok.

Just add follow code to your code:

sed 's/_/) /' a | sed 's/_/ /g'

so the total code should be:

awk '{print $9}' | sort -n | sed 's/_/) /' | sed 's/_/ /g'