kakoune/src/file.cc

#include "file.hh"

#include "buffer.hh"
#include "buffer_manager.hh"
#include "assert.hh"

#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <cstring>


namespace Kakoune
{

bool isidentifier(char c)
{
    return std::isalnum(c) or c == '_';
}

std::string parse_filename(const std::string& filename)
{
    if (filename.length() > 2 and filename[0] == '~' and filename[1] == '/')
        return parse_filename("$HOME/" + filename.substr(2));

    size_t pos = 0;
    std::string result;
    for (size_t i = 0; i < filename.length(); ++i)
    {
        if (filename[i] == '$' and (i == 0 or filename[i-1] != '\\'))
        {
            result += filename.substr(pos, i - pos);
            size_t end = i+1;
            while (end != filename.length() and isidentifier(filename[end]))
                ++end;
            std::string var_name = filename.substr(i+1, end - i - 1);
            const char* var_value = getenv(var_name.c_str());
            if (var_value)
                result += var_value;

            pos = end;
        }
    }
    if (pos != filename.length())
        result += filename.substr(pos);

    return result;
}

std::string read_file(const std::string& filename)
{
   int fd = open(filename.c_str(), O_RDONLY);
    if (fd == -1)
    {
        if (errno == ENOENT)
            throw file_not_found(filename);

        throw file_access_error(filename, strerror(errno));
    }

    std::string content;
    char buf[256];
    while (true)
    {
        ssize_t size = read(fd, buf, 256);
        if (size == -1 or size == 0)
            break;

        content += std::string(buf, size);
    }
    close(fd);
    return content;
}

Buffer* create_buffer_from_file(const std::string& filename)
{
    std::string content = read_file(filename);

    if (Buffer* buffer = BufferManager::instance().get_buffer(filename))
        delete buffer;

    return new Buffer(filename, Buffer::Type::File, content);
}

void write_buffer_to_file(const Buffer& buffer, const std::string& filename)
{
    int fd = open(filename.c_str(), O_CREAT | O_WRONLY | O_TRUNC, 0644);
    if (fd == -1)
        throw file_access_error(filename, strerror(errno));

    for (size_t i = 0; i < buffer.line_count(); ++i)
    {
        const String& content = buffer.line_content(i);
        ssize_t count = content.length() * sizeof(BufferChar);
        const char* ptr = content.c_str();

        while (count)
        {
            ssize_t written = write(fd, ptr, count);
            ptr += written;
            count -= written;

            if (written == -1)
                throw file_access_error(filename, strerror(errno));
        }
    }
    close(fd);
}

}