Ver Fonte

FileManager: Create FileUtils

This could later be replaced with library-provided utility functions
when/if they exist.
Aaron Malpas há 5 anos atrás
pai
commit
7841aadb41

+ 133 - 0
Applications/FileManager/FileUtils.cpp

@@ -0,0 +1,133 @@
+#include "FileUtils.h"
+#include <AK/FileSystemPath.h>
+#include <AK/StringBuilder.h>
+#include <LibCore/CDirIterator.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <sys/stat.h>
+#include <unistd.h>
+
+namespace FileUtils {
+
+bool copy_file_or_directory(const String& src_path, const String& dst_path)
+{
+    int duplicate_count = 0;
+    while (access(get_duplicate_name(dst_path, duplicate_count).characters(), F_OK) == 0) {
+        ++duplicate_count;
+    }
+    if (duplicate_count != 0) {
+        return copy_file_or_directory(src_path, get_duplicate_name(dst_path, duplicate_count));
+    }
+
+    int src_fd = open(src_path.characters(), O_RDONLY);
+    if (src_fd < 0) {
+        return false;
+    }
+
+    struct stat src_stat;
+    int rc = fstat(src_fd, &src_stat);
+    if (rc < 0) {
+        return false;
+    }
+
+    if (S_ISDIR(src_stat.st_mode)) {
+        return copy_directory(src_path, dst_path);
+    }
+    return copy_file(src_path, dst_path, src_stat, src_fd);
+}
+
+bool copy_directory(const String& src_path, const String& dst_path)
+{
+    int rc = mkdir(dst_path.characters(), 0755);
+    if (rc < 0) {
+        return false;
+    }
+    CDirIterator di(src_path, CDirIterator::SkipDots);
+    if (di.has_error()) {
+        return false;
+    }
+    while (di.has_next()) {
+        String filename = di.next_path();
+        bool is_copied = copy_file_or_directory(
+            String::format("%s/%s", src_path.characters(), filename.characters()),
+            String::format("%s/%s", dst_path.characters(), filename.characters()));
+        if (!is_copied) {
+            return false;
+        }
+    }
+    return true;
+}
+
+bool copy_file(const String& src_path, const String& dst_path, const struct stat& src_stat, int src_fd)
+{
+    int dst_fd = creat(dst_path.characters(), 0666);
+    if (dst_fd < 0) {
+        if (errno != EISDIR) {
+            return false;
+        }
+        StringBuilder builder;
+        builder.appendf("%s/%s", dst_path, FileSystemPath(src_path).basename());
+        String dst_path = builder.to_string();
+        dst_fd = creat(dst_path.characters(), 0666);
+        if (dst_fd < 0) {
+            return false;
+        }
+    }
+
+    for (;;) {
+        char buffer[BUFSIZ];
+        ssize_t nread = read(src_fd, buffer, sizeof(buffer));
+        if (nread < 0) {
+            return false;
+        }
+        if (nread == 0)
+            break;
+        ssize_t remaining_to_write = nread;
+        char* bufptr = buffer;
+        while (remaining_to_write) {
+            ssize_t nwritten = write(dst_fd, bufptr, remaining_to_write);
+            if (nwritten < 0) {
+                return false;
+            }
+            assert(nwritten > 0);
+            remaining_to_write -= nwritten;
+            bufptr += nwritten;
+        }
+    }
+
+    auto my_umask = umask(0);
+    umask(my_umask);
+    int rc = fchmod(dst_fd, src_stat.st_mode & ~my_umask);
+    if (rc < 0) {
+        return false;
+    }
+
+    close(src_fd);
+    close(dst_fd);
+    return true;
+}
+
+String get_duplicate_name(const String& path, int duplicate_count)
+{
+    if (duplicate_count == 0) {
+        return path;
+    }
+    FileSystemPath fsp(path);
+    StringBuilder duplicated_name;
+    duplicated_name.append('/');
+    for (int i = 0; i < fsp.parts().size() - 1; ++i) {
+        duplicated_name.appendf("%s/", fsp.parts()[i].characters());
+    }
+    auto prev_duplicate_tag = String::format("(%d)", duplicate_count);
+    auto title = fsp.title();
+    if (title.ends_with(prev_duplicate_tag)) {
+        // remove the previous duplicate tag "(n)" so we can add a new tag.
+        title = title.substring(0, title.length() - prev_duplicate_tag.length());
+    }
+    duplicated_name.appendf("%s (%d)", fsp.title().characters(), duplicate_count);
+    if (!fsp.extension().is_empty()) {
+        duplicated_name.appendf(".%s", fsp.extension().characters());
+    }
+    return duplicated_name.build();
+}
+}

+ 10 - 0
Applications/FileManager/FileUtils.h

@@ -0,0 +1,10 @@
+#pragma once
+#include <AK/String.h>
+
+namespace FileUtils {
+bool copy_file_or_directory(const String& src_path, const String& dst_path);
+
+String get_duplicate_name(const String& path, int duplicate_count);
+bool copy_file(const String& src_path, const String& dst_path, const struct stat& src_stat, int src_fd);
+bool copy_directory(const String& src_path, const String& dst_path);
+}

+ 1 - 0
Applications/FileManager/Makefile

@@ -2,6 +2,7 @@ include ../../Makefile.common
 
 OBJS = \
     DirectoryView.o \
+    FileUtils.o \
     main.o
 
 APP = FileManager