Add mediapipe::file::IsDirectory helper

PiperOrigin-RevId: 596637172
google-ai-edge · Jan 8, 2024 · 0b53972 · 0b53972
1 parent bdcf474
commit 0b53972
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 0 deletions.
diff --git a/mediapipe/framework/deps/BUILD b/mediapipe/framework/deps/BUILD
@@ -117,6 +117,7 @@ cc_library(
     deps = [
         ":file_path",
         "//mediapipe/framework/port:status",
+        "@com_google_absl//absl/status",
         "@com_google_absl//absl/strings",
     ],
 )

diff --git a/mediapipe/framework/deps/file_helpers.cc b/mediapipe/framework/deps/file_helpers.cc
@@ -28,7 +28,9 @@
 #include <sys/stat.h>
 
 #include <cerrno>
+#include <string>
 
+#include "absl/status/status.h"
 #include "mediapipe/framework/deps/file_path.h"
 #include "mediapipe/framework/port/canonical_errors.h"
 #include "mediapipe/framework/port/status.h"
@@ -251,6 +253,24 @@ absl::Status Exists(absl::string_view file_name) {
   }
 }
 
+absl::Status IsDirectory(absl::string_view file_name) {
+  struct stat buffer;
+  int status;
+  status = stat(std::string(file_name).c_str(), &buffer);
+  if (status == 0) {
+    if ((buffer.st_mode & S_IFMT) == S_IFDIR) {
+      return absl::OkStatus();
+    }
+    return absl::FailedPreconditionError("The path is not a directory.");
+  }
+  switch (errno) {
+    case EACCES:
+      return absl::PermissionDeniedError("Insufficient permissions.");
+    default:
+      return absl::NotFoundError("The path does not exist.");
+  }
+}
+
 #ifndef _WIN32
 int mkdir(std::string path) {
   return ::mkdir(path.c_str(), S_IRWXU | S_IRWXG | S_IRWXO);

diff --git a/mediapipe/framework/deps/file_helpers.h b/mediapipe/framework/deps/file_helpers.h
@@ -36,6 +36,8 @@ absl::Status MatchFileTypeInDirectory(const std::string& directory,
 
 absl::Status Exists(absl::string_view file_name);
 
+absl::Status IsDirectory(absl::string_view file_name);
+
 absl::Status RecursivelyCreateDir(absl::string_view path);
 
 }  // namespace file