[Feat] 한투 계좌 조회 + cors 설정

app.js

@@ -1,3 +1,4 @@
+require("dotenv").config();
 var createError = require("http-errors");
 var express = require("express");
 var path = require("path");
@@ -6,8 +7,20 @@ var logger = require("morgan");
 
 var indexRouter = require("./routes/index");
 var usersRouter = require("./routes/users");
+var balanceRouter = require("./routes/real");
 
 var app = express();
+const cors = require("cors");
+
+/* CORS */
+const allowedOrigins = ["http://localhost:3000"];
+
+app.use(
+  cors({
+    origin: allowedOrigins,
+    credentials: true,
+  })
+);
 
 app.use(logger("dev"));
 app.use(express.json());
@@ -17,6 +30,7 @@ app.use(express.static(path.join(__dirname, "public")));
 
 app.use("/", indexRouter);
 app.use("/users", usersRouter);
+app.use("/api/real", balanceRouter);
 
 // catch 404 and forward to error handler
 app.use(function (req, res, next) {
@@ -34,4 +48,10 @@ app.use(function (err, req, res, next) {
   res.render("error");
 });
 
+const PORT = process.env.PORT || 3001;
+
+app.listen(PORT, () => {
+  console.log(`🚀 서버가 실행 중: http://localhost:${PORT}`);
+});
+
 module.exports = app;

package.json

@@ -19,10 +19,13 @@
   "homepage": "https://github.com/InserToken/UpAndDown-Server#readme",
   "dependencies": {
     "cookie-parser": "~1.4.4",
+    "cors": "^2.8.5",
     "debug": "~2.6.9",
+    "dotenv": "^17.2.0",
     "ejs": "~2.6.1",
     "express": "~4.16.1",
     "http-errors": "~1.6.3",
-    "morgan": "~1.9.1"
+    "morgan": "~1.9.1",
+    "node-fetch": "^2.7.0"
   }
 }

routes/real.js

@@ -0,0 +1,57 @@
+const express = require("express");
+const router = express.Router();
+const { getBalance } = require("../services/stockService");
+
+router.get("/", async (req, res) => {
+  const cano = "50143725";
+  const acnt = "01";
+
+  try {
+    const data = await getBalance(cano, acnt);
+    res.json(data);
+  } catch (err) {
+    console.error("잔고 조회 오류:", err.message);
+    res.status(500).json({ error: "잔고 조회 실패" });
+  }
+});
+
+module.exports = router;
+
+// const express = require("express");
+// const router = express.Router();
+// const { getBalance } = require("../services/getBalance");
+// const Stock = require("../models/Stock");
+// const UserStock = require("../models/UserStock");
+
+// router.post("/", async (req, res) => {
+//   try {
+//     const { userId, cano, acntPrdtCd } = req.body;
+//     const data = await getBalance(cano, acntPrdtCd);
+//     const stocks = data.output1;
+
+//     for (const item of stocks) {
+//       const { pdno, prdt_name } = item;
+
+//       // 주식 저장 (중복 처리)
+//       await Stock.updateOne(
+//         { stock_code: pdno },
+//         { $set: { company: prdt_name, state: true } },
+//         { upsert: true }
+//       );
+
+//       // 보유 주식 저장
+//       await UserStock.updateOne(
+//         { user_id: userId, stock_code: pdno },
+//         { $setOnInsert: { cumulative_score: 0 } },
+//         { upsert: true }
+//       );
+//     }
+
+//     res.json({ count: stocks.length, stocks: stocks.map((s) => s.prdt_name) });
+//   } catch (err) {
+//     console.error("잔고 조회 오류:", err.message);
+//     res.status(500).json({ message: err.message });
+//   }
+// });
+
+// module.exports = router;

services/stockService.js

@@ -0,0 +1,53 @@
+const fetch = require("node-fetch");
+const { getAccessToken } = require("./tokenService");
+
+async function getBalance(cano, acntPrdtCd) {
+  try {
+    const token = await getAccessToken();
+
+    const url = new URL(
+      "/uapi/domestic-stock/v1/trading/inquire-balance",
+      process.env.API_DOMAIN
+    );
+
+    url.searchParams.set("CANO", cano);
+    url.searchParams.set("ACNT_PRDT_CD", acntPrdtCd);
+    url.searchParams.set("AFHR_FLPR_YN", "N");
+    url.searchParams.set("INQR_DVSN", "02");
+    url.searchParams.set("UNPR_DVSN", "01");
+    url.searchParams.set("FUND_STTL_ICLD_YN", "N");
+    url.searchParams.set("FNCG_AMT_AUTO_RDPT_YN", "N");
+    url.searchParams.set("PRCS_DVSN", "00");
+    url.searchParams.set("OFL_YN", "");
+    url.searchParams.set("CTX_AREA_FK100", "");
+    url.searchParams.set("CTX_AREA_NK100", "");
+
+    console.log("요청 URL:", url.toString());
+
+    const res = await fetch(url.toString(), {
+      method: "GET",
+      headers: {
+        authorization: `Bearer ${token}`,
+        appkey: process.env.API_APPKEY,
+        appsecret: process.env.API_APPSECRET,
+        tr_id: "VTTC8434R",
+        accept: "*/*",
+        "accept-encoding": "gzip, deflate, br",
+        connection: "keep-alive",
+        "user-agent": "PostmanRuntime/7.44.1",
+      },
+    });
+
+    if (!res.ok) {
+      const errText = await res.text();
+      throw new Error(`잔고 조회 실패: ${res.status} - ${errText}`);
+    }
+
+    return await res.json();
+  } catch (err) {
+    console.error("🚨 getBalance 오류:", err.message);
+    throw err;
+  }
+}
+
+module.exports = { getBalance };

services/tokenService.js

@@ -0,0 +1,25 @@
+const fetch = require("node-fetch");
+
+let cachedToken = null;
+
+async function getAccessToken() {
+  if (cachedToken) return cachedToken;
+
+  const res = await fetch(`${process.env.TOKEN_DOMAIN}`, {
+    method: "POST",
+    headers: { "content-type": "application/json" },
+    body: JSON.stringify({
+      grant_type: "client_credentials",
+      appkey: process.env.API_APPKEY,
+      appsecret: process.env.API_APPSECRET,
+    }),
+  });
+
+  if (!res.ok) throw new Error("토큰 발급 실패");
+
+  const data = await res.json();
+  cachedToken = data.access_token;
+  return cachedToken;
+}
+
+module.exports = { getAccessToken };