xref: /openbmc/bmcweb/include/webassets.hpp (revision 4859bdbafbb2b78ae414fb050ad197db2f2cb28b)
1 #pragma once
2 
3 #include <experimental/filesystem>
4 #include <fstream>
5 #include <string>
6 #include <crow/app.h>
7 #include <crow/http_request.h>
8 #include <crow/http_response.h>
9 #include <crow/routing.h>
10 #include <boost/algorithm/string/replace.hpp>
11 #include <boost/container/flat_set.hpp>
12 
13 namespace crow {
14 namespace webassets {
15 
16 namespace filesystem = std::experimental::filesystem;
17 
18 struct CmpStr {
19   bool operator()(const char* a, const char* b) const {
20     return std::strcmp(a, b) < 0;
21   }
22 };
23 
24 static boost::container::flat_set<std::string> routes;
25 
26 template <typename... Middlewares>
27 void requestRoutes(Crow<Middlewares...>& app) {
28   const static boost::container::flat_map<const char*, const char*, CmpStr>
29       contentTypes{
30           {{".css", "text/css;charset=UTF-8"},
31            {".html", "text/html;charset=UTF-8"},
32            {".js", "text/html;charset=UTF-8"},
33            {".png", "image/png;charset=UTF-8"},
34            {".woff", "application/x-font-woff"},
35            {".woff2", "application/x-font-woff2"},
36            {".gif", "image/gif"},
37            {".ico", "image/x-icon"},
38            {".ttf", "application/x-font-ttf"},
39            {".svg", "image/svg+xml"},
40            {".eot", "application/vnd.ms-fontobject"},
41            {".xml", "application/xml"},
42            {".jpg", "image/jpeg"},
43            {".jpeg", "image/jpeg"},
44            // dev tools don't care about map type, setting to json causes
45            // browser to show as text
46            // https://stackoverflow.com/questions/19911929/what-mime-type-should-i-use-for-javascript-source-map-files
47            {".map", "application/json"}}};
48   filesystem::path rootpath{"/usr/share/www/"};
49   filesystem::recursive_directory_iterator dirIter(rootpath);
50 
51   for (const filesystem::directory_entry& dir : dirIter) {
52     filesystem::path absolutePath = dir.path();
53     filesystem::path relativePath{
54         absolutePath.string().substr(rootpath.string().size() - 1)};
55     // make sure we don't recurse into certain directories
56     // note: maybe check for is_directory() here as well...
57 
58     if (filesystem::is_directory(dir)) {
59       // don't recurse into hidden directories or symlinks
60       if (boost::starts_with(dir.path().filename().string(), ".") ||
61           filesystem::is_symlink(dir)) {
62         dirIter.disable_recursion_pending();
63       }
64     } else if (filesystem::is_regular_file(dir)) {
65       std::string extension = relativePath.extension();
66       filesystem::path webpath = relativePath;
67       const char* contentEncoding = nullptr;
68 
69       if (extension == ".gz") {
70         webpath = webpath.replace_extension("");
71         // Use the non-gzip version for determining content type
72         extension = webpath.extension().string();
73         contentEncoding = "gzip";
74       }
75 
76       if (boost::starts_with(webpath.filename().string(), "index.")) {
77         webpath = webpath.parent_path();
78         if (webpath.string().size() == 0 || webpath.string().back() != '/') {
79           // insert the non-directory version of this path
80           routes.insert(webpath);
81           webpath += "/";
82         }
83       }
84 
85       routes.insert(webpath);
86       const char* contentType = nullptr;
87 
88       auto contentTypeIt = contentTypes.find(extension.c_str());
89       if (contentTypeIt == contentTypes.end()) {
90         BMCWEB_LOG_ERROR << "Cannot determine content-type for " << absolutePath
91                          << " with extension " << extension;
92       } else {
93         contentType = contentTypeIt->second;
94       }
95 
96       app.routeDynamic(webpath)(
97           [absolutePath, contentType, contentEncoding](const crow::Request& req,
98                                                        crow::Response& res) {
99             if (contentType != nullptr) {
100               res.addHeader("Content-Type", contentType);
101             }
102 
103             if (contentEncoding != nullptr) {
104               res.addHeader("Content-Encoding", contentEncoding);
105             }
106 
107             // res.set_header("Cache-Control", "public, max-age=86400");
108             std::ifstream inf(absolutePath);
109             if (!inf) {
110               BMCWEB_LOG_DEBUG << "failed to read file";
111               res.result(boost::beast::http::status::internal_server_error);
112               res.end();
113               return;
114             }
115 
116             res.body() = {std::istreambuf_iterator<char>(inf),
117                           std::istreambuf_iterator<char>()};
118             res.end();
119           });
120     }
121   }
122 }  // namespace webassets
123 }  // namespace webassets
124 }  // namespace crow
125