installer/ncurses-gui/diskpartition.sh

# $Id$

# trims whitespaces from variables
# bash do it itself automatically if the var is not escaped with ' or "
trim() { echo $1; }

adjust_column()
{
	local maxlen="$1"
	local var="$2"
	local len
	local space
	local i

	# get len of var
	len="${#var}"

	for ((i=len; i < maxlen; i++))
	do
		space+=" "
	done

	echo "${var}${space}"
}

get_list()
{
	:> ${mylist}

	# print dialog setup
	echo "--extra-button --extra-label Done --ok-label Edit --colors \\" >> ${mylist}
	# highlite
	echo "--default-item \"${highlite}\" \\" >> ${mylist}
	echo "--menu \"Select a disk or partition:\" \\" >> ${mylist}
	echo "30 70 25 \\" >> ${mylist}

	# print header
	echo "\"$(adjust_column 5 disk) $(adjust_column 10 partition)\" \"\Zb\Z4$(adjust_column 11 size) $(adjust_column 8 fstype) $(adjust_column 4 flag) \Zb\Z6$(adjust_column 10 mountpoint) \Zb\Z6$(adjust_column 8 format)\"" >> ${mylist}
	echo "\"\" \"\"" >> ${mylist}

	OLDIFS="${IFS}"
	IFS="|"
	declare -i count=0
	while read disk partition size fstype boot mount format
	do
		# line is a disk
		if [[ ${disk} = ${partition} ]]
		then
			if [[ ${count} -gt 0 ]]
			then
				echo "\"\" \"\" ${off}" >> ${mylist}
			fi
			echo "\"${disk}\" \"\Zb\Z3[setup partition table]\"" >> ${mylist}
		# line is a partition
		else
			echo "\"$(adjust_column 5 "") $(adjust_column 10 ${partition})\" \"$(adjust_column 11 ${size}) $(adjust_column 8 ${fstype}) $(adjust_column 4 ${boot}) \Zb\Z6$(adjust_column 10 ${mount}) \Zb\Z6$(adjust_column 8 ${format})\"" >> ${mylist}
		fi
		((count++))
	done << EOF
$(include/read_disk.sh)
EOF
	IFS="${OLDIFS}"
}

partition_drive_list()
{
	local mylist=$(mktemp)
	local retval

	# get partition list and write dialog file
	get_list
	# show dialog
	dialog --stdout --file ${mylist}
	retval="$?"
	# remove temp files
	rm ${mylist}

	return ${retval}
}

setup_partition_filesystem()
{
	dialog \
		--stdout \
		--colors \
		--title "${SELECTED_DISK}" \
		--backtitle "${TITLE}" \
		--ok-label "Next" \
		--menu "Select a filesystem:" 20 40 10 \
		"keep-as-is" "" \
		"ext2" "" \
		"ext3" "" \
		"reiserfs" "" \
		"swap" ""
}

setup_partition_mountpoint()
{
	dialog \
		--stdout \
		--colors \
		--title "${SELECTED_DISK}" \
		--backtitle "${TITLE}" \
		--ok-label "Next" \
		--menu "Select a mountpoint:" 20 40 10 \
		"/" "" \
		"/boot" "" \
		"/home" "" \
		"/tmp" "" \
		"/usr" "" \
		"/var" "" \
		"/svr" "" \
		"/opt" ""
}

setup_partition()
{
	FILESYSTEM="$(setup_partition_filesystem)"
	case $? in
		-1)	die "Error!" ;;
		255)	die "aborted by user" ;;

		1)
			partition_drive_list
			# aborted, delete settings for sure
			FILESYSTEM=""
			;;
		0)
			if [[ ${FILESYSTEM} = swap ]]
			then
				MOUNTPOINT="swap"
			else
				MOUNTPOINT="$(setup_partition_mountpoint)"
				case $? in
					-1)	die "Error!" ;;
					255)	die "aborted by user" ;;

					1)
						partition_drive_list
						# aborted, delete settings for sure
						MOUNTPOINT=""
						;;
					0)
						true
						;;
				esac
			fi
			;;
	esac

	export FILESYSTEM
	export MOUNTPOINT
}

query_partition_disk_setup()
{
	local i
	local partition
	local filesystem
	local mountpoint
	local command="$1"
	local given_partition="$2"

	#
	#--overwrite

	for i in ${PARTITION_DISK_SETUP}
	do
		partition=$(echo ${i} | cut -d: -f1)
		filesystem=$(echo ${i} | cut -d: -f2)
		mountpoint=$(echo ${i} | cut -d: -f3)

		if [[ ${partition} = ${given_partition} ]]
		then
			case ${command} in
				--filesystem) echo "${filesystem}"; return 0;;
				--mountpoint) echo "${mountpoint}"; return 0;;
				--delete) export PARTITION_DISK_SETUP="${PARTITION_DISK_SETUP/${i}/}"; return 0;;
			esac
		fi
	done
}

