v0.5.10: /export — attachment default, ?view=1 inline, ?filename=custom

package.json

@@ -1,6 +1,6 @@
 {
   "name": "ipwhitelist",
-  "version": "0.5.9",
+  "version": "0.5.10",
   "description": "IP WhiteList microservice for cloud provider",
   "main": "server.js",
   "scripts": {

src/api/routes/entries.js

@@ -95,7 +95,9 @@ function createEntriesRouter({ q }) {
       const cidrs = await q.getExportCIDRs(companyId);
       const aggregated = aggregateCIDRs(cidrs);
       res.setHeader('Content-Type', 'text/plain; charset=utf-8');
-      res.setHeader('Content-Disposition', 'inline; filename="whitelist.txt"');
+      const fname = req.query.filename || 'white-list.txt';
+      const disp  = req.query.view === '1' ? 'inline' : 'attachment';
+      res.setHeader('Content-Disposition', `${disp}; filename="${fname}"`);
       res.send(aggregated.join('\n') + (aggregated.length ? '\n' : ''));
     } catch (e) {
       res.status(500).json({ error: e.message });

src/routes/export.js

@@ -39,8 +39,12 @@ function createRouter({ q, exportLimiter, aggregateCIDRs }) {
 
       if (!req.user.isAdmin) {
         // Обычный пользователь: только его компания.
-        // getOrCreateCompany — атомарная UPSERT, безопасна для параллельных вызовов.
+        // ?client_id= для мульти-компании
-        const company = await q.getOrCreateCompany(req.user.clientId, req.user.companyName);
+        const requestedId = (req.query.client_id || '').trim();
+        const effectiveClientId = (requestedId && req.user.allClientIds && req.user.allClientIds.includes(requestedId))
+          ? requestedId
+          : req.user.clientId;
+        const company = await q.getOrCreateCompany(effectiveClientId, effectiveClientId);
         companyId = company.id;
       } else if (req.query.company) {
         // Администратор с необязательным фильтром по компании.
@@ -56,9 +60,12 @@ function createRouter({ q, exportLimiter, aggregateCIDRs }) {
       const aggregated = aggregateCIDRs(cidrs);
 
       // Content-Type: text/plain — чтобы браузер не пытался разбирать как HTML.
-      // Content-Disposition: inline — открывается в браузере, а не скачивается.
+      // ?view=1 → inline (просмотр в браузере), иначе → attachment (скачивание)
+      // ?filename=X → своё имя файла, по умолчанию white-list.txt
       res.setHeader('Content-Type', 'text/plain; charset=utf-8');
-      res.setHeader('Content-Disposition', 'inline; filename="whitelist.txt"');
+      const fname = req.query.filename || 'white-list.txt';
+      const disp  = req.query.view === '1' ? 'inline' : 'attachment';
+      res.setHeader('Content-Disposition', `${disp}; filename="${fname}"`);
 
       // Один CIDR на строку. Если список пустой — возвращаем пустую строку (не ошибку).
       res.send(aggregated.join('\n') + (aggregated.length ? '\n' : ''));

ui/routes/export.js

@@ -14,11 +14,20 @@ function createRouter() {
 
   router.get('/export', async (req, res) => {
     const token = api.token(req);
-    const companyQuery = req.query.company ? '?company=' + parseInt(req.query.company, 10) : '';
+    // Прокидываем query-параметры в API
+    const params = new URLSearchParams();
+    if (req.query.company) params.set('company', req.query.company);
+    if (req.query.client_id) params.set('client_id', req.query.client_id);
+    const qs = params.toString();
+    const companyQuery = qs ? '?' + qs : '';
     try {
       const r = await api.get('/api/v1/entries/export' + companyQuery, token);
       res.set('Content-Type', 'text/plain; charset=utf-8');
-      res.set('Content-Disposition', 'inline; filename="whitelist.txt"');
+      // ?view=1 → inline (просмотр), иначе → attachment (скачивание)
+      // ?filename=X → своё имя, по умолчанию white-list.txt
+      const fname = req.query.filename || 'white-list.txt';
+      const disp  = req.query.view === '1' ? 'inline' : 'attachment';
+      res.set('Content-Disposition', disp + '; filename="' + fname + '"');
       res.send(r.data);
     } catch (e) {
       res.status(500).send('Ошибка экспорта: ' + e.message);