# If you are curious why partman-auto is so slow, it is because
# update-all is slow
update_all () {
	local dev num id size type fs path name partitions
	for dev in $DEVICES/*; do
		[ -d "$dev" ] || continue
		cd $dev
		partitions=''
		open_dialog PARTITIONS
		while { read_line num id size type fs path name; [ "$id" ]; }; do
			partitions="$partitions $id"
		done
		close_dialog
		for id in $partitions; do
			update_partition $dev $id
		done
	done
}

autopartitioning_failed () {
	db_input critical partman-auto/autopartitioning_failed || true
	db_go || true
	update_all
	exit 1
}

find_method () {
	local num id size type fs path name method found minsize
	minsize=$((${2:-0}*1000000))
	found=
	open_dialog PARTITIONS
	while { read_line num id size type fs path name; [ "$id" ]; }; do
		# skip partition if size < minimum size
		[ "$size" -ge "$minsize" ] || continue
		[ -f $id/method-old ] || continue
		method="$(cat $id/method-old)"
		if [ "$method" = "$1" ]; then
			found="$id"
		fi
	done
	close_dialog
	echo "$found"
}

cap_ram () {
    local ram
    ram="$1"
    db_get partman-auto/cap-ram
    # test that return string is all numbers, otherwise do not cap
    if [ $(expr "x$RET" : "x[0-9]*$") -gt 1 ]; then
	if [ $ram -gt "$RET" ]; then
	    ram=$RET
	fi
    fi
    echo "$ram"
}

unnamed=0

decode_recipe () {
	local ignore ram line word min factor max fs iflabel label map map_end -
	local reusemethod method id
	local reuse=$3
	ignore="${2:+${2}ignore}"
	unnamed=$(($unnamed + 1))
	ram=
	for map in /sys/firmware/memmap/*; do
		[ -d "$map" ] || continue
		if [ "$(cat $map/type)" = "System RAM" ]; then
			map_start="$(printf %d "$(cat $map/start)")"
			map_end="$(printf %d "$(cat $map/end)")"
			ram="$(expr "${ram:-0}" + \
				    "$map_end" - "$map_start" + 1)"
		fi
	done
	if [ -z "$ram" ]; then
		ram=$(grep ^Mem: /proc/meminfo | { read x y z; echo $y; }) # in bytes
	fi
	if [ -z "$ram" ]; then
		ram=$(grep ^MemTotal: /proc/meminfo | { read x y z; echo $y; })000
	fi
	ram=$(convert_to_megabytes $ram)
	ram=$(cap_ram $ram)
	name="Unnamed.${unnamed}"
	scheme=''
	line=''
	for word in $(cat $1); do
		case $word in
		    :)
			name=$line
			line=''
			;;
		    ::)
			db_metaget $line description || RET=''
			name="${RET:-Unnamed.$unnamed}"
			line=''
			;;
		    .)
			# we correct errors in order not to crash parted_server
			set -- $line
			if expr "$1" : '[0-9][0-9]*$' >/dev/null; then
				min=$1
			elif expr "$1" : '[0-9][0-9]*+[0-9][0-9]*%$' >/dev/null; then
				ram_percent="${1#*+}"
				ram_percent="${ram_percent%?}"
				min=$((${1%%+*} + $ram * $ram_percent / 100))
			elif expr "$1" : '[0-9][0-9]*%$' >/dev/null; then
				min=$(($ram * ${1%?} / 100))
			else # error
				min=2200000000 # there is no so big storage device jet
			fi
			if expr "$2" : '[0-9][0-9]*+[0-9][0-9]*%$' >/dev/null; then
				ram_percent="${2#*+}"
				ram_percent="${ram_percent%?}"
				factor=$((${2%%+*} + $ram * $ram_percent / 100))
			elif expr "$2" : '[0-9][0-9]*%$' >/dev/null; then
				factor=$(($ram * ${2%?} / 100))
			elif expr "$2" : '[0-9][0-9]*$' >/dev/null; then
				factor=$2
			else # error
				factor=$min # do not enlarge the partition
			fi
			if [ $factor -lt $min ]; then
				factor=$min
			fi
			if [ "$3" = "-1" ] || \
			   expr "$3" : '[0-9][0-9]*$' >/dev/null; then
				max=$3
			elif expr "$3" : '[0-9][0-9]*+[0-9][0-9]*%$' >/dev/null; then
				ram_percent="${3#*+}"
				ram_percent="${ram_percent%?}"
				max=$((${3%%+*} + $ram * $ram_percent / 100))
			elif expr "$3" : '[0-9][0-9]*%$' >/dev/null; then
				max=$(($ram * ${3%?} / 100))
			else # error
				max=$min # do not enlarge the partition
			fi
			if [ $max -ne -1 ] && [ $max -lt $min ]; then
				max=$min
			fi
			case "$4" in # allow only valid file systems
			    btrfs|ext2|ext3|ext4|xfs|jfs|linux-swap|fat16|fat32|hfs)
				fs="$4"
				;;
			    \$default_filesystem)
				db_get partman/default_filesystem
				fs="$RET"
				;;
			    *)
				fs=ext2
				;;
			esac
			shift; shift; shift; shift
			line="$min $factor $max $fs $*"

			# Exclude partitions that have ...ignore set
			if [ "$ignore" ] && [ "$(echo $line | grep "$ignore")" ]; then
				line=
				continue
			fi

			# Exclude partitions that are only for a different
			# disk label.  The $PWD check avoids problems when
			# running from older versions of partman-auto-lvm,
			# where we weren't in a subdirectory of $DEVICES
			# while decoding the recipe; we preserve it in case
			# of custom code with the same problem.
			iflabel="$(echo $line | sed -n 's/.*\$iflabel{ \([^}]*\) }.*/\1/p')"
			if [ "$iflabel" ]; then
				if [ "${PWD#$DEVICES/}" = "$PWD" ]; then
					line=''
					continue
				fi

				open_dialog GET_LABEL_TYPE
				read_line label
				close_dialog
				# multiple space-separated disk labels are supported
				iflabel=" $iflabel "
				if [ -n "${iflabel%%* $label *}" ]; then
					line=''
					continue
				fi
			fi

			# Check if we can reuse an existing partition.
			if [ -n "$reuse" ] && echo "$line" | grep -q '\$reusemethod{'; then
				if [ "${PWD#$DEVICES/}" != "$PWD" ]; then
					method="$(echo "$line" | sed -n 's/.* method{ \([^}]*\) }.*/\1/p')"
					id="$(find_method "$method" "$min")"
					if [ "$id" ]; then
						line="$(echo "$line" | sed 's/\$reusemethod{[^}]*}/$reuse{ '"$id"' }/')"
						# strip format{ } from a reused partition
						line="$(echo "$line" | sed 's/format{[^}]*}//g')"
					fi
				fi
			fi

			scheme="${scheme:+$scheme$NL}$line"
			line=''
			;;
		    *)
			line="${line:+$line }$word"
			;;
		esac
	done
}

