From 21a4542d7ba4d20481b72cbfaadbaf24b47e43e3 Mon Sep 17 00:00:00 2001
From: tangyuxin <462747508@qq.com>
Date: Fri, 9 Oct 2020 11:54:42 +0800
Subject: [PATCH] [msh] rm command supports recursive deletion of folders

---
 components/finsh/msh_file.c | 117 ++++++++++++++++++++++++++++++++++--
 1 file changed, 113 insertions(+), 4 deletions(-)

diff --git a/components/finsh/msh_file.c b/components/finsh/msh_file.c
index 10025965f0..1368838b94 100644
--- a/components/finsh/msh_file.c
+++ b/components/finsh/msh_file.c
@@ -265,22 +265,131 @@ int cmd_cat(int argc, char **argv)
 }
 FINSH_FUNCTION_EXPORT_ALIAS(cmd_cat, __cmd_cat, Concatenate FILE(s));
 
+static void directory_delete_for_msh(const char *pathname, char f, char v)
+{
+    DIR *dir = NULL;
+    struct dirent *dirent = NULL;
+    char *full_path;
+
+    if (pathname == RT_NULL)
+        return;
+
+    full_path = (char *)rt_malloc(DFS_PATH_MAX);
+    if (full_path == RT_NULL)
+        return;
+
+    dir = opendir(pathname);
+    if (dir == RT_NULL)
+    {
+        if (f == 0)
+        {
+            rt_kprintf("cannot remove '%s'\n", pathname);
+        }
+        rt_free(full_path);
+        return;
+    }
+
+    while (1)
+    {
+        dirent = readdir(dir);
+        if (dirent == RT_NULL)
+            break;
+        if (rt_strcmp(".", dirent->d_name) != 0 &&
+            rt_strcmp("..", dirent->d_name) != 0)
+        {
+            rt_sprintf(full_path, "%s/%s", pathname, dirent->d_name);
+            if (dirent->d_type == DT_REG)
+            {
+                if (unlink(full_path) != 0)
+                {
+                    if (f == 0)
+                        rt_kprintf("cannot remove '%s'\n", full_path);
+                }
+                else if (v)
+                {
+                    rt_kprintf("removed '%s'\n", full_path);
+                }
+            }
+            else if (dirent->d_type == DT_DIR)
+            {
+                directory_delete_for_msh(full_path, f, v);
+            }
+        }
+    }
+    closedir(dir);
+    rt_free(full_path);
+    if (unlink(pathname) != 0)
+    {
+        if (f == 0)
+            rt_kprintf("cannot remove '%s'\n", pathname);
+    }
+    else if (v)
+    {
+        rt_kprintf("removed directory '%s'\n", pathname);
+    }
+}
+
 int cmd_rm(int argc, char **argv)
 {
-    int index;
+    int index, n;
+    char f = 0, r = 0, v = 0;
 
     if (argc == 1)
     {
-        rt_kprintf("Usage: rm FILE...\n");
+        rt_kprintf("Usage: rm option(s) FILE...\n");
         rt_kprintf("Remove (unlink) the FILE(s).\n");
         return 0;
     }
 
-    for (index = 1; index < argc; index ++)
+    if (argv[1][0] == '-')
     {
-        unlink(argv[index]);
+        for (n = 0; argv[1][n]; n++)
+        {
+            switch (argv[1][n])
+            {
+                case 'f': f = 1; break;
+                case 'r': r = 1; break;
+                case 'v': v = 1; break;
+                case '-': break;
+                default:
+                    rt_kprintf("Error: Bad option: %c\n", argv[1][n]);
+                    return 0;
+            }
+        }
+        argc -= 1;
+        argv = argv + 1;
     }
 
+    for (index = 1; index < argc; index ++)
+    {
+        struct stat s;
+        if (stat (argv[index], &s) == 0)
+        {
+            if (s.st_mode & S_IFDIR)
+            {
+                if (r == 0)
+                    rt_kprintf("cannot remove '%s': Is a directory\n", argv[index]);
+                else
+                    directory_delete_for_msh(argv[index], f, v);
+            }
+            else if (s.st_mode & S_IFREG)
+            {
+                if (unlink(argv[index]) != 0)
+                {
+                    if (f == 0)
+                        rt_kprintf("cannot remove '%s'\n", argv[index]);
+                }
+                else if (v)
+                {
+                    rt_kprintf("removed '%s'\n", argv[index]);
+                }
+            }
+        }
+        else if (f == 0)
+        {
+            rt_kprintf("cannot remove '%s': No such file or directory\n", argv[index]);
+        }
+    }
     return 0;
 }
 FINSH_FUNCTION_EXPORT_ALIAS(cmd_rm, __cmd_rm, Remove(unlink) the FILE(s).);