libhv/http/server/FileCache.h

#ifndef HW_FILE_CACHE_H_
#define HW_FILE_CACHE_H_

#include <map>
#include <string>

#include "hbuf.h"
#include "hfile.h"
#include "hstring.h"
#include "hscope.h"
#include "hdir.h"

#include "md5.h"
#include "HttpRequest.h" // for get_content_type_str_by_suffix
#include "http_page.h"

#define HTTP_HEADER_MAX_LENGTH      1024 // 1k


typedef struct file_cache_s {
    //std::string filepath;
    struct stat st;
    time_t      open_time;
    time_t      stat_time;
    uint32_t    stat_cnt;
    HBuf        buf; // http_header + file_content
    hbuf_t      filebuf;
    hbuf_t      httpbuf;
    char        last_modified[64];
    char        etag[64];
    const char* content_type;

    file_cache_s() {
        stat_cnt = 0;
        content_type = NULL;
    }

    void resize_buf(int filesize) {
        buf.resize(HTTP_HEADER_MAX_LENGTH + filesize);
        filebuf.base = buf.base + HTTP_HEADER_MAX_LENGTH;
        filebuf.len = filesize;
    }

    void prepend_header(const char* header, int len) {
        if (len > HTTP_HEADER_MAX_LENGTH) return;
        httpbuf.base = filebuf.base - len;
        httpbuf.len = len + filebuf.len;
        memcpy(httpbuf.base, header, len);
    }
} file_cache_t;

// filepath => file_cache_t
typedef std::map<std::string, file_cache_t*> FileCacheMap;

#define DEFAULT_FILE_STAT_INTERVAL  10 // s
#define DEFAULT_FILE_CACHED_TIME    60 // s
class FileCache {
public:
    int file_stat_interval;
    int file_cached_time;
    FileCacheMap cached_files;

    FileCache() {
        file_stat_interval  = DEFAULT_FILE_STAT_INTERVAL;
        file_cached_time    = DEFAULT_FILE_CACHED_TIME;
    }

    ~FileCache() {
        for (auto& pair : cached_files) {
            delete pair.second;
        }
        cached_files.clear();
    }

    file_cache_t* Open(const char* filepath, void* ctx) {
        file_cache_t* fc = Get(filepath);
        bool modified = false;
        if (fc) {
            time_t tt;
            time(&tt);
            if (tt - fc->stat_time > file_stat_interval) {
                time_t mtime = fc->st.st_mtime;
                stat(filepath, &fc->st);
                fc->stat_time = tt;
                fc->stat_cnt++;
                if (mtime != fc->st.st_mtime) {
                    modified = true;
                    fc->stat_cnt = 1;
                }
            }
        }
        if (fc == NULL || modified) {
            int fd = open(filepath, O_RDONLY);
            if (fd < 0) {
                return NULL;
            }
            ScopeCleanup _(close, fd);
            if (fc == NULL) {
                struct stat st;
                fstat(fd, &st);
                if (S_ISREG(st.st_mode) ||
                    (S_ISDIR(st.st_mode) &&
                     filepath[strlen(filepath)-1] == '/')) {
                    fc = new file_cache_t;
                    //fc->filepath = filepath;
                    fc->st = st;
                    time(&fc->open_time);
                    fc->stat_time = fc->open_time;
                    fc->stat_cnt = 1;
                    cached_files[filepath] = fc;
                }
                else {
                    return NULL;
                }
            }
            if (S_ISREG(fc->st.st_mode)) {
                // FILE
                fc->resize_buf(fc->st.st_size);
                read(fd, fc->filebuf.base, fc->filebuf.len);
                const char* suffix = strrchr(filepath, '.');
                if (suffix) {
                    fc->content_type = http_content_type_str_by_suffix(++suffix);
                }
            }
            else if (S_ISDIR(fc->st.st_mode)) {
                // DIR
                std::string page;
                make_index_of_page(filepath, page, (const char*)ctx);
                fc->resize_buf(page.size());
                memcpy(fc->filebuf.base, page.c_str(), page.size());
                fc->content_type = http_content_type_str(TEXT_HTML);
            }
            time_t tt = fc->st.st_mtime;
            strftime(fc->last_modified, sizeof(fc->last_modified), "%a, %d %b %Y %H:%M:%S GMT", gmtime(&tt));
            MD5_CTX md5_ctx;
            MD5Init(&md5_ctx);
            MD5Update(&md5_ctx, (unsigned char*)fc->filebuf.base, fc->filebuf.len);
            unsigned char digital[16];
            MD5Final(digital, &md5_ctx);
            char* md5 = fc->etag;
            for (int i = 0; i < 16; ++i) {
                sprintf(md5, "%02x", digital[i]);
                md5 += 2;
            }
            fc->etag[32] = '\0';
        }
        return fc;
    }

    int Close(const char* filepath) {
        auto iter = cached_files.find(filepath);
        if (iter != cached_files.end()) {
            delete iter->second;
            iter = cached_files.erase(iter);
            return 0;
        }
        return -1;
    }

protected:
    file_cache_t* Get(const char* filepath) {
        auto iter = cached_files.find(filepath);
        if (iter != cached_files.end()) {
            return iter->second;
        }
        return NULL;
    }
};

#endif // HW_FILE_CACHE_H_