--- libaitcli/src/cli.c	2013/10/08 10:45:08	1.6.4.4
+++ libaitcli/src/cli.c	2022/12/05 22:23:38	1.12
@@ -3,7 +3,7 @@
 *  by Michael Pounov <misho@openbsd-bg.org>
 *
 * $Author: misho $
-* $Id: cli.c,v 1.6.4.4 2013/10/08 10:45:08 misho Exp $
+* $Id: cli.c,v 1.12 2022/12/05 22:23:38 misho Exp $
 *
 **************************************************************************
 The ELWIX and AITNET software is distributed under the following
@@ -12,7 +12,7 @@ terms:
 All of the documentation and software included in the ELWIX and AITNET
 Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>
 
-Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012, 2013
+Copyright 2004 - 2022
 	by Michael Pounov <misho@elwix.org>.  All rights reserved.
 
 Redistribution and use in source and binary forms, with or without
@@ -58,6 +58,7 @@ int
 cli_Cmd_Exit(void * __restrict cli_buffer, int idx, char ** __restrict args)
 {
 	cli_Printf(cli_buffer, "\n");
+	((linebuffer_t*) cli_buffer)->line_kill = 42;
 	return RETCODE_EOF;
 }
 
@@ -83,7 +84,7 @@ cli_Cmd_Help(void * __restrict cli_buffer, int idx, ch
 	cli_Printf(buf, "\n");
 	if (!args) {
 		SLIST_FOREACH(cmd, &buf->line_cmds, cmd_next) {
-			if (cmd->cmd_level != idx)
+			if (!(cmd->cmd_level & (1 << idx)))
 				continue;
 
 			cli_Printf(buf, "%s\t\t%s\n", cmd->cmd_name, cmd->cmd_help);
@@ -96,9 +97,10 @@ cli_Cmd_Help(void * __restrict cli_buffer, int idx, ch
 				return RETCODE_OK;
 
 		SLIST_FOREACH(cmd, &buf->line_cmds, cmd_next) {
-			if (cmd->cmd_level != idx)
+			if (!(cmd->cmd_level & (1 << idx)))
 				continue;
-			if (args[1] && (cmd->cmd_level != idx || strcmp(args[1], cmd->cmd_name)))
+			if (args[1] && (!(cmd->cmd_level & (1 << idx)) || 
+						strcmp(args[1], cmd->cmd_name)))
 				continue;
 
 			cli_Printf(buf, "%s%s\t\t%s\n", args[1] ? "Syntax::\n\t" : "", cmd->cmd_name, 
@@ -137,11 +139,15 @@ int
 cli_Cmd_Config(void * __restrict cli_buffer, int idx, char ** __restrict args)
 {
 	linebuffer_t *buf = cli_buffer;
+	char szPrompt[STRSIZ + 16] = {[0 ... STRSIZ + 15] = 0};
 
-	if (idx == sizeof idx * 8)
+	if (idx == sizeof idx * 8 - 1)
 		return RETCODE_ERR;
 
 	buf->line_level++;
+	snprintf(szPrompt, sizeof szPrompt, "%s{%d}> ", buf->line_porigin, buf->line_level);
+	cli_setPrompt(buf, szPrompt);
+	cli_Printf(buf, "Enter to config level %d\n", buf->line_level);
 	return RETCODE_OK;
 }
 
@@ -157,11 +163,15 @@ int
 cli_Cmd_End(void * __restrict cli_buffer, int idx, char ** __restrict args)
 {
 	linebuffer_t *buf = cli_buffer;
+	char szPrompt[STRSIZ + 16] = {[0 ... STRSIZ + 15] = 0};
 
 	if (!idx)
-		cli_Cmd_Exit(cli_buffer, idx, args);
+		return cli_Cmd_Exit(cli_buffer, idx, args);
 
 	buf->line_level--;
+	snprintf(szPrompt, sizeof szPrompt, "%s{%d}> ", buf->line_porigin, buf->line_level);
+	cli_setPrompt(buf, szPrompt);
+	cli_Printf(buf, "Enter to config level %d\n", buf->line_level);
 	return RETCODE_OK;
 }
 
@@ -177,7 +187,28 @@ int
 cli_Cmd_Top(void * __restrict cli_buffer, int idx, char ** __restrict args)
 {
 	linebuffer_t *buf = cli_buffer;
+	char szPrompt[STRSIZ + 16] = {[0 ... STRSIZ + 15] = 0};
 
 	buf->line_level ^= buf->line_level;
+	snprintf(szPrompt, sizeof szPrompt, "%s{%d}> ", buf->line_porigin, buf->line_level);
+	cli_setPrompt(buf, szPrompt);
+	cli_Printf(buf, "Enter to top config level %d\n", buf->line_level);
+	return RETCODE_OK;
+}
+
+/*
+ * cli_Cmd_WhereAmI() - Builtin helper function for query level of Cli
+ *
+ * @cli_buffer = CLI buffer
+ * @idx = Config level
+ * @args = Parsed arguments array
+ * return: RETCODE_OK ok
+*/
+int
+cli_Cmd_WhereAmI(void * __restrict cli_buffer, int idx, char ** __restrict args)
+{
+	linebuffer_t *buf = cli_buffer;
+
+	cli_Printf(buf, "Current config level is %d\n", buf->line_level);
 	return RETCODE_OK;
 }