From 375825c1e8f873aaa1cb7e66518ffc0507aa6df0 Mon Sep 17 00:00:00 2001
From: oddlama <oddlama@oddlama.org>
Date: Thu, 29 Oct 2020 01:19:45 +0100
Subject: [PATCH] Fix wrong partition types

---
 scripts/functions.sh       |  2 +-
 scripts/internal_config.sh | 11 ++++++++---
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/scripts/functions.sh b/scripts/functions.sh
index 9c82bdf..fc097a7 100644
--- a/scripts/functions.sh
+++ b/scripts/functions.sh
@@ -387,7 +387,7 @@ disk_format_btrfs() {
 
 	# Collect extra arguments
 	extra_args=()
-	if [[ -v "arguments[raid_type]" ]]; then
+	if [[ "${#devices}" -gt 1 ]] && [[ -v "arguments[raid_type]" ]]; then
 		extra_args+=("-d" "$raid_type")
 	fi
 
diff --git a/scripts/internal_config.sh b/scripts/internal_config.sh
index 444cc13..2abca09 100644
--- a/scripts/internal_config.sh
+++ b/scripts/internal_config.sh
@@ -216,7 +216,12 @@ format() {
 	declare -A arguments; parse_arguments "$@"
 
 	verify_existing_id id
-	verify_option type bios efi swap ext4
+	verify_option type bios efi swap ext4 btrfs
+
+	local type="${arguments[type]}"
+	if [[ "$type" == "btrfs" ]]; then
+		USED_BTRFS=true
+	fi
 
 	DISK_ACTIONS+=("action=format" "$@" ";")
 }
@@ -399,8 +404,8 @@ create_btrfs_raid_layout() {
 	create_gpt new_id="gpt_dev0" device="${extra_arguments[0]}"
 	create_partition new_id="part_${type}_dev0" id="gpt_dev0" size=256MiB       type="$type"
 	[[ $size_swap != "false" ]] && \
-	create_partition new_id="part_swap_dev0"    id="gpt_dev0" size="$size_swap" type=raid
-	create_partition new_id="part_root_dev0"    id="gpt_dev0" size=remaining    type=raid
+	create_partition new_id="part_swap_dev0"    id="gpt_dev0" size="$size_swap" type=swap
+	create_partition new_id="part_root_dev0"    id="gpt_dev0" size=remaining    type=linux
 
 	local root_id
 	local root_ids=""