uploads และ downloads
This commit is contained in:
@@ -1,10 +1,11 @@
|
||||
import { ProjectAddService } from '../services/projectAddService.js'
|
||||
import { sendError } from '../utils/response.js'
|
||||
// import { OftenError } from '../utils/oftenError.js'
|
||||
import { GeneralService } from '../share/generalservice.js';
|
||||
import { trim_all_array } from '../utils/trim.js'
|
||||
import { verifyToken, generateToken } from '../utils/token.js'
|
||||
import { Interface } from '../interfaces/Interface.js';
|
||||
import fs from 'fs';
|
||||
import path from 'path';
|
||||
|
||||
|
||||
export class projectAdd {
|
||||
@@ -17,7 +18,12 @@ export class projectAdd {
|
||||
|
||||
async onNavigate(req, res) {
|
||||
this.generalService.devhint(1, 'projectAdd.js', 'onNavigate() start');
|
||||
let organization = req.body.organization;
|
||||
|
||||
// Check Content-Type: ต้องเป็น multipart/form-data เท่านั้น ถ้ามีการส่งไฟล์
|
||||
// (multer เช็คให้ระดับนึงแล้ว แต่เพิ่มความชัวร์ถ้าต้องการ)
|
||||
|
||||
let organization = '';
|
||||
|
||||
const prommis = await this.onProjectAdd(req, res, organization);
|
||||
return prommis;
|
||||
}
|
||||
@@ -30,63 +36,89 @@ export class projectAdd {
|
||||
let token = req.headers.authorization?.split(' ')[1];
|
||||
const decoded = verifyToken(token);
|
||||
|
||||
// ✅ รองรับทั้ง JSON { request: {...} } และ Form-Data (Flat body)
|
||||
const requestData = req.body.request ? req.body.request : req.body;
|
||||
|
||||
// Form-Data Body
|
||||
const requestData = req.body;
|
||||
let name = requestData.prjnam;
|
||||
|
||||
// Override Database จาก Token ตาม Pattern เดิม
|
||||
database = decoded.organization || 'dbo';
|
||||
|
||||
aryResult = await this.projectAddService.getProjectAdd(database, name);
|
||||
latSeq = await this.projectAddService.getLatestProjectSeq(database);
|
||||
// this.generalService.devhint(1, 'budgetSearch.js', 'Login success');
|
||||
|
||||
} catch (error) {
|
||||
idx = 1;
|
||||
console.error(error); // เพิ่ม log error เพื่อ debug
|
||||
console.error(error);
|
||||
} finally {
|
||||
if (idx === 1) return sendError('เกิดข้อผิดพลาดไม่คาดคิดเกิดขึ้น', 'Unexpected error');
|
||||
// if (!aryResult) return sendError('ไม่พบการมีอยู่ของข้อมูล', 'Cannot Find Any Data');
|
||||
|
||||
if (aryResult == 0) {
|
||||
// ส่ง latSeq เข้าไป (ต้อง handle กรณี null ถ้า table ว่าง)
|
||||
const currentSeq = (latSeq && latSeq[0] && latSeq[0].prjseq) ? latSeq[0].prjseq : 0;
|
||||
let prommis = await this.makeArySave(req, currentSeq);
|
||||
return prommis
|
||||
} else {
|
||||
// Cleanup Temp Files if Duplicate
|
||||
if (req.files) {
|
||||
req.files.forEach(f => {
|
||||
if (fs.existsSync(f.path)) fs.unlinkSync(f.path);
|
||||
});
|
||||
}
|
||||
return sendError('คีย์หลักซ้ำในระบบ', 'Duplicate Primary Key');
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
async makeArySave(req, latseq) {
|
||||
// Extract Data ให้รองรับทั้ง 2 แบบ
|
||||
const requestData = req.body.request ? req.body.request : req.body;
|
||||
const requestData = req.body;
|
||||
const nextSeq = latseq + 1;
|
||||
|
||||
// ✅ แก้ไข: Sanitise Input ป้องกัน Error numeric: ""
|
||||
// ถ้าเป็นค่าว่าง ให้แปลงเป็น null หรือ 0.00 ตามประเภทข้อมูล
|
||||
const prjwntbdg = (requestData.prjwntbdg && requestData.prjwntbdg !== '') ? requestData.prjwntbdg : '0.00';
|
||||
const prjusrseq = (requestData.prjusrseq && requestData.prjusrseq !== '') ? requestData.prjusrseq : null;
|
||||
const typ = requestData.typ;
|
||||
|
||||
let arysave = {
|
||||
methods: 'post',
|
||||
prjseq: latseq + 1,
|
||||
prjseq: nextSeq,
|
||||
prjnam: requestData.prjnam,
|
||||
prjusrseq: prjusrseq,
|
||||
prjwntbdg: prjwntbdg,
|
||||
prjacpbdg: '0.00',
|
||||
prjbdgcod: '',
|
||||
prjcomstt: requestData.prjcomstt || 'UAC', // Default UAC ถ้ายิงมาแค่ชื่อกับงบ
|
||||
prjcomstt: requestData.prjcomstt || 'UAC',
|
||||
prjacpdtm: requestData.prjacpdtm || null
|
||||
}
|
||||
|
||||
// เพิ่ม Logic จัดการไฟล์
|
||||
if (req.file) {
|
||||
arysave.prjdoc = req.file.filename // บันทึกชื่อไฟล์ลง DB
|
||||
// Logic ย้ายหลายไฟล์
|
||||
let savedFileNames = [];
|
||||
if (req.files && req.files.length > 0) {
|
||||
if (typ === 'prj') {
|
||||
const targetDir = `uploads/projects/${nextSeq}`;
|
||||
|
||||
try {
|
||||
if (!fs.existsSync(targetDir)) {
|
||||
fs.mkdirSync(targetDir, { recursive: true });
|
||||
}
|
||||
|
||||
// Loop ย้ายไฟล์ทีละไฟล์
|
||||
req.files.forEach(file => {
|
||||
const targetPath = path.join(targetDir, file.filename);
|
||||
fs.renameSync(file.path, targetPath);
|
||||
savedFileNames.push(file.filename);
|
||||
});
|
||||
|
||||
// บันทึกชื่อไฟล์ลง DB (คั่นด้วย Comma)
|
||||
arysave.prjdoc = savedFileNames.join(',');
|
||||
|
||||
} catch (err) {
|
||||
console.error('Error moving files:', err);
|
||||
return sendError('ไม่สามารถบันทึกไฟล์ลงโฟลเดอร์โครงการได้');
|
||||
}
|
||||
} else {
|
||||
// ไม่ใช่ typ prj ไม่ย้าย (แต่ต้องเก็บชื่อไว้ หรือลบทิ้ง แล้วแต่ Business Logic)
|
||||
// ในที่นี้เก็บชื่อ Temp ไว้ก่อน
|
||||
arysave.prjdoc = req.files.map(f => f.filename).join(',');
|
||||
}
|
||||
}
|
||||
|
||||
// กรณี User Seq ไม่ได้ส่งมา (เป็น null) ให้ใช้จาก Token
|
||||
if (!arysave.prjusrseq) {
|
||||
const token = req.headers.authorization?.split(' ')[1];
|
||||
const decoded = verifyToken(token);
|
||||
|
||||
@@ -0,0 +1,74 @@
|
||||
import { GeneralService } from '../share/generalservice.js';
|
||||
import { sendError } from '../utils/response.js';
|
||||
import fs from 'fs';
|
||||
import path from 'path';
|
||||
import archiver from 'archiver'; // ⚠️ ต้อง npm install archiver
|
||||
|
||||
export class projectDownload {
|
||||
|
||||
constructor() {
|
||||
this.generalService = new GeneralService();
|
||||
}
|
||||
|
||||
// GET Route ไม่รับ Body ดังนั้นรับ Parameter จาก Query String หรือ Params
|
||||
async onNavigate(req, res) {
|
||||
this.generalService.devhint(1, 'projectDownload.js', 'onNavigate() start');
|
||||
|
||||
// รับ prjseq จาก Query String (เช่น /projectdownload?prjseq=123)
|
||||
const prjseq = req.query.prjseq;
|
||||
|
||||
if (!prjseq) {
|
||||
return res.status(400).json(sendError('กรุณาระบุ prjseq', 'Missing prjseq parameter'));
|
||||
}
|
||||
|
||||
return await this.onProjectDownload(req, res, prjseq);
|
||||
}
|
||||
|
||||
async onProjectDownload(req, res, prjseq) {
|
||||
try {
|
||||
const folderPath = `uploads/projects/${prjseq}`;
|
||||
|
||||
// 1. ตรวจสอบว่า Folder มีอยู่จริงหรือไม่
|
||||
if (!fs.existsSync(folderPath)) {
|
||||
return res.status(404).json(sendError('ไม่พบเอกสารของโครงการนี้', 'Project documents not found'));
|
||||
}
|
||||
|
||||
// 2. ตั้งค่า Response Header ให้เป็นไฟล์ Zip
|
||||
const archive = archiver('zip', {
|
||||
zlib: { level: 9 } // บีบอัดสูงสุด
|
||||
});
|
||||
|
||||
const zipFilename = `project_${prjseq}_documents.zip`;
|
||||
|
||||
res.attachment(zipFilename); // บอก Browser ว่าให้ Download เป็นชื่อนี้
|
||||
|
||||
// 3. Handle Events
|
||||
archive.on('error', function(err) {
|
||||
console.error('Zip Error:', err);
|
||||
res.status(500).send({error: err.message});
|
||||
});
|
||||
|
||||
// เมื่อบีบอัดเสร็จและส่งข้อมูลครบ
|
||||
archive.on('end', function() {
|
||||
console.log(`Archive wrote ${archive.pointer()} bytes`);
|
||||
});
|
||||
|
||||
// 4. Pipe Archive -> Response (Stream)
|
||||
archive.pipe(res);
|
||||
|
||||
// 5. เอาไฟล์ใน Folder ใส่ Zip
|
||||
// name: false หมายถึงเอาไฟล์ใส่ root ของ zip เลย ไม่ต้องซ้อน folder
|
||||
archive.directory(folderPath, false);
|
||||
|
||||
// 6. สั่งจบการบีบอัด (จะเริ่มส่งข้อมูลทันที)
|
||||
await archive.finalize();
|
||||
|
||||
// ⚠️ หมายเหตุ: เราไม่ return อะไรกลับไป เพราะเรา Pipe Stream ใส่ res โดยตรงแล้ว
|
||||
// globalResponseHandler จะไม่ทำงานเพราะเราไม่ได้ res.json() ซึ่งถูกต้องแล้วสำหรับการโหลดไฟล์
|
||||
|
||||
} catch (error) {
|
||||
console.error('Download Controller Error:', error);
|
||||
return res.status(500).json(sendError('เกิดข้อผิดพลาดในการดาวน์โหลด', 'Download Error'));
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user