forked from SergKolo/sergrep
-
Notifications
You must be signed in to change notification settings - Fork 0
/
safe_shutdown.sh
168 lines (149 loc) · 4.2 KB
/
safe_shutdown.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
#!/usr/bin/env bash
#
###########################################################
# Author: Serg Kolo , contact: [email protected]
# Date: May 14th , 2016
# Purpose: Ensure that user closes all or specific
# running windows and exits without any work
# lost
# Written for: http://askubuntu.com/q/771227/295286
# Tested on: Ubuntu 14.04 LTS
###########################################################
# Copyright: Serg Kolo , 2016
#
# Permission to use, copy, modify, and distribute this software is hereby granted
# without fee, provided that the copyright notice above and this permission statement
# appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
# THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
# DEALINGS IN THE SOFTWARE.
ARGV0="$0"
ARGC=$#
_notify_user()
{
# Close the shutdown dialog and display
# graphical popup which will ask user's shutdown
# confirmation. If user clicks OK , we shutdown.
# If cancel - no action.
qdbus com.canonical.Unity \
/com/canonical/Unity/Session \
com.canonical.Unity.Session.CancelAction
if zenity --question --title='WARNING!' \
--text="You have running apps. Shutdown anyway ?" \
2> /dev/null
then
qdbus com.canonical.Unity \
/com/canonical/Unity/Session \
com.canonical.Unity.Session.Shutdown
fi
}
_get_running_apps()
{
# Gets list of .desktop files for each
# running app
qdbus org.ayatana.bamf \
/org/ayatana/bamf/matcher \
org.ayatana.bamf.matcher.RunningApplicationsDesktopFiles
}
_check_any_running()
{
# Among the running apps there's always one
# .desktop file, which is compiz.desktop.
# We want to know if there's anything besides that
if [ $( _get_running_apps | wc -l ) -gt 1 ];
then
_notify_user
fi
}
_check_specific_running()
{
# Get list of running apps and see if
# the .desktop file we got is on the list
if _get_running_apps | grep -q "$1"
then
_notify_user
fi
}
_select_app()
{
# xwininfo provides nice interface which allows selecting
# a window. The rest is just simple parsing and passing
# around the XID of the app.
notify-send 'Select a window you would like to monitor '
XID=$(xwininfo -int | awk '/xwininfo: Window id/{print $4}')
APP=$(qdbus org.ayatana.bamf \
/org/ayatana/bamf/matcher \
org.ayatana.bamf.matcher.ApplicationForXid $XID )
qdbus org.ayatana.bamf \
"$APP" org.ayatana.bamf.application.DesktopFile
}
_print_usage()
{
cat <<EOF
safe_shutdown.sh [-a | -c |-s DESKTOP_FILE | -h ]
Options:
-a Monitor any open applications.
-c Graphically select an app
-s specify .desktop file for app on command line
-h print this text
Copyright Serg Kolo , 2016
EOF
}
parse_args()
{
if [ $ARGC -eq 0 ] ; then
printf "%s: No option specified\n Usage:\n" ${ARGV0##*/}
_print_usage
exit 1
fi
local OPTIND opt
while getopts "acs:" opt
do
case ${opt} in
a) FUNCTION="_check_any_running"
break
;;
c)
DESK_FILE=$(_select_app )
FUNCTION=" _check_specific_running $DESK_FILE "
break
;;
s) DESK_FILE=${OPTARG}
FUNCTION=" _check_specific_running $DESK_FILE "
break
;;
h) _print_usage
exit 0
;;
\?)
echo "Invalid option: -$OPTARG" >&2
exit 1
;;
esac
done
shift $((OPTIND-1))
}
main()
{
# Basic idea is to let user chose what to do
# then monitor dbus for appropriate signal
# Once the RebootRequested signal is received
# then perform appropriate checks ( for a specific
# or all apps ).
local FUNCTION
parse_args "$@"
dbus-monitor --profile \
"interface='com.canonical.Unity.Session',type=signal" |
while read -r line;
do
case "$line" in
*RebootRequested*) $FUNCTION ;;
esac
done
}
main "$@"