forked from sqlitebrowser/sqlitebrowser
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathsqlitetypes.h
More file actions
134 lines (110 loc) · 3.48 KB
/
sqlitetypes.h
File metadata and controls
134 lines (110 loc) · 3.48 KB
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
126
127
128
129
130
131
132
133
134
#pragma once
#ifndef SQLITETYPES_H
#define SQLITETYPES_H
#include "antlr/ASTRefCount.hpp"
#include <QString>
#include <QSharedPointer>
#include <QVector>
#include <QStringList>
namespace sqlb {
class Field
{
public:
Field(const QString& name,
const QString& type,
bool notnull = false,
const QString& defaultvalue = "",
const QString& check = "",
bool pk = false)
: m_name(name)
, m_type(type)
, m_notnull(notnull)
, m_check(check)
, m_defaultvalue(defaultvalue)
, m_autoincrement(false)
, m_primaryKey(pk)
{}
QString toString(const QString& indent = "\t", const QString& sep = "\t") const;
void setName(const QString& name) { m_name = name; }
void setType(const QString& type) { m_type = type; }
void setNotNull(bool notnull = true) { m_notnull = notnull; }
void setCheck(const QString& check) { m_check = check; }
void setDefaultValue(const QString& defaultvalue) { m_defaultvalue = defaultvalue; }
void setAutoIncrement(bool autoinc) { m_autoincrement = autoinc; }
void setPrimaryKey(bool pk) { m_primaryKey = pk; }
bool isText() const;
bool isInteger() const;
const QString& name() const { return m_name; }
const QString& type() const { return m_type; }
bool notnull() const { return m_notnull; }
const QString& check() const { return m_check; }
const QString& defaultValue() const { return m_defaultvalue; }
bool autoIncrement() const { return m_autoincrement; }
bool primaryKey() const { return m_primaryKey; }
static QStringList Datatypes;
private:
QString m_name;
QString m_type;
bool m_notnull;
QString m_check;
QString m_defaultvalue;
bool m_autoincrement; //! this is stored here for simplification
bool m_primaryKey;
};
typedef QSharedPointer<Field> FieldPtr;
typedef QVector< FieldPtr > FieldVector;
class Table
{
public:
Table(const QString& name): m_name(name) {}
virtual ~Table();
void setName(const QString& name) { m_name = name; }
const QString& name() const { return m_name; }
const FieldVector& fields() const { return m_fields; }
/**
* @brief Creates an empty insert statement.
* @return An sqlite conform INSERT INTO statement with empty values. (NULL,'',0)
*/
QString emptyInsertStmt() const;
/**
* @brief Returns the CREATE TABLE statement for this table object
* @return A QString with the CREATE TABLE object.
*/
QString sql() const;
void addField(const FieldPtr& f);
bool removeField(const QString& sFieldName);
void setFields(const FieldVector& fields);
void setField(int index, FieldPtr f) { m_fields[index] = f; }
void clear();
/**
* @brief findField Finds a field and returns the index.
* @param sname
* @return The field index if the field was found.
* -1 if field coulnd't be found.
*/
int findField(const QString& sname);
static Table parseSQL(const QString& sSQL);
private:
QStringList fieldList() const;
bool hasAutoIncrement() const;
private:
QString m_name;
FieldVector m_fields;
};
/**
* @brief The CreateTableWalker class
* Goes trough the createtable AST and returns
* Table object.
*/
class CreateTableWalker
{
public:
CreateTableWalker(antlr::RefAST r) : m_root(r) {}
Table table();
private:
void parsecolumn(FieldPtr& f, antlr::RefAST c);
private:
antlr::RefAST m_root;
};
} //namespace sqlb
#endif // SQLITETYPES_H