Spaces:

Raybilhelp
/

Secret7

Paused

App Files Files Community

Raybilhelp commited on Jun 15

Commit

acd0769

verified ·

1 Parent(s): 62261e6

Update server.js

Browse files

Files changed (1) hide show

server.js +79 -134

server.js CHANGED Viewed

@@ -1,158 +1,103 @@
-require('dotenv').config();
 const express = require('express');
 const cors = require('cors');
 const admin = require('firebase-admin');
-const Groq = require('groq-sdk');
-const { Octokit } = require('@octokit/rest');
 const app = express();
-app.use(cors({ origin: '*' }));
-app.use(express.json({ limit: '10mb' }));
-// ১. ফায়ারবেস অ্যাডমিন এসডিকে ইনিশিয়ালিলাইজেশন
-try {
-  const serviceAccount = JSON.parse(process.env.FIREBASE_ADMIN_API);
-  admin.initializeApp({
-    credential: admin.credential.cert(serviceAccount),
-    databaseURL: process.env.FIREBASE_REALTIME_URL
-  });
-  console.log("Firebase Connected.");
-} catch (error) {
-  console.error("Firebase Init Error:", error.message);
-}
-const db = admin.database();
-// ২. Groq এবং GitHub ইনিশিয়ালিলাইজেশন
-const groq = new Groq({ apiKey: process.env.GROQ_API });
-const octokit = new Octokit({ auth: process.env.GITHUB_API_TOKEN });
-const REPO_OWNER = 'chaingroupworld-rgb';
-const REPO_NAME = 'Website-visit-1';
-// 🤖 মাস্টার প্রম্পট: এটি এআই-কে পিওর এসইও এইচটিএমএল ফাইল বানাতে বাধ্য করবে
-const MASTER_PROMPT = `You are an expert web developer and storyteller. Generate a unique, interesting short story in Bengali.
-The output MUST be a complete, production-ready, standalone HTML page code using Tailwind CSS.
-Do NOT include any markdown code blocks like \`\`\`html or explanation. Just start with <!DOCTYPE html> and end with </html>.
-SEO & Design Requirements:
-1. Include <title>, <meta name="description"> (max 160 chars), <meta name="keywords">, and JSON-LD Schema Markup for BlogPosting.
-2. In og:image, use a placeholder or your brand asset logo, but ensure the content has a stunning typographic banner.
-3. For the main body hero section, create a beautiful typography card using Tailwind CSS with dark gradients (slate-900 to indigo-950), glowing neon borders, and large glowing text displaying the story title. This acts as the visual image thumbnail.
-4. The story layout must be clean, readable, responsive for mobile, with elegant fonts. Use appropriate H1, H2 tags.`;
-// 🚀 কোর অটোমেশন ফাংশন (গল্প তৈরি ও গিটহাবে পুশ)
-async function generateAndPushStory() {
-  console.log("Automation triggered: Generating new story...");
-  try {
-    // ক) Groq (Llama 4 Scout 17B) থেকে সম্পূর্ণ HTML কোড নেওয়া
-    const chatCompletion = await groq.chat.completions.create({
-      messages: [{ role: "user", content: MASTER_PROMPT }],
-      model: "meta-llama/llama-4-scout-17b-16e-instruct", // বড় কাজের জন্য ১৭বি মডেল
-    });
-    const htmlContent = chatCompletion.choices[0]?.message?.content || "";
-    if (!htmlContent.includes("<!DOCTYPE html>")) {
-      console.error("Invalid HTML output from AI");
-      return;
-    }
-    // খ) একটি ইউনিক স্লাগ বা ফাইলের নাম তৈরি (টাইমস্ট্যাম্প দিয়ে)
-    const timestamp = Date.now();
-    const fileName = `story-${timestamp}.html`;
-    // গ) সরাসরি নতুন গল্পের ফাইলটি GitHub-এ পুশ করা
-    const base64Content = Buffer.from(htmlContent).toString('base64');
-    await octokit.repos.createOrUpdateFileContents({
-      owner: REPO_OWNER,
-      repo: REPO_NAME,
-      path: fileName,
-      message: `Automated Story Upload: ${fileName}`,
-      content: base64Content
-    });
-    console.log(`Successfully uploaded: ${fileName}`);
-    // ঘ) ফায়ারবেসে ট্র্যাক বা মেটাডাটা সেভ করা
-    const storyRef = db.ref('automated_stories').push();
-    await storyRef.set({
-      fileName: fileName,
-      timestamp: admin.database.ServerValue.TIMESTAMP
     });
-    // 🔄 ঙ) মূল index.html ফাইলটি রিড করে নতুন গল্পের লিংক যুক্ত করা
-    await updateIndexHtml(fileName);
-  } catch (error) {
-    console.error("Automation Logic Error:", error.message);
   }
-}
-// 🔄 index.html ডাইনামিকালি আপডেট করার ফাংশน
-async function updateIndexHtml(newStoryFile) {
   try {
-    let indexSha, indexContentRaw;
-    // গিটহাব থেকে বর্তমান index.html তুলে আনা
-    try {
-      const { data } = await octokit.repos.getContent({
-        owner: REPO_OWNER,
-        repo: REPO_NAME,
-        path: 'index.html',
-      });
-      indexSha = data.sha;
-      indexContentRaw = Buffer.from(data.content, 'base64').toString('utf-8');
-    } catch (e) {
-      // যদি index.html না থাকে, তবে একটি বেসিক স্ট্রাকচার তৈরি হবে
-      indexContentRaw = `<!DOCTYPE html><html><head><title>Raybil Stories</title></head><body class="bg-slate-950 text-white"><h1>গল্পের তালিকা</h1><div id="story-list"></div></body></html>`;
-    }
-    // নতুন গল্পের জন্য লিংকের এইচটিএমএল এলিমেন্ট
-    const newLinkHtml = `<p><a href="${newStoryFile}" class="text-indigo-400 hover:underline">নতুন গল্প (${new Date().toLocaleTimeString()})</a></p>\n`;
-    // কোডের ভেতরের নির্দিষ্ট জায়গাতে (যেমন <div id="story-list"> এর ঠিক নিচে) লিংকটি পুশ করা
-    let updatedIndexContent;
-    if (indexContentRaw.includes('<div id="story-list">')) {
-      updatedIndexContent = indexContentRaw.replace(
-        '<div id="story-list">',
-        `<div id="story-list">\n${newLinkHtml}`
-      );
-    } else {
-      // সেফটি ফলব্যাক: যদি ট্যাগ না মেলে তবে বডির শেষে পুশ করবে
-      updatedIndexContent = indexContentRaw.replace('</body>', `${newLinkHtml}</body>`);
-    }
-    // আপডেট হওয়া index.html গিটহাবে পুশ ব্যাক করা
-    await octokit.repos.createOrUpdateFileContents({
-      owner: REPO_OWNER,
-      repo: REPO_NAME,
-      path: 'index.html',
-      message: 'Updated index.html with latest story link',
-      content: Buffer.from(updatedIndexContent).toString('base64'),
-      sha: indexSha
     });
-    console.log("index.html successfully updated on GitHub!");
-  } catch (err) {
-    console.error("Error updating index.html:", err.message);
   }
-}
-// ⏱️ টেস্ট করার ���ন্য প্রতি ২ মিনিট (১২০০০০ মিলিসেকেন্ড) পর পর রান হবে
-// যখন আপনি ৪ ঘণ্টা করতে চাইবেন, তখন ১২০০০০ বদলে ১৪৪০০০০০ (14400000) দিয়ে দেবেন।
-setInterval(generateAndPushStory, 120000);
-// বেসিক রুট ও হেলথ চেক
-app.get('/', (req, res) => {
-  res.status(200).send('Raybil Autonomous AI Engine is Running...');
 });
