cwd/cwd-api/src/api/public/like.ts

import type { Context } from 'hono';
import type { Bindings } from '../../bindings';
import { decodePostSlug } from '../../utils/decodePostSlug';

type LikeStatusResponse = {
	liked: boolean;
	alreadyLiked: boolean;
	totalLikes: number;
};

type LikeRequestBody = {
	postSlug?: string;
	postTitle?: string;
	postUrl?: string;
	siteId?: string;
};

function getUserIdFromRequest(c: Context<{ Bindings: Bindings }>): string {
	const header =
		c.req.header('X-CWD-Like-User') ||
		c.req.header('x-cwd-like-user') ||
		'';
	const fromHeader = header.trim();
	if (fromHeader) {
		return fromHeader;
	}
	const ip = c.req.header('cf-connecting-ip') || '';
	const trimmedIp = ip.trim();
	if (trimmedIp) {
		return `ip:${trimmedIp}`;
	}
	return 'anonymous';
}

export const getLikeStatus = async (
	c: Context<{ Bindings: Bindings }>
): Promise<Response> => {
	try {
		const rawPostSlug = c.req.query('post_slug') || '';
		const postSlug = decodePostSlug(rawPostSlug);
		const siteId = c.req.query('siteId') || '';

		if (!postSlug) {
			return c.json({ message: 'post_slug is required' }, 400);
		}

		const userIdHeader =
			c.req.header('X-CWD-Like-User') ||
			c.req.header('x-cwd-like-user') ||
			'';
		const userId = userIdHeader.trim();

		const totalRow = await c.env.CWD_DB.prepare(
			'SELECT COUNT(*) AS count FROM Likes WHERE page_slug = ? AND site_id = ?'
		)
			.bind(postSlug, siteId)
			.first<{ count: number }>();

		let liked = false;
		if (userId) {
			const row = await c.env.CWD_DB.prepare(
				'SELECT id FROM Likes WHERE page_slug = ? AND user_id = ? AND site_id = ?'
			)
				.bind(postSlug, userId, siteId)
				.first<{ id: number }>();
			liked = !!row;
		}

		const totalLikes = totalRow?.count || 0;

		const payload: LikeStatusResponse = {
			liked,
			alreadyLiked: false,
			totalLikes
		};

		return c.json(payload);
	} catch (e: any) {
		return c.json(
			{ message: e?.message || '获取点赞状态失败' },
			500
		);
	}
};

export const likePage = async (
	c: Context<{ Bindings: Bindings }>
): Promise<Response> => {
	try {
		const body = ((await c.req
			.json()
			.catch(() => ({}))) || {}) as LikeRequestBody;

		const rawPostSlug =
			typeof body.postSlug === 'string' ? body.postSlug.trim() : '';
		const postSlug = decodePostSlug(rawPostSlug);
		const rawPostTitle =
			typeof body.postTitle === 'string' ? body.postTitle.trim() : '';
		const rawPostUrl =
			typeof body.postUrl === 'string' ? body.postUrl.trim() : '';
		const siteId = typeof body.siteId === 'string' ? body.siteId.trim() : '';

		if (!postSlug) {
			return c.json({ message: 'postSlug is required' }, 400);
		}

		const userId = getUserIdFromRequest(c);

		const now = Date.now();

		const existingLike = await c.env.CWD_DB.prepare(
			'SELECT id FROM Likes WHERE page_slug = ? AND user_id = ? AND site_id = ?'
		)
			.bind(postSlug, userId, siteId)
			.first<{ id: number }>();

		let alreadyLiked = false;

		if (!existingLike) {
			await c.env.CWD_DB.prepare(
				'INSERT INTO Likes (page_slug, user_id, created_at, site_id) VALUES (?, ?, ?, ?)'
			)
				.bind(postSlug, userId, now, siteId)
				.run();
		} else {
			alreadyLiked = true;
		}

		const pageStatsRow = await c.env.CWD_DB.prepare(
			'SELECT id FROM page_stats WHERE post_slug = ? AND site_id = ?'
		)
			.bind(postSlug, siteId)
			.first<{ id: number }>();

		if (!pageStatsRow) {
			await c.env.CWD_DB.prepare(
				'INSERT INTO page_stats (post_slug, post_title, post_url, pv, last_visit_at, created_at, updated_at, site_id) VALUES (?, ?, ?, ?, ?, ?, ?, ?)'
			)
				.bind(
					postSlug,
					rawPostTitle || null,
					rawPostUrl || null,
					0,
					now,
					now,
					now,
					siteId
				)
				.run();
		} else if (rawPostTitle || rawPostUrl) {
			await c.env.CWD_DB.prepare(
				'UPDATE page_stats SET post_title = COALESCE(?, post_title), post_url = COALESCE(?, post_url), updated_at = ? WHERE id = ?'
			)
				.bind(
					rawPostTitle || null,
					rawPostUrl || null,
					now,
					pageStatsRow.id
				)
				.run();
		}

		const totalRow = await c.env.CWD_DB.prepare(
			'SELECT COUNT(*) AS count FROM Likes WHERE page_slug = ? AND site_id = ?'
		)
			.bind(postSlug, siteId)
			.first<{ count: number }>();

		const totalLikes = totalRow?.count || 0;

		const payload: LikeStatusResponse = {
			liked: true,
			alreadyLiked,
			totalLikes
		};

		return c.json(payload);
	} catch (e: any) {
		return c.json(
			{ message: e?.message || '点赞失败' },
			500
		);
	}
};