FRPCWebUI/controller/StaticWebPageController.cpp

//
// Created by xcbosa on 2023/1/28.
//

#include "../processor/processor.h"
#include "../webuiconf.h"

#include "../processor/templates/framework7/Framework7Document.hpp"

#include <sys/stat.h>

using namespace std;
using namespace xc::processor;
using namespace xc::processor::templates;
using namespace xc::processor::templates::framework7;
using namespace configor;

namespace xc::controller {

    ContentGeneratorDefine({
        if (!conf::enableStaticAssetsController) { return false; }
        if (request.getMethod() != "GET") { return false; }
        if (request.getURL().length() == 0) { return false; }
        struct stat buffer;
        string filePath = "html";
        if (request.getURL()[0] == '/') {
            filePath += request.getURL();
        } else {
            filePath += "/" + request.getURL();
        }
        if (stat(filePath.c_str(), &buffer) == 0) {
            if (S_ISREG(buffer.st_mode)) {
                return true;
            } else {
                for (auto file: conf::defaultFiles) {
                    string newFilePath = filePath;
                    if (filePath[filePath.length() - 1] == '/') {
                        newFilePath += file;
                    } else {
                        newFilePath += "/" + file;
                    }
                    if (stat(newFilePath.c_str(), &buffer) == 0) {
                        return S_ISREG(buffer.st_mode);
                    }
                }
            }
        }
        return false;
        }, {
        struct stat buffer;
        string filePath = "html";
        if (request.getURL()[0] == '/') {
            filePath += request.getURL();
        } else {
            filePath += "/" + request.getURL();
        }
        if (stat(filePath.c_str(), &buffer) == 0) {
            if (S_ISREG(buffer.st_mode)) {
                return (ResponseData *) new BinaryResponseData(200, filePath,
                                                               mimeTypeOfFile(filePath));
            } else {
                for (auto file: conf::defaultFiles) {
                    string newFilePath = filePath;
                    if (filePath[filePath.length() - 1] == '/') {
                        newFilePath += file;
                    } else {
                        newFilePath += "/" + file;
                    }
                    if (stat(newFilePath.c_str(), &buffer) == 0) {
                        if (S_ISREG(buffer.st_mode)) {
                            return (ResponseData *) new BinaryResponseData(200, newFilePath, mimeTypeOfFile(newFilePath));
                        }
                    }
                }
            }
        }
        return (ResponseData *) new FileResponseData(conf::errorPage404);
    })

}