fdisk: use libfdisk to ask for partition type

Signed-off-by: Karel Zak <kzak@redhat.com>

1 files changed, 27 insertions, 0 deletions

diff --git a/fdisks/fdisk-ask.c b/fdisks/fdisk-ask.c
index 0c4e200f..a7e86e1c 100644
--- a/fdisks/fdisk-ask.c
+++ b/fdisks/fdisk-ask.c
@@ -220,3 +220,30 @@ int ask_callback(struct fdisk_context *cxt, struct fdisk_ask *ask,
 	}
 	return rc;
 }
+
+
+struct fdisk_parttype *ask_partition_type(struct fdisk_context *cxt)
+{
+	const char *q;
+
+	if (!cxt || !cxt->label || !cxt->label->nparttypes)
+		return NULL;
+
+        q = fdisk_is_parttype_string(cxt) ?
+		_("Partition type (type L to list all types): ") :
+		_("Hex code (type L to list all codes): ");
+	do {
+		char buf[256];
+		int rc = get_user_reply(cxt, q, buf, sizeof(buf));
+
+		if (rc)
+			break;
+
+		if (buf[1] == '\0' && toupper(*buf) == 'L')
+			list_partition_types(cxt);
+		else if (*buf)
+			return fdisk_parse_parttype(cxt, buf);
+        } while (1);
+
+	return NULL;
+}