mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2025-01-08 20:32:56 +03:00
FileManager: Create FileUtils
This could later be replaced with library-provided utility functions when/if they exist.
This commit is contained in:
parent
b894803a30
commit
7841aadb41
Notes:
sideshowbarker
2024-07-19 12:10:10 +09:00
Author: https://github.com/malpas Commit: https://github.com/SerenityOS/serenity/commit/7841aadb41b Pull-request: https://github.com/SerenityOS/serenity/pull/539 Reviewed-by: https://github.com/awesomekling
133
Applications/FileManager/FileUtils.cpp
Normal file
133
Applications/FileManager/FileUtils.cpp
Normal file
@ -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
Applications/FileManager/FileUtils.h
Normal file
10
Applications/FileManager/FileUtils.h
Normal file
@ -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);
|
||||
}
|
@ -2,6 +2,7 @@ include ../../Makefile.common
|
||||
|
||||
OBJS = \
|
||||
DirectoryView.o \
|
||||
FileUtils.o \
|
||||
main.o
|
||||
|
||||
APP = FileManager
|
||||
|
Loading…
Reference in New Issue
Block a user