foreach_partition () {
	local - doing IFS pcount last partition
	doing=$1
	pcount=$(echo "$scheme" | wc -l)
	last=no

	IFS="$NL"
	for partition in $scheme; do
		restore_ifs
		[ $pcount -gt 1 ] || last=yes
		set -- $partition
		eval "$doing"
		pcount=$(($pcount - 1))
	done
}

min_size () {
	local size
	size=0
	foreach_partition '
		size=$(($size + $1))'
	echo $size
}

factor_sum () {
	local factor
	factor=0
	foreach_partition '
		factor=$(($factor + $2))'
	echo $factor
}

partition_before () {
	local num id size type fs path name result found
	result=''
	found=no
	open_dialog PARTITIONS
	while { read_line num id size type fs path name; [ "$id" ]; }; do
		if [ "$id" = "$1" ]; then
			found=yes
		fi
		if [ $found = no ]; then
			result=$id
		fi
	done
	close_dialog
	echo $result
}

partition_after () {
	local num id size type fs path name result found
	result=''
	found=no
	open_dialog PARTITIONS
	while { read_line num id size type fs path name; [ "$id" ]; }; do
		if [ $found = yes ] && [ -z "$result" ]; then
			result=$id
		fi
		if [ "$id" = "$1" ]; then
			found=yes
		fi
	done
	close_dialog
	echo $result
}

