mirror of
https://github.com/RGBCube/serenity
synced 2025-07-25 13:37:45 +00:00
LibCore: Add utility class for temporary files and directories
Core::TempFile is a RAII type that creates a temporary file or directory on construction and deletes it on destruction.
This commit is contained in:
parent
28a6686f2c
commit
95545648bd
3 changed files with 97 additions and 0 deletions
59
Userland/Libraries/LibCore/TempFile.cpp
Normal file
59
Userland/Libraries/LibCore/TempFile.cpp
Normal file
|
@ -0,0 +1,59 @@
|
|||
/*
|
||||
* Copyright (c) 2020-2021, the SerenityOS developers.
|
||||
*
|
||||
* SPDX-License-Identifier: BSD-2-Clause
|
||||
*/
|
||||
|
||||
#include "TempFile.h"
|
||||
#include <AK/Random.h>
|
||||
#include <AK/Vector.h>
|
||||
#include <LibCore/File.h>
|
||||
#include <fcntl.h>
|
||||
#include <stdlib.h>
|
||||
#include <sys/stat.h>
|
||||
|
||||
namespace Core {
|
||||
|
||||
NonnullOwnPtr<TempFile> TempFile::create(Type type)
|
||||
{
|
||||
return adopt_own(*new TempFile(type));
|
||||
}
|
||||
|
||||
String TempFile::create_temp(Type type)
|
||||
{
|
||||
char name_template[] = "/tmp/tmp.XXXXXX";
|
||||
switch (type) {
|
||||
case Type::File: {
|
||||
auto fd = mkstemp(name_template);
|
||||
VERIFY(fd >= 0);
|
||||
close(fd);
|
||||
break;
|
||||
}
|
||||
case Type::Directory: {
|
||||
auto fd = mkdtemp(name_template);
|
||||
VERIFY(fd != nullptr);
|
||||
break;
|
||||
}
|
||||
}
|
||||
return String { name_template };
|
||||
}
|
||||
|
||||
TempFile::TempFile(Type type)
|
||||
: m_type(type)
|
||||
, m_path(create_temp(type))
|
||||
{
|
||||
}
|
||||
|
||||
TempFile::~TempFile()
|
||||
{
|
||||
File::RecursionMode recursion_allowed { File::RecursionMode::Disallowed };
|
||||
if (m_type == Type::Directory)
|
||||
recursion_allowed = File::RecursionMode::Allowed;
|
||||
|
||||
auto rc = File::remove(m_path.characters(), recursion_allowed, false);
|
||||
if (rc.is_error()) {
|
||||
warnln("File::remove failed: {}", rc.error().string_literal());
|
||||
}
|
||||
}
|
||||
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue