-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathapp.js
105 lines (91 loc) · 2.76 KB
/
app.js
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
require('dotenv').config();
const express = require('express');
const multer = require('multer');
const path = require('path');
const fs = require('fs');
const sqlite3 = require('sqlite3').verbose();
const bodyParser = require('body-parser');
const cors = require('cors');
const developmentPlansRouter = require('./routes/developmentPlans');
const developmentStepsRouter = require('./routes/developmentSteps');
const { handleOpenAIMessage } = require('./handlers/openaiMessageHandler');
const app = express();
app.use(express.static('public'));
app.use(bodyParser.json());
app.use(cors());
const storage = multer.diskStorage({
destination: function (req, file, cb) {
const uploadDir = 'uploads';
fs.mkdirSync(uploadDir, { recursive: true });
cb(null, uploadDir);
},
filename: function (req, file, cb) {
const dbName = req.body.dbname;
const ext = path.extname(file.originalname).toLowerCase();
cb(null, dbName + ext);
}
});
const upload = multer({
storage: storage,
fileFilter: function (req, file, cb) {
if (!(/\.(sqlite|db)$/i).test(file.originalname)) {
return cb(new Error('Only SQLite database files are allowed!'));
}
cb(null, true);
}
});
app.post('/upload', upload.single('dbfile'), (req, res) => {
res.status(200).send('File uploaded successfully');
}, (error, req, res, next) => {
res.status(400).send(error.message);
});
app.get('/databases', (req, res) => {
const uploadDir = 'uploads';
fs.readdir(uploadDir, (err, files) => {
if (err) {
res.status(500).send('Failed to list databases');
return;
}
const databases = files.filter(file => /\.(sqlite|db)$/i.test(file)).map(file => {
return { name: file };
});
res.json(databases);
});
});
app.get('/apps', (req, res) => {
const dbName = req.query.db;
if (!dbName) {
return res.status(400).send('Database name is required');
}
const dbPath = path.join('uploads', `${dbName}.sqlite`);
if (!fs.existsSync(dbPath)) {
return res.status(404).send('Database file not found');
}
let db = new sqlite3.Database(dbPath, sqlite3.OPEN_READONLY, (err) => {
if (err) {
res.status(500).send("Error opening database: " + err.message);
return;
}
});
db.serialize(() => {
db.all("SELECT * FROM app", (err, rows) => {
if (err) {
res.status(500).send("Error reading from database: " + err.message);
return;
}
res.json(rows);
});
});
db.close((err) => {
if (err) {
console.error("Error closing database: " + err.message);
}
});
});
app.use(developmentPlansRouter);
app.use(developmentStepsRouter);
app.post('/submit_messages', handleOpenAIMessage);
const port = process.env.PORT || 3000;
app.listen(port, () => {
console.log(`Server running on port ${port}`);
});