How to write practical shell scripts

In the last post, we discussed Linux sed command and awk command.

During the series, we wrote small shell scripts, but we didn’t mix things up, I think we should take a small step further and write a useful shell script.

However, the scripts in this post will help you to empower your scriptwriting skills.

 

Sending messages

We are going to build a bash script that will send a message to a user who is logged into the Linux system.

For this simple shell script, we need only a few functions. We will use some of the basic commands we covered before; you can review the previous posts.

First, we need to know the logged in users. You can do this using the who command, which retrieves all logged-in users.

$ who

shell scripts who command

To send a message, you need the username and his current terminal.

Then, you need to know if the users are allowing messages or not using the mesg command.

$ mesg

mesg command

If the result shows “is y”, that means the user allows messages. If the result shows “is n”, that means the opposite.

To check any logged user message status, use the who command with -T option.

$ who -T

If you see a dash (-), that means the user turned off the messages, and if you see plus sign (+), that means the opposite.

To allow messages, type mesg command with the “y” option like this:

$ mesg y

allow messages

Sure enough, it shows “is y”, which means messages are permitted for this user.

Of course, we need another user to be able to communicate with him, so in my case, I’m going to connect to my PC using SSH from another PC, and I’m already on the system with my user, so we have two users logged onto the system.

Let’s see how to send a message.

 

Write command

You can use the write command to send messages between users using the username and current terminal.

For those users who logged into the graphical environment (KDE, Gnome, Cinnamon, or any), they can’t receive messages. The user must be logged onto the terminal.

We will send a message to testuser user from my user likegeeks like this:

$ write testuser pts/1

write command

Type the write command followed by the user and the terminal and hit Enter.

When you hit Enter, you can start typing your message. After finishing the message, you can send the message by pressing the Ctrl+D key combination, which is the end of file signal.

Receive message

The receiver can recognize which user on which terminal sends the message. EOF means that the message is finished.

I think now we have all the parts to build our shell script.

 

Creating the send script

Before we create our shell script, we need to determine whether the user we want to send a message to him is currently on the system, you can do this using the who command to determine that.

logged=$(who | awk -v IGNORECASE=1 -v usr=$1 '{ if ($1==usr) { print $1 }exit }')

We get the logged in users using the who command and pipe it to awk and check if it is matching the entered user.

The final output from the awk command is stored in the variable logged.

Then we need to check the variable if it contains something or not:

if [ -z $logged ]; then

	echo "$1 is not logged on."

	echo "Exit"

	exit

fi

I recommend you to read the post about the if statement and how to use it Bash Script.

Check logged user

We tested the logged variable to check if it is a zero or not.

If it is zero, the script prints the message, and the script terminates.

If the user is logged, the logged variable contains the username.

Checking if the user accepts messages

You can use the who command with -T option to check if the user is allowing messages or not.

check=$(who -T | grep -i -m 1 $1 | awk '{print $2}')

if [ "$check" != "+" ]; then

	echo "$1 disable messaging."

	echo "Exit"

	exit

fi

Check message allowed

Notice that we use the who command with -T. This shows a (+) beside the username if he is allowing messaging. Otherwise, it shows a (-) beside the username if he turned it off.

Finally, we check for a messaging indicator if the indicator not equal to the plus sign (+).

Checking if the message was included

You can check if the sender typed the message or not like this:

if [ -z $2 ]; then

	echo "Message not found"

	echo "Exit"

	exit

fi

Getting the current terminal

Before we send a message, we need to get the current user terminal and store it in a variable.

terminal=$(who | grep -i -m 1 $1 | awk '{print $2}')

Then we can send the message:

echo $2 | write $logged $terminal

Now we can test the whole shell script to see how it goes:

$ ./senderscript likegeeks welcome

Let’s see the other shell window:

Send message

Good!  You can now send simple one-word messages.

Sending a long message

If you try to send more than one word:

$ ./senderscript likegeeks welcome to shell scripting

One word message

It didn’t work. Only we sent the first word of the message.

To fix this problem, we will use the shift command with the while loop.

shift

while [ -n "$1" ]; do

	message=$message' '$1

	shift

done

