improve copy directory

tboox · Feb 6, 2024 · 548aa6b · 548aa6b
1 parent 517a8a5
commit 548aa6b
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/src/tbox/platform/windows/directory.c b/src/tbox/platform/windows/directory.c
@@ -62,6 +62,9 @@ static tb_long_t tb_directory_walk_copy(tb_char_t const* path, tb_file_info_t co
  tb_size_t size = tuple[1].ul;
  tb_char_t const* name = path + size;
 
+ // the copy flags
+ tb_size_t flags = tuple[2].ul;
+
  // the dest file path
  tb_char_t dpath[8192] = {0};
  tb_snprintf(dpath, 8192, "%s\\%s", dest, name[0] == '\\'? name + 1 : name);
@@ -77,10 +80,12 @@ static tb_long_t tb_directory_walk_copy(tb_char_t const* path, tb_file_info_t co
  }
 
  // copy
+ tb_bool_t ok = tb_true;
+ tb_bool_t skip_recursion = tb_false;
  switch (info->type)
  {
  case TB_FILE_TYPE_FILE:
- if (!tb_file_copy(path, dpath, TB_FILE_COPY_NONE)) tuple[2].b = tb_false;
+ ok = tb_file_copy(path, dpath, flags);
  break;
  case TB_FILE_TYPE_DIRECTORY:
  {
@@ -97,7 +102,11 @@ static tb_long_t tb_directory_walk_copy(tb_char_t const* path, tb_file_info_t co
  default:
  break;
  }
- return TB_DIRECTORY_WALK_CODE_CONTINUE;
+ tuple[3].b = ok;
+ tb_size_t retcode = TB_DIRECTORY_WALK_CODE_CONTINUE;
+ if (skip_recursion)
+ retcode |= TB_DIRECTORY_WALK_CODE_SKIP_RECURSION;
+ return retcode;
 }
 static tb_long_t tb_directory_walk_impl(tb_wchar_t const* path, tb_long_t recursion, tb_bool_t prefix, tb_directory_walk_func_t func, tb_cpointer_t priv)
 {
@@ -330,7 +339,8 @@ tb_bool_t tb_directory_copy(tb_char_t const* path, tb_char_t const* dest, tb_siz
  tb_value_t tuple[3];
  tuple[0].cstr = dest;
  tuple[1].ul = tb_strlen(path);
- tuple[2].b = tb_true;
+ tuple[2].ul = flags;
+ tuple[3].b = tb_true;
  tb_directory_walk(path, -1, tb_true, tb_directory_walk_copy, tuple);
 
  // ok?