Initial commit: PicUpper webcam upload service
This commit is contained in:
sebseb7
220
server.js
Normal file
220
server.js
Normal file
@@ -0,0 +1,220 @@
|
||||
import express from 'express';
|
||||
import multer from 'multer';
|
||||
import path from 'path';
|
||||
import fs from 'fs';
|
||||
import { fileURLToPath } from 'url';
|
||||
import { validateApiKey, loadUsers } from './users.js';
|
||||
|
||||
const __filename = fileURLToPath(import.meta.url);
|
||||
const __dirname = path.dirname(__filename);
|
||||
|
||||
// Configuration from environment variables
|
||||
const PORT = process.env.PICUPPER_PORT;
|
||||
const UPLOAD_DIR = process.env.PICUPPER_UPLOAD_DIR || path.join(__dirname, 'uploads');
|
||||
const MAX_FILE_SIZE = 5 * 1024 * 1024; // 5MB - typical webcam snapshot size
|
||||
|
||||
if (!PORT) {
|
||||
console.error('ERROR: PICUPPER_PORT environment variable is required');
|
||||
process.exit(1);
|
||||
}
|
||||
|
||||
// Check for users
|
||||
const users = loadUsers();
|
||||
if (Object.keys(users).length === 0) {
|
||||
console.warn('WARNING: No users configured. Use "node manage-users.js add <username>" to create users.');
|
||||
}
|
||||
|
||||
// Ensure upload directory exists
|
||||
if (!fs.existsSync(UPLOAD_DIR)) {
|
||||
fs.mkdirSync(UPLOAD_DIR, { recursive: true });
|
||||
}
|
||||
|
||||
const app = express();
|
||||
|
||||
// API Key authentication middleware
|
||||
function authenticate(req, res, next) {
|
||||
const providedKey = req.headers['x-api-key'] || req.query.apiKey;
|
||||
|
||||
if (!providedKey) {
|
||||
return res.status(401).json({ error: 'Unauthorized: Missing API key' });
|
||||
}
|
||||
|
||||
const username = validateApiKey(providedKey);
|
||||
if (!username) {
|
||||
return res.status(401).json({ error: 'Unauthorized: Invalid API key' });
|
||||
}
|
||||
|
||||
// Attach username to request for logging/tracking
|
||||
req.authenticatedUser = username;
|
||||
next();
|
||||
}
|
||||
|
||||
// Configure multer storage
|
||||
const storage = multer.diskStorage({
|
||||
destination: (req, file, cb) => {
|
||||
const cameraId = req.body.cameraId || req.query.cameraId || 'default';
|
||||
const now = new Date();
|
||||
|
||||
// Create directory structure: uploads/{cameraId}/{YYYY}/{MM}/{DD}/
|
||||
const year = now.getFullYear().toString();
|
||||
const month = String(now.getMonth() + 1).padStart(2, '0');
|
||||
const day = String(now.getDate()).padStart(2, '0');
|
||||
|
||||
const destPath = path.join(UPLOAD_DIR, cameraId, year, month, day);
|
||||
|
||||
fs.mkdirSync(destPath, { recursive: true });
|
||||
cb(null, destPath);
|
||||
},
|
||||
filename: (req, file, cb) => {
|
||||
const cameraId = req.body.cameraId || req.query.cameraId || 'default';
|
||||
const now = new Date();
|
||||
|
||||
// Filename format: {cameraId}_{timestamp}.{ext}
|
||||
const timestamp = now.toISOString().replace(/[:.]/g, '-');
|
||||
const ext = path.extname(file.originalname) || '.jpg';
|
||||
|
||||
cb(null, `${cameraId}_${timestamp}${ext}`);
|
||||
}
|
||||
});
|
||||
|
||||
// File filter - only accept images
|
||||
function fileFilter(req, file, cb) {
|
||||
const allowedMimes = ['image/jpeg', 'image/png', 'image/webp', 'image/gif'];
|
||||
|
||||
if (allowedMimes.includes(file.mimetype)) {
|
||||
cb(null, true);
|
||||
} else {
|
||||
cb(new Error(`Invalid file type: ${file.mimetype}. Allowed: ${allowedMimes.join(', ')}`), false);
|
||||
}
|
||||
}
|
||||
|
||||
const upload = multer({
|
||||
storage,
|
||||
fileFilter,
|
||||
limits: {
|
||||
fileSize: MAX_FILE_SIZE
|
||||
}
|
||||
});
|
||||
|
||||
// Health check endpoint (no auth required)
|
||||
app.get('/health', (req, res) => {
|
||||
res.json({
|
||||
status: 'ok',
|
||||
timestamp: new Date().toISOString(),
|
||||
uptime: process.uptime()
|
||||
});
|
||||
});
|
||||
|
||||
// Upload endpoint
|
||||
app.post('/upload', authenticate, upload.single('image'), (req, res) => {
|
||||
if (!req.file) {
|
||||
return res.status(400).json({ error: 'No image file provided. Use form field "image".' });
|
||||
}
|
||||
|
||||
const cameraId = req.body.cameraId || req.query.cameraId || 'default';
|
||||
|
||||
res.json({
|
||||
success: true,
|
||||
cameraId,
|
||||
filename: req.file.filename,
|
||||
path: req.file.path,
|
||||
size: req.file.size,
|
||||
timestamp: new Date().toISOString()
|
||||
});
|
||||
});
|
||||
|
||||
// List cameras (directories in upload folder)
|
||||
app.get('/cameras', authenticate, (req, res) => {
|
||||
try {
|
||||
if (!fs.existsSync(UPLOAD_DIR)) {
|
||||
return res.json({ cameras: [] });
|
||||
}
|
||||
|
||||
const cameras = fs.readdirSync(UPLOAD_DIR, { withFileTypes: true })
|
||||
.filter(dirent => dirent.isDirectory())
|
||||
.map(dirent => dirent.name);
|
||||
|
||||
res.json({ cameras });
|
||||
} catch (error) {
|
||||
res.status(500).json({ error: 'Failed to list cameras', details: error.message });
|
||||
}
|
||||
});
|
||||
|
||||
// Get statistics for a camera
|
||||
app.get('/stats/:cameraId', authenticate, (req, res) => {
|
||||
const { cameraId } = req.params;
|
||||
const cameraPath = path.join(UPLOAD_DIR, cameraId);
|
||||
|
||||
if (!fs.existsSync(cameraPath)) {
|
||||
return res.status(404).json({ error: `Camera '${cameraId}' not found` });
|
||||
}
|
||||
|
||||
try {
|
||||
let totalImages = 0;
|
||||
let totalSize = 0;
|
||||
let oldestImage = null;
|
||||
let newestImage = null;
|
||||
|
||||
// Recursively count files
|
||||
function countFiles(dir) {
|
||||
const entries = fs.readdirSync(dir, { withFileTypes: true });
|
||||
|
||||
for (const entry of entries) {
|
||||
const fullPath = path.join(dir, entry.name);
|
||||
|
||||
if (entry.isDirectory()) {
|
||||
countFiles(fullPath);
|
||||
} else if (entry.isFile()) {
|
||||
totalImages++;
|
||||
const stats = fs.statSync(fullPath);
|
||||
totalSize += stats.size;
|
||||
|
||||
if (!oldestImage || stats.mtime < oldestImage) {
|
||||
oldestImage = stats.mtime;
|
||||
}
|
||||
if (!newestImage || stats.mtime > newestImage) {
|
||||
newestImage = stats.mtime;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
countFiles(cameraPath);
|
||||
|
||||
res.json({
|
||||
cameraId,
|
||||
totalImages,
|
||||
totalSizeBytes: totalSize,
|
||||
totalSizeMB: (totalSize / (1024 * 1024)).toFixed(2),
|
||||
oldestImage: oldestImage ? oldestImage.toISOString() : null,
|
||||
newestImage: newestImage ? newestImage.toISOString() : null
|
||||
});
|
||||
} catch (error) {
|
||||
res.status(500).json({ error: 'Failed to get camera stats', details: error.message });
|
||||
}
|
||||
});
|
||||
|
||||
// Error handling middleware
|
||||
app.use((error, req, res, next) => {
|
||||
if (error instanceof multer.MulterError) {
|
||||
if (error.code === 'LIMIT_FILE_SIZE') {
|
||||
return res.status(413).json({
|
||||
error: `File too large. Maximum size: ${MAX_FILE_SIZE / (1024 * 1024)}MB`
|
||||
});
|
||||
}
|
||||
return res.status(400).json({ error: error.message });
|
||||
}
|
||||
|
||||
if (error) {
|
||||
return res.status(400).json({ error: error.message });
|
||||
}
|
||||
|
||||
next();
|
||||
});
|
||||
|
||||
// Start server
|
||||
app.listen(PORT, () => {
|
||||
console.log(`PicUpper service running on port ${PORT}`);
|
||||
console.log(`Upload directory: ${UPLOAD_DIR}`);
|
||||
console.log(`Max file size: ${MAX_FILE_SIZE / (1024 * 1024)}MB`);
|
||||
});
|
||||
Reference in New Issue
Block a user