And now, one thing we need to fix, which is the message parameter.

echo $whole_message | write $logged $terminal

So now the whole script should be like this:

#!/bin/bash

logged=$(who | awk -v IGNORECASE=1 -v usr=$1 '{ if ($1==usr) { print $1 }exit }')

if [ -z $logged ]; then

	echo "$1 is not logged on."

	echo "Exit"

	exit

fi

check=$(who -T | grep -i -m 1 $1 | awk '{print $2}')

if [ "$check" != "+" ]; then

	echo "$1 disable messaging."

	echo "Exit"

	exit

fi

if [ -z $2 ]; then

	echo "Message not found"

	echo "Exit"

	exit

fi

terminal=$(who | grep -i -m 1 $1 | awk '{print $2}')

shift

while [ -n "$1" ]; do

	message=$message' '$1

	shift

done

echo $message | write $logged $terminal

If you try now:

$ ./senderscript likegeeks welcome to shell scripting

Complete message

Awesome!! It worked. Again, I’m not here to make a script to send the message to the user, but the main goal is to review our shell scripting knowledge and use all the parts we’ve learned together and seen how things work together.

 

Monitoring disk space

Let’s build a script that monitors the biggest top ten directories.

If you add -s option to the du command, it will show summarized totals.

$ du -s /var/log/

We use the -S option to show the subdirectories totals.

$ du -S /var/log/

du command

You should use the sort command to sort the results generated by the du command to get the largest directories like this:

$ du -S /var/log/ | sort -rn

sort command

The -n to sort numerically and the -r option to reverse the order, so it shows the bigger first.

We use the N command to label each line with a number:

sed '{11,$D; =}' |

sed 'N; s/\n/ /' |

Then we can clean the output using the awk command:

awk '{printf $1 ":" "\t" $2 "\t" $3 "\n"}'

Then we add a colon and a tab, so it appears much better.

$ du -S /var/log/ |

sort -rn |

sed '{11,$D; =}' |

# pipe the first result for another one to clean it

sed 'N; s/\n/ /' |

# formated printing using printf

awk '{printf $1 ":" "\t" $2 "\t" $3 "\n"}'

Format output with sed and awk

Suppose we have a variable called  MY_DIRECTORIES that holds two folders.

MY_DIRECTORIES=”/home /var/log”

We will iterate over each directory from MY_DIRECTORIES variable and get the disk usage using du command.

So the shell script will look like this:

#!/bin/bash

MY_DIRECTORIES="/home /var/log"

echo "Top Ten Directories"

for DIR in $MY_DIRECTORIES; do

	echo "The $DIR Directory:"

	du -S $DIR 2>/dev/empty |
		sort -rn |
		sed '{11,$D; =}' |

		# pipe the first result for another one to clean it
		sed 'N; s/\n/ /' |

		# formated printing using printf
		awk '{printf $1 "\t" "\t" $2 "\t" $3 "\r\n"}'

done

exit

Monitor disk usage

Good!! Both directories /home and /var/log are on the same report.

You can filter files, so instead of calculating the consumption of all files, you can calculate the consumption for a specific extension like *.log or whatever.

One thing I have to mention here, in production systems, you can’t rely on disk space reports; instead, you should use disk quotas.

Quota package is suitable for that, but here we are learning how bash scripts work.

Again the shell scripts we’ve introduced here are for showing you how shell scripting works. There are a ton of ways to implement any task in Linux.

I tried to reduce the post length and make everything as simple as possible, hope you like it.

Keep coming back. Thank you.

6 thoughts on “How to write practical shell scripts
  1. Very good, but can’t get ‘Sending Messages’ to work.
    I have a ‘Ubuntu 10.04’
    Could you Please send me a list of other information you have writen!
    Than you
    Philip

    1. Thanks!

      There are no other steps to run the script. try to run it and see what you get.

      Regards,

  2. Hello Mokhtar,

    could you please explain what does the ” =” mean in the `sed ‘{11,$D; =}’ `?

    Thanks a lot

    1. Hi,
      The equal sign in Linux sed means print the current input line number with newline after it.

      Regards,

Leave a Reply

Your email address will not be published. Required fields are marked *