pull_primary () {
	primary=''
	scheme_rest=''
	foreach_partition '
		if [ -z "$primary" ] && \
		   echo $* | grep '\''\$primary{'\'' >/dev/null; then
			primary="$*"
		else
			scheme_rest="${scheme_rest:+$scheme_rest$NL}$*"
		fi'
}

setup_partition () {
	local id flags file line label
	id=$1; shift
	open_dialog GET_LABEL_TYPE
	read_line label
	close_dialog
	while [ "$1" ]; do
		case "$1" in
		    \$bootable{)
			while [ "$1" != '}' ] && [ "$1" ]; do
				shift
			done
			if [ "$label" != "gpt" ]; then
				open_dialog GET_FLAGS $id
				flags=$(read_paragraph)
				close_dialog
				open_dialog SET_FLAGS $id
				write_line "$flags"
				write_line boot
				write_line NO_MORE
				close_dialog
			fi
			;;
		    \$legacy_boot{)
			while [ "$1" != '}' ] && [ "$1" ]; do
				shift
			done
			if [ "$label" = "gpt" ]; then
				open_dialog GET_FLAGS $id
				flags=$(read_paragraph)
				close_dialog
				open_dialog SET_FLAGS $id
				write_line "$flags"
				write_line legacy_boot
				write_line NO_MORE
				close_dialog
			fi
			;;
		    \$default_filesystem{)
			while [ "$1" != '}' ] && [ "$1" ]; do
				shift
			done
			mkdir -p $id
			db_get partman/default_filesystem
			echo "$RET" >$id/filesystem
			;;
		    \$*{)
			while [ "$1" != '}' ] && [ "$1" ]; do
				shift
			done
			;;
		    *{)
			file=${1%?}
			mkdir -p $id
			case $file in
			    */*)
				mkdir -p $id/${file%/*}
				;;
			esac
			>$id/$file
			shift
			line=''
			while [ "$1" != '}' ] && [ "$1" ]; do
				if [ "$1" = ';' ]; then
					echo "$line" >>$id/$file
					line=''
				else
					line="${line:+$line }$1"
				fi
				shift
			done
			echo "$line" >>$id/$file
		esac
		shift
	done
	return 0
}

get_recipedir () {
	local archdetect arch sub recipedir

	if type archdetect >/dev/null 2>&1; then
		archdetect=$(archdetect)
	else
		archdetect=unknown/generic
	fi
	arch=${archdetect%/*}
	sub=${archdetect#*/}

	for recipedir in \
	    /usr/lib/partman/recipes-$arch-$sub \
	    /usr/lib/partman/recipes-$arch \
	    /usr/lib/partman/recipes; do
	if [ -d $recipedir ]; then
		echo $recipedir
		break
	fi
	done
}