-// ম্যানুয়ালি টেস্ট ট্রিগার করার জন্য একটি গোপন এপিআই রুট
-app.get('/api/trigger-now', async (req, res) => {
-  await generateAndPushStory();
-  res.json({ success: true, message: "Story generation forced successfully." });
 });
-const PORT = process.env.PORT || 7860;
-app.listen(PORT, () => {
-  console.log(`Server running on port ${PORT}`);
 });

 const express = require('express');
 const cors = require('cors');
 const admin = require('firebase-admin');
 const app = express();
+const PORT = process.env.PORT || 7860;
+app.use(cors());
+app.use(express.json());
+// ফায়ারবেস এডমিন এসডিকে ইনিশিয়ালাইজ করা
+const serviceAccount = require("./firebase-key.json");
+admin.initializeApp({
+  credential: admin.credential.cert(serviceAccount)
+});
+// ==========================================
+// ১. হেলথ চেক রাউট (/health এবং /check)
+// এখানে GET, POST, PUT যাই আসুক না কেন, সার্ভার 'ok' রেসপন্স করবে (ক্রোন জবের জন্য)
+// ==========================================
+const healthCheckHandler = (req, res) => {
+  res.status(200).send('ok');
+};
+app.all('/health', healthCheckHandler);
+app.all('/check', healthCheckHandler);
+// ==========================================
+// ২. সিকিউরিটি মিডলওয়্যার (Middleware)
+// /subscribe এবং /unsubscribe-এ POST ছাড়া অন্য কোনো রিকোয়েست আসলে তা ব্লক করে দেবে
+// ==========================================
+const enforcePostOnly = (req, res, next) => {
+  if (req.method !== 'POST') {
+    return res.status(405).json({
+      success: false,
+      message: `Method ${req.method} Not Allowed. Only POST requests are accepted on this endpoint.`
     });
+  }
+  next();
+};
+// ==========================================
+// ৩. মূল কার্যকারী রাউট (/subscribe এবং /unsubscribe)
+// ==========================================
+// --- সাবস্ক্রাইব সেকশন ---
+app.post('/subscribe', enforcePostOnly, async (req, res) => {
+  const { fcm_token, selected_category, action } = req.body;
+  // ভ্যালিডেশন এবং অ্যাকশন চেক (অ্যাকশন 'subscribe' হওয়া বাধ্যতামূলক)
+  if (!fcm_token || !selected_category || action !== 'subscribe') {
+    return res.status(400).json({ success: false, message: "Invalid payload or wrong action for subscription." });
   }
   try {
+    // ফায়ারবেস টপিক এপিআই কল
+    await admin.messaging().subscribeToTopic(fcm_token, selected_category);
+    console.log(`Successfully subscribed token to topic: ${selected_category}`);
+    res.status(200).json({
+      success: true,
+      message: `Successfully registered to ${selected_category} category.`
     });
+  } catch (error) {
+    console.error("Firebase subscription error:", error);
+    res.status(500).json({ success: false, error: error.message });
+  }
+});
+// --- আন-সাবস্ক্রাইব সেকশন ---
+app.post('/unsubscribe', enforcePostOnly, async (req, res) => {
+  const { fcm_token, selected_category, action } = req.body;
+  // ভ্যালিডেশন এবং অ্যাকশন চেক (অ্যাকশন 'unsubscribe' হওয়া বাধ্যতামূলক)
+  if (!fcm_token || !selected_category || action !== 'unsubscribe') {
+    return res.status(400).json({ success: false, message: "Invalid payload or wrong action for unsubscription." });
   }
+  try {
+    // ফায়ারবেস টপিক থেকে রিমুভ করার এপিআই কল
+    await admin.messaging().unsubscribeFromTopic(fcm_token, selected_category);
+    console.log(`Successfully unsubscribed token from topic: ${selected_category}`);
+    res.status(200).json({
+      success: true,
+      message: `Successfully removed from ${selected_category} category.`
+    });
+  } catch (error) {
+    console.error("Firebase unsubscription error:", error);
+    res.status(500).json({ success: false, error: error.message });
+  }
 });
+// ==========================================
+// ৪. গ্লোবাল ক্যাচ-অল রাউট (অন্য যেকোনো ভুল পাথের জন্য)
+// ==========================================
+app.use((req, res) => {
+  res.status(404).json({ success: false, message: "Endpoint not found." });
 });
+app.listen(PORT, '0.0.0.0', () => {
+  console.log(`Server is running on port ${PORT}`);
 });