edit_disk_or_partition()
{
	if [[ ! -z ${SELECTED_DISK} ]]
	then
		${sudo} cfdisk ${SELECTED_DISK}
		rundialog_partition_drive_list
	else
		rundialog_setup_partition
	fi
}

rundialog_partition_drive_list()
{
	local retval
	SELECTED_DISK=$(partition_drive_list)
	retval=$?

	# highlite the current item
	highlite="${SELECTED_DISK}"

	# fix partitions, remove trailing empty spaces
	SELECTED_DISK="$(trim ${SELECTED_DISK})"

	# save given variables in env
	# they will be shown as default on error
	export SELECTED_DISK
	export PARTITION_DISK_SETUP

	# convert to some useable vars
	local disk
	local mount
	local fs
	for disk in ${PARTITION_DISK_SETUP}
	do
		disk="$(echo ${disk} | cut -d: -f1)"
		mount=$(query_partition_disk_setup --mountpoint ${disk})
		fs=$(query_partition_disk_setup --filesystem ${disk})

		case ${mount} in
			/)
				export PARTITION_DISK_ROOT="${disk}"
				export PARTITION_FS_ROOT="${fs}"
				;;
			/boot)
				export PARTITION_DISK_BOOT="${disk}"
				export PARTITION_FS_BOOT="${fs}"
				;;
			/home)
				export PARTITION_DISK_HOME="${disk}"
				export PARTITION_FS_HOME="${fs}"
				;;
			/tmp)
				export PARTITION_DISK_TMP="${disk}"
				export PARTITION_FS_TMP="${fs}"
				;;
			/usr)
				export PARTITION_DISK_USER="${disk}"
				export PARTITION_FS_USER="${fs}"
				;;
			/var)
				export PARTITION_DISK_VAR="${disk}"
				export PARTITION_FS_VAR="${fs}"
				;;
			/svr)
				export PARTITION_DISK_SVR="${disk}"
				export PARTITION_FS_SVR="${fs}"
				;;
			/opt)
				export PARTITION_DISK_OPT="${disk}"
				export PARTITION_FS_OPT="${fs}"
				;;
			swap)
				export PARTITION_DISK_SWAP="${disk}"
				export PARTITION_FS_SWAP="swap"
				;;
		esac
	done

	case ${retval} in
		-1)	die "Error!" ;;
		255)	die "aborted by user" ;;

		1)
			partition_drive_list
			;;

		3)
			if [[ -z ${PARTITION_DISK_ROOT} ]] || [[ -z ${PARTITION_DISK_SWAP} ]]
			then
				messagebox "Warning" "You must at least select root (/) and a swap (SWAP) partition!" 10 40
				rundialog_partition_drive_list
			fi
			;;

		0)
			# do nothing if header or emptylines where selected
			if [[ -z ${SELECTED_DISK} ]] || [[ ${SELECTED_DISK} = disk* ]]
			then
				rundialog_partition_drive_list

			# a partition without a number must be the disk
			elif [[ ${SELECTED_DISK/[0-9]/} = ${SELECTED_DISK} ]]
			then
				${sudo} cfdisk ${SELECTED_DISK}
				rundialog_partition_drive_list
			else
				rundialog_setup_partition
			fi
			;;
	esac

	debug "SELECTED_DISK='${SELECTED_DISK}'"
}

rundialog_setup_partition()
{
	setup_partition

	# delete stanza
	query_partition_disk_setup --delete "${SELECTED_DISK}"
	# set the new
	export PARTITION_DISK_SETUP="${PARTITION_DISK_SETUP} ${SELECTED_DISK}:${FILESYSTEM}:${MOUNTPOINT}"

	rundialog_partition_drive_list
}