filter_reused () {
	scheme_reused=$(
	    foreach_partition '
		if echo "$*" | grep -q '\''\$reuse{'\''; then
			echo "$*"
		fi'
	)
	scheme=$(
	    foreach_partition '
		if ! echo "$*" | grep -q '\''\$reuse{'\''; then
			echo "$*"
		fi'
	)
}

choose_recipe () {
	local recipes recipedir free_size choices min_size type target

	type=$1
	target="$2"
	free_size=$3
	local reuse="$4" # optional, boolean, empty=false

	# Preseeding of recipes
	db_get partman-auto/expert_recipe
	if [ -n "$RET" ]; then
		echo "$RET" > /tmp/expert_recipe
		db_set partman-auto/expert_recipe_file /tmp/expert_recipe
	fi
	db_get partman-auto/expert_recipe_file
	if [ ! -z "$RET" ] && [ -e "$RET" ]; then
		recipe="$RET"
		decode_recipe $recipe $type $reuse
		[ -n "$reuse" ] && filter_reused
		min_size=$(min_size)
		if [ $min_size -le $free_size ]; then
			return 0
		else
			logger -t partman-auto \
			"Available disk space ($free_size) too small for expert recipe ($min_size); skipping"
			hookdir=/usr/lib/partman/not-enough-space.d
			if [ -d $hookdir ] ; then
				for h in $hookdir/* ; do
					if [ -x $h ] ; then
						$h $recipe $free_size $min_size
					fi
				done
			fi
		fi
	fi

	recipedir=$(get_recipedir)

	choices=''
	default_recipe=no
	db_get partman-auto/choose_recipe
	old_default_recipe="$RET"
	for recipe in $recipedir/*; do
		[ -f "$recipe" ] || continue
		decode_recipe $recipe $type $reuse
		# Make sure the recipe contains lvmok tags if using LVM
		if [ "$type" = lvm ] && ! echo "$scheme" | grep -Fq lvmok; then
			continue
		fi
		[ -n "$reuse" ] && filter_reused
		if [ $(min_size) -le $free_size ]; then
			choices="${choices}${recipe}${TAB}${name}${NL}"
			if [ "$default_recipe" = no ]; then
				default_recipe="$recipe"
			fi
			if [ "$old_default_recipe" = "$name" ]; then
				default_recipe="$recipe"
			else
				local base="$(basename "$recipe")"
				if [ "$old_default_recipe" = "$base" ] || \
				   [ "$old_default_recipe" = "${base#[0-9][0-9]}" ]; then
					default_recipe="$recipe"
				fi
			fi
		fi
	done

	if [ -z "$choices" ]; then
		db_input critical partman-auto/no_recipe || true
		db_go || true # TODO handle backup right
		return 1
	fi

	db_subst partman-auto/choose_recipe TARGET "$target"
	debconf_select high partman-auto/choose_recipe \
		"$choices" "$default_recipe"
	if [ $? = 255 ]; then
		return 255
	fi
	recipe="$RET"
}

expand_scheme() {
	# Filter out reused partitions first, as we don't want to take
	# account of their size.
	filter_reused

	# Make factors small numbers so we can multiply on them.
	# Also ensure that fact, max and fs are valid
	# (Ofcourse in valid recipes they must be valid.)
	factsum=$(($(factor_sum) - $(min_size)))
	if [ $factsum -eq 0 ]; then
		factsum=100
	fi
	scheme=$(
	    foreach_partition '
		local min fact max fs
		min=$1
		fact=$((($2 - $min) * 100 / $factsum))
		max=$3
		fs=$4
		case "$fs" in
		    btrfs|ext2|ext3|ext4|xfs|jfs|linux-swap|fat16|fat32|hfs)
			true
			;;
		    *)
			fs=ext2
			;;
		esac
		shift; shift; shift; shift
		echo $min $fact $max $fs $*'
	)

	oldscheme=''
	while [ "$scheme" != "$oldscheme" ]; do
		oldscheme="$scheme"
		factsum=$(factor_sum)
		unallocated=$(($free_size - $(min_size)))
		if [ $unallocated -lt 0 ]; then
			unallocated=0
		fi
		scheme=$(
		    foreach_partition '
			local min fact max newmin
			min=$1
			fact=$2
			max=$3
			shift; shift; shift
			if [ $factsum -eq 0 ]; then
				newmin=$min
				if [ $fact -lt 0 ]; then
					fact=0
				fi
			else
				newmin=$(($min + $unallocated * $fact / $factsum))
			fi
			if [ $max -ne -1 ] && [ $newmin -gt $max ]; then
				echo $max 0 $max $*
			elif [ $newmin -lt $min ]; then
				echo $min 0 $min $*
			else
				echo $newmin $fact $max $*
			fi'
		)
	done
}

clean_method() {
	for device in $DEVICES/*; do
		[ -d "$device" ] || continue
		cd $device
		open_dialog PARTITIONS
		while { read_line num id size type fs path name; [ "$id" ]; }; do
			[ -e $id/method ] || continue
			mv $id/method $id/method-old
		done
		close_dialog
	done
}