VileFile/model.cpp at main · sargonkadnezar/VileFile · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
#include "model.hpp"
#include <ctime>
#include <filesystem>
#include <iomanip>
#include <sstream>

namespace fs = std::filesystem;

std::string permsToString(fs::perms p) {
  auto r = [](fs::perms m) -> char { return (m != fs::perms::none) ? 'r' : '-'; };
  auto w = [](fs::perms m) -> char { return (m != fs::perms::none) ? 'w' : '-'; };
  auto x = [](fs::perms m) -> char { return (m != fs::perms::none) ? 'x' : '-'; };
  std::string s;
  s += r(p & fs::perms::owner_read);
  s += w(p & fs::perms::owner_write);
  s += x(p & fs::perms::owner_exec);
  s += r(p & fs::perms::group_read);
  s += w(p & fs::perms::group_write);
  s += x(p & fs::perms::group_exec);
  s += r(p & fs::perms::others_read);
  s += w(p & fs::perms::others_write);
  s += x(p & fs::perms::others_exec);
  return s;
}

/// Convert file_time_type to ISO datetime string
std::string getTimeString(fs::file_time_type ftime) {
  using namespace std::chrono;
  auto sysTime = time_point_cast<system_clock::duration>(
      ftime - fs::file_time_type::clock::now() + system_clock::now());
  std::time_t tt = system_clock::to_time_t(sysTime);
  std::ostringstream oss;
  oss << std::put_time(std::localtime(&tt), "%Y-%m-%d %H:%M");
  return oss.str();
}
std::vector<FileData>
FileSystemModel::getDirectoryContents(const std::string &path) {
  std::vector<FileData> result;
  try {
    for (const auto &entry : fs::directory_iterator(
             path, fs::directory_options::skip_permission_denied)) {
      std::error_code ec;
      bool isDir = entry.is_directory(ec);
      bool isSym = entry.is_symlink(ec);
      if (ec || isSym || (!isDir && !entry.is_regular_file(ec)))
        continue;

      uintmax_t size = isDir ? 0 : entry.file_size(ec);
      if (ec)
        size = 0;

      std::string dateStr;
      if (auto ftime = entry.last_write_time(ec); !ec)
        dateStr = getTimeString(ftime);

      std::string permStr = permsToString(entry.status(ec).permissions());

      result.push_back({entry.path().filename().string(), entry.path().string(),
                        isDir, size, dateStr, permStr});
    }
  } catch (const fs::filesystem_error &) {
  }
  return result;
}
std::vector<std::string>
FileSystemModel::getSubDirectories(const std::string &path) {
  std::vector<std::string> out;
  for (const auto &fd : getDirectoryContents(path))
    if (fd.isDirectory)
      out.push_back(fd.fullPath);
  return out;
}