FileIconProvider.cpp 4.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151
  1. /*
  2. * Copyright (c) 2020, Andreas Kling <kling@serenityos.org>
  3. * All rights reserved.
  4. *
  5. * Redistribution and use in source and binary forms, with or without
  6. * modification, are permitted provided that the following conditions are met:
  7. *
  8. * 1. Redistributions of source code must retain the above copyright notice, this
  9. * list of conditions and the following disclaimer.
  10. *
  11. * 2. Redistributions in binary form must reproduce the above copyright notice,
  12. * this list of conditions and the following disclaimer in the documentation
  13. * and/or other materials provided with the distribution.
  14. *
  15. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  16. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  17. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  18. * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
  19. * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  20. * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  21. * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
  22. * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  23. * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  24. * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25. */
  26. #include <AK/String.h>
  27. #include <LibGUI/FileIconProvider.h>
  28. #include <LibGUI/Icon.h>
  29. #include <LibGfx/Bitmap.h>
  30. #include <sys/stat.h>
  31. namespace GUI {
  32. #define ENUMERATE_FILETYPES(F) \
  33. F(cplusplus, ".cpp") \
  34. F(form, ".frm") \
  35. F(hackstudio, ".hsp") \
  36. F(header, ".h") \
  37. F(html, ".html") \
  38. F(image, ".png") \
  39. F(java, ".java") \
  40. F(javascript, ".js") \
  41. F(library, ".so", ".a") \
  42. F(markdown, ".md") \
  43. F(object, ".o", ".obj") \
  44. F(pdf, ".pdf") \
  45. F(python, ".py") \
  46. F(sound, ".wav") \
  47. F(ini, ".ini") \
  48. F(text, ".txt") \
  49. F(spreadsheet, ".sheets")
  50. #define __ENUMERATE_FILETYPE(filetype_name, ...) \
  51. static Icon s_filetype_##filetype_name##_icon;
  52. ENUMERATE_FILETYPES(__ENUMERATE_FILETYPE)
  53. #undef __ENUMERATE_FILETYPE
  54. static Icon s_directory_icon;
  55. static Icon s_directory_open_icon;
  56. static Icon s_home_directory_icon;
  57. static Icon s_home_directory_open_icon;
  58. static Icon s_file_icon;
  59. static Icon s_symlink_icon;
  60. static Icon s_socket_icon;
  61. static Icon s_executable_icon;
  62. static void initialize_if_needed()
  63. {
  64. static bool s_initialized = false;
  65. if (s_initialized)
  66. return;
  67. s_directory_icon = Icon::default_icon("filetype-folder");
  68. s_directory_open_icon = Icon::default_icon("filetype-folder-open");
  69. s_home_directory_icon = Icon::default_icon("home-directory");
  70. s_home_directory_open_icon = Icon::default_icon("home-directory-open");
  71. s_file_icon = Icon::default_icon("filetype-unknown");
  72. s_symlink_icon = Icon::default_icon("filetype-symlink");
  73. s_socket_icon = Icon::default_icon("filetype-socket");
  74. s_executable_icon = Icon::default_icon("filetype-executable");
  75. #define __ENUMERATE_FILETYPE(filetype_name, ...) \
  76. s_filetype_##filetype_name##_icon = Icon::default_icon("filetype-" #filetype_name);
  77. ENUMERATE_FILETYPES(__ENUMERATE_FILETYPE)
  78. #undef __ENUMERATE_FILETYPE
  79. s_initialized = true;
  80. }
  81. Icon FileIconProvider::directory_icon()
  82. {
  83. initialize_if_needed();
  84. return s_directory_icon;
  85. }
  86. Icon FileIconProvider::directory_open_icon()
  87. {
  88. initialize_if_needed();
  89. return s_directory_open_icon;
  90. }
  91. Icon FileIconProvider::home_directory_icon()
  92. {
  93. initialize_if_needed();
  94. return s_home_directory_icon;
  95. }
  96. Icon FileIconProvider::home_directory_open_icon()
  97. {
  98. initialize_if_needed();
  99. return s_home_directory_open_icon;
  100. }
  101. Icon FileIconProvider::filetype_image_icon()
  102. {
  103. initialize_if_needed();
  104. return s_filetype_image_icon;
  105. }
  106. Icon FileIconProvider::icon_for_path(const String& path)
  107. {
  108. struct stat stat;
  109. if (::stat(path.characters(), &stat) < 0)
  110. return {};
  111. return icon_for_path(path, stat.st_mode);
  112. }
  113. Icon FileIconProvider::icon_for_path(const String& path, mode_t mode)
  114. {
  115. initialize_if_needed();
  116. if (S_ISDIR(mode))
  117. return s_directory_icon;
  118. if (S_ISLNK(mode))
  119. return s_symlink_icon;
  120. if (S_ISSOCK(mode))
  121. return s_socket_icon;
  122. if (mode & (S_IXUSR | S_IXGRP | S_IXOTH))
  123. return s_executable_icon;
  124. #define __ENUMERATE_FILETYPE(filetype_name, filetype_extensions...) \
  125. for (auto& extension : { filetype_extensions }) { \
  126. if (path.ends_with(extension, CaseSensitivity::CaseInsensitive)) \
  127. return s_filetype_##filetype_name##_icon; \
  128. }
  129. ENUMERATE_FILETYPES(__ENUMERATE_FILETYPE)
  130. #undef __ENUMERATE_FILETYPE
  131. return s_file_icon;
  132. }
  133. }