server.js

require('dotenv').config();
const express = require('express');
const mongoose = require('mongoose');
const bodyParser = require('body-parser');
const cors = require('cors');
const { URL } = require('url');

// [수정] 우리가 만든 모델 파일 불러오기! (이 한 줄로 스키마 정의 끝)
const Url = require('./models/Url'); 

const app = express();

const PORT = process.env.PORT || 3000;
const MONGO_URI = process.env.MONGO_URI || 'mongodb://127.0.0.1:27017/my_tinyurl';
const BASE_URL = process.env.BASE_URL || `http://localhost:${PORT}`;

// 미들웨어
app.use(cors());
app.use(bodyParser.json());
app.use(bodyParser.urlencoded({ extended: true }));
app.use(express.static('public'));

// DB 연결
mongoose.connect(MONGO_URI)
    .then(() => console.log('✅ MongoDB 연결 성공!'))
    .catch(err => console.error('❌ MongoDB 연결 실패:', err));

// --- (여기 있던 스키마 정의 코드는 싹 지워졌습니다!) ---

// 랜덤 ID 생성 함수
function generateShortId(length = 5) {
    const chars = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
    let result = '';
    for (let i = 0; i < length; i++) {
        result += chars.charAt(Math.floor(Math.random() * chars.length));
    }
    return result;
}

// 🚀 라우팅 (API)

// URL 단축 (POST)
app.post('/shorten', async (req, res) => {
    try {
        let { originalUrl } = req.body;

        // 유효성 검사 로직들...
        if (!originalUrl.match(/^https?:\/\//)) originalUrl = 'http://' + originalUrl;

        try {
            const parsedUrl = new URL(originalUrl);
            if (parsedUrl.hostname.indexOf('.') === -1 && parsedUrl.hostname !== 'localhost') {
                 throw new Error("Invalid Domain");
            }
            const myDomain = new URL(BASE_URL).hostname;
            if (myDomain === parsedUrl.hostname) {
                return res.status(400).json({ error: '이미 단축된 URL입니다.' });
            }
        } catch (err) {
            return res.status(400).json({ error: '유효하지 않은 URL입니다.' });
        }

        // DB 조회 및 저장
        let urlData = await Url.findOne({ originalUrl }); // 여기서 models/Url.js를 사용함

        if (urlData) {
            return res.json({ shortUrl: `${BASE_URL}/${urlData.shortId}` });
        }

        let shortId = generateShortId();
        while (await Url.findOne({ shortId })) {
            shortId = generateShortId();
        }

        urlData = new Url({ originalUrl, shortId });
        await urlData.save();

        res.json({ shortUrl: `${BASE_URL}/${urlData.shortId}` });

    } catch (err) {
        console.error(err);
        res.status(500).json({ error: '서버 에러' });
    }
});

// 리다이렉트 (GET)
app.get('/:shortId', async (req, res) => {
    try {
        const { shortId } = req.params;
        const urlData = await Url.findOne({ shortId });

        if (urlData) {
            urlData.clicks++;
            await urlData.save();
            return res.redirect(urlData.originalUrl);
        } else {
            return res.status(404).send('<h1>존재하지 않는 URL입니다.</h1>');
        }
    } catch (err) {
        console.error(err);
        res.status(500).send('서버 에러');
    }
});

app.listen(PORT, () => {
    console.log(`🚀 서버 실행 중: ${PORT}`);
});

// server.js의 리다이렉트 부분 찾아서 수정

app.get('/:shortId', async (req, res) => {
    try {
        const { shortId } = req.params;
        const urlData = await Url.findOne({ shortId });

        if (urlData) {
            // [추가] 진짜 IP 알아내기 (Cloudflare 헤더 우선, 없으면 일반 IP)
            const userIp = req.headers['cf-connecting-ip'] || req.headers['x-forwarded-for'] || req.socket.remoteAddress;

            // [추가] DB에 IP와 시간 기록 (배열에 추가)
            urlData.visitHistory.push({ ip: userIp });
            
            urlData.clicks++;
            await urlData.save();
            return res.redirect(urlData.originalUrl);
        } else {
            // ... (에러 처리)
        }
    } catch (err) {
        // ...
    }
});