main_diskpartition()
{
	rundialog_partition_drive_list
}
1	# $Id$
2
3	# trims whitespaces from variables
4	# bash do it itself automatically if the var is not escaped with ' or "
5	trim() { echo $1; }
6
7	adjust_column()
8	{
9	local maxlen="$1"
10	local var="$2"
11	local len
12	local space
13	local i
14
15	# get len of var
16	len="${#var}"
17
18	for ((i=len; i < maxlen; i++))
19	do
20	space+=" "
21	done
22
23	echo "${var}${space}"
24	}
25
26	get_list()
27	{
28	:> ${mylist}
29
30	# print dialog setup
31	echo "--extra-button --extra-label Done --ok-label Edit --colors \\" >> ${mylist}
32	# highlite
33	echo "--default-item \"${highlite}\" \\" >> ${mylist}
34	echo "--menu \"Select a disk or partition:\" \\" >> ${mylist}
35	echo "30 70 25 \\" >> ${mylist}
36
37	# print header
38	echo "\"$(adjust_column 5 disk) $(adjust_column 10 partition)\" \"\Zb\Z4$(adjust_column 11 size) $(adjust_column 8 fstype) $(adjust_column 4 flag) \Zb\Z6$(adjust_column 10 mountpoint) \Zb\Z6$(adjust_column 8 format)\"" >> ${mylist}
39	echo "\"\" \"\"" >> ${mylist}
40
41	OLDIFS="${IFS}"
42	IFS="\|"
43	declare -i count=0
44	while read disk partition size fstype boot mount format
45	do
46	# line is a disk
47	if [[ ${disk} = ${partition} ]]
48	then
49	if [[ ${count} -gt 0 ]]
50	then
51	echo "\"\" \"\" ${off}" >> ${mylist}
52	fi
53	echo "\"${disk}\" \"\Zb\Z3[setup partition table]\"" >> ${mylist}
54	# line is a partition
55	else
56	echo "\"$(adjust_column 5 "") $(adjust_column 10 ${partition})\" \"$(adjust_column 11 ${size}) $(adjust_column 8 ${fstype}) $(adjust_column 4 ${boot}) \Zb\Z6$(adjust_column 10 ${mount}) \Zb\Z6$(adjust_column 8 ${format})\"" >> ${mylist}
57	fi
58	((count++))
59	done << EOF
60	$(include/read_disk.sh)
61	EOF
62	IFS="${OLDIFS}"
63	}
64
65	partition_drive_list()
66	{
67	local mylist=$(mktemp)
68	local retval
69
70	# get partition list and write dialog file
71	get_list
72	# show dialog
73	dialog --stdout --file ${mylist}
74	retval="$?"
75	# remove temp files
76	rm ${mylist}
77
78	return ${retval}
79	}
80
81	setup_partition_filesystem()
82	{
83	dialog \
84	--stdout \
85	--colors \
86	--title "${SELECTED_DISK}" \
87	--backtitle "${TITLE}" \
88	--ok-label "Next" \
89	--menu "Select a filesystem:" 20 40 10 \
90	"keep-as-is" "" \
91	"ext2" "" \
92	"ext3" "" \
93	"reiserfs" "" \
94	"swap" ""
95	}
96
97	setup_partition_mountpoint()
98	{
99	dialog \
100	--stdout \
101	--colors \
102	--title "${SELECTED_DISK}" \
103	--backtitle "${TITLE}" \
104	--ok-label "Next" \
105	--menu "Select a mountpoint:" 20 40 10 \
106	"/" "" \
107	"/boot" "" \
108	"/home" "" \
109	"/tmp" "" \
110	"/usr" "" \
111	"/var" "" \
112	"/svr" "" \
113	"/opt" ""
114	}
115
116	setup_partition()
117	{
118	FILESYSTEM="$(setup_partition_filesystem)"
119	case $? in
120	-1) die "Error!" ;;
121	255) die "aborted by user" ;;
122
123	1)
124	partition_drive_list
125	# aborted, delete settings for sure
126	FILESYSTEM=""
127	;;
128	0)
129	if [[ ${FILESYSTEM} = swap ]]
130	then
131	MOUNTPOINT="swap"
132	else
133	MOUNTPOINT="$(setup_partition_mountpoint)"
134	case $? in
135	-1) die "Error!" ;;
136	255) die "aborted by user" ;;
137
138	1)
139	partition_drive_list
140	# aborted, delete settings for sure
141	MOUNTPOINT=""
142	;;
143	0)
144	true
145	;;
146	esac
147	fi
148	;;
149	esac
150
151	export FILESYSTEM
152	export MOUNTPOINT
153	}
154
155	query_partition_disk_setup()
156	{
157	local i
158	local partition
159	local filesystem
160	local mountpoint
161	local command="$1"
162	local given_partition="$2"
163
164	#
165	#--overwrite
166
167	for i in ${PARTITION_DISK_SETUP}
168	do
169	partition=$(echo ${i} \| cut -d: -f1)
170	filesystem=$(echo ${i} \| cut -d: -f2)
171	mountpoint=$(echo ${i} \| cut -d: -f3)
172
173	if [[ ${partition} = ${given_partition} ]]
174	then
175	case ${command} in
176	--filesystem) echo "${filesystem}"; return 0;;
177	--mountpoint) echo "${mountpoint}"; return 0;;
178	--delete) export PARTITION_DISK_SETUP="${PARTITION_DISK_SETUP/${i}/}"; return 0;;
179	esac
180	fi
181	done
182	}
183
184	edit_disk_or_partition()
185	{
186	if [[ ! -z ${SELECTED_DISK} ]]
187	then
188	${sudo} cfdisk ${SELECTED_DISK}
189	rundialog_partition_drive_list
190	else
191	rundialog_setup_partition
192	fi
193	}
194
195	rundialog_partition_drive_list()
196	{
197	local retval
198	SELECTED_DISK=$(partition_drive_list)
199	retval=$?
200
201	# highlite the current item
202	highlite="${SELECTED_DISK}"
203
204	# fix partitions, remove trailing empty spaces
205	SELECTED_DISK="$(trim ${SELECTED_DISK})"
206
207	# save given variables in env
208	# they will be shown as default on error
209	export SELECTED_DISK
210	export PARTITION_DISK_SETUP
211
212	# convert to some useable vars
213	local disk
214	local mount
215	local fs
216	for disk in ${PARTITION_DISK_SETUP}
217	do
218	disk="$(echo ${disk} \| cut -d: -f1)"
219	mount=$(query_partition_disk_setup --mountpoint ${disk})
220	fs=$(query_partition_disk_setup --filesystem ${disk})
221
222	case ${mount} in
223	/)
224	export PARTITION_DISK_ROOT="${disk}"
225	export PARTITION_FS_ROOT="${fs}"
226	;;
227	/boot)
228	export PARTITION_DISK_BOOT="${disk}"
229	export PARTITION_FS_BOOT="${fs}"
230	;;
231	/home)
232	export PARTITION_DISK_HOME="${disk}"
233	export PARTITION_FS_HOME="${fs}"
234	;;
235	/tmp)
236	export PARTITION_DISK_TMP="${disk}"
237	export PARTITION_FS_TMP="${fs}"
238	;;
239	/usr)
240	export PARTITION_DISK_USER="${disk}"
241	export PARTITION_FS_USER="${fs}"
242	;;
243	/var)
244	export PARTITION_DISK_VAR="${disk}"
245	export PARTITION_FS_VAR="${fs}"
246	;;
247	/svr)
248	export PARTITION_DISK_SVR="${disk}"
249	export PARTITION_FS_SVR="${fs}"
250	;;
251	/opt)
252	export PARTITION_DISK_OPT="${disk}"
253	export PARTITION_FS_OPT="${fs}"
254	;;
255	swap)
256	export PARTITION_DISK_SWAP="${disk}"
257	export PARTITION_FS_SWAP="swap"
258	;;
259	esac
260	done
261
262	case ${retval} in
263	-1) die "Error!" ;;
264	255) die "aborted by user" ;;
265
266	1)
267	partition_drive_list
268	;;
269
270	3)
271	if [[ -z ${PARTITION_DISK_ROOT} ]] \|\| [[ -z ${PARTITION_DISK_SWAP} ]]
272	then
273	messagebox "Warning" "You must at least select root (/) and a swap (SWAP) partition!" 10 40
274	rundialog_partition_drive_list
275	fi
276	;;
277
278	0)
279	# do nothing if header or emptylines where selected
280	if [[ -z ${SELECTED_DISK} ]] \|\| [[ ${SELECTED_DISK} = disk* ]]
281	then
282	rundialog_partition_drive_list
283
284	# a partition without a number must be the disk
285	elif [[ ${SELECTED_DISK/[0-9]/} = ${SELECTED_DISK} ]]
286	then
287	${sudo} cfdisk ${SELECTED_DISK}
288	rundialog_partition_drive_list
289	else
290	rundialog_setup_partition
291	fi
292	;;
293	esac
294
295	debug "SELECTED_DISK='${SELECTED_DISK}'"
296	}
297
298	rundialog_setup_partition()
299	{
300	setup_partition
301
302	# delete stanza
303	query_partition_disk_setup --delete "${SELECTED_DISK}"
304	# set the new
305	export PARTITION_DISK_SETUP="${PARTITION_DISK_SETUP} ${SELECTED_DISK}:${FILESYSTEM}:${MOUNTPOINT}"
306
307	rundialog_partition_drive_list
308	}
309
310	main_diskpartition()
311	{
312	rundialog_partition_drive_list
313	}