LjTools/LjasFileCache.cpp at master · rochus-keller/LjTools · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
/*
* Copyright 2019 Rochus Keller <mailto:me@rochus-keller.ch>
*
* This file is part of the LjAsm parser library.
*
* The following is the license that applies to this copy of the
* library. For a license to use the library under conditions
* other than those described here, please email to me@rochus-keller.ch.
*
* GNU General Public License Usage
* This file may be used under the terms of the GNU General Public
* License (GPL) versions 2.0 or 3.0 as published by the Free Software
* Foundation and appearing in the file LICENSE.GPL included in
* the packaging of this file. Please review the following information
* to ensure GNU General Public Licensing requirements will be met:
* http://www.fsf.org/licensing/licenses/info/GPLv2.html and
* http://www.gnu.org/copyleft/gpl.html.
*/

#include "LjasFileCache.h"
#include <QFile>
#include <QBuffer>
#include <QFileInfo>
using namespace Ljas;

FileCache::FileCache(QObject *parent) : QObject(parent)
{
}

#define _USE_CANONOCALS

void FileCache::addFile(const QString& path, const QByteArray& content)
{
#ifdef _USE_CANONOCALS
    const QString cpath = QFileInfo(path).canonicalFilePath();
#else
    const QString cpath = path;
#endif
    d_lock.lockForWrite();
    d_files[cpath] = content;
    d_lock.unlock();
}

void FileCache::removeFile(const QString& path)
{
#ifdef _USE_CANONOCALS
    const QString cpath = QFileInfo(path).canonicalFilePath();
#else
    const QString cpath = path;
#endif
    d_lock.lockForWrite();
    d_files.remove(cpath);
    d_lock.unlock();
}

QByteArray FileCache::getFile(const QString& path, bool* found) const
{
    QByteArray res;
#ifdef _USE_CANONOCALS
    const QString cpath = QFileInfo(path).canonicalFilePath();
#else
    const QString cpath = path;
#endif

    d_lock.lockForRead();

    if( found )
        *found = false;
    Files::const_iterator i = d_files.find(cpath);
    if( i != d_files.end() )
    {
        res = i.value();
        if( found )
            *found = true;
    }

    d_lock.unlock();

    return res;
}

QByteArray FileCache::fetchTextLineFromFile(const QString& path, int line, const QByteArray& defaultString)
{
    QIODevice* in = createFileStreamForReading( path );
    if( in == 0 )
        return defaultString;
    QByteArray str;
    for( int i = 0; i < line; i++ )
    {
        if( in->atEnd() )
        {
            delete in;
            return defaultString;
        }
        str = in->readLine();
    }
    delete in;
    str.chop(1);
    if( str.endsWith('\r') )
        str.chop(1);
    return str;
}

QIODevice*FileCache::createFileStreamForReading(const QString& path) const
{
    bool found;
    QByteArray bytes = getFile( path, &found );
    if( found )
    {
        QBuffer* buf = new QBuffer();
        buf->setData(bytes);
        buf->open(QIODevice::ReadOnly);
        return buf;
    }else if( QFileInfo(path).isReadable() )
    {
        QFile* file = new QFile( path );
        if( !file->open(QIODevice::ReadOnly) )
        {
            delete file;
            return 0;
        }else
            return file;
    }
    return 0;
}