Code-MCP

import Redis from "ioredis"; // ============================================ // Configuration // ============================================ const redis = new Redis({ host: process.env.REDIS_HOST || "localhost", port: parseInt(process.env.REDIS_PORT || "6379"), password: process.env.REDIS_PASSWORD, db: parseInt(process.env.REDIS_DB || "0"), retryStrategy: (times) => Math.min(times * 50, 2000), }); redis.on("connect", () => console.log("Redis connected")); redis.on("error", (err) => console.error("Redis error:", err)); // ============================================ // Basic Operations // ============================================ export async function set( key: string, value: unknown, ttlSeconds?: number, ): Promise<void> { const serialized = JSON.stringify(value); if (ttlSeconds) { await redis.setex(key, ttlSeconds, serialized); } else { await redis.set(key, serialized); } } export async function get<T>(key: string): Promise<T | null> { const value = await redis.get(key); return value ? JSON.parse(value) : null; } export async function del(key: string): Promise<boolean> { const result = await redis.del(key); return result > 0; } export async function exists(key: string): Promise<boolean> { const result = await redis.exists(key); return result === 1; } // ============================================ // Cache Helper // ============================================ export async function cached<T>( key: string, ttlSeconds: number, fetchFn: () => Promise<T>, ): Promise<T> { const cached = await get<T>(key); if (cached !== null) { return cached; } const fresh = await fetchFn(); await set(key, fresh, ttlSeconds); return fresh; } // ============================================ // Rate Limiting // ============================================ export interface RateLimitResult { allowed: boolean; remaining: number; resetIn: number; } export async function rateLimit( key: string, maxRequests: number, windowSeconds: number, ): Promise<RateLimitResult> { const current = await redis.incr(key); if (current === 1) { await redis.expire(key, windowSeconds); } const ttl = await redis.ttl(key); return { allowed: current <= maxRequests, remaining: Math.max(0, maxRequests - current), resetIn: ttl > 0 ? ttl : windowSeconds, }; } // ============================================ // Session Store // ============================================ export interface Session { userId: string; data: Record<string, unknown>; createdAt: number; } const SESSION_PREFIX = "session:"; const SESSION_TTL = 86400; // 24 hours export const sessionStore = { async create( sessionId: string, userId: string, data: Record<string, unknown> = {}, ): Promise<void> { const session: Session = { userId, data, createdAt: Date.now(), }; await set(`${SESSION_PREFIX}${sessionId}`, session, SESSION_TTL); }, async get(sessionId: string): Promise<Session | null> { return get<Session>(`${SESSION_PREFIX}${sessionId}`); }, async update( sessionId: string, data: Record<string, unknown>, ): Promise<void> { const session = await this.get(sessionId); if (session) { session.data = { ...session.data, ...data }; await set(`${SESSION_PREFIX}${sessionId}`, session, SESSION_TTL); } }, async destroy(sessionId: string): Promise<boolean> { return del(`${SESSION_PREFIX}${sessionId}`); }, async refresh(sessionId: string): Promise<void> { await redis.expire(`${SESSION_PREFIX}${sessionId}`, SESSION_TTL); }, }; // ============================================ // Pub/Sub // ============================================ const subscriber = redis.duplicate(); type MessageHandler = (message: string, channel: string) => void; const handlers = new Map<string, MessageHandler[]>(); subscriber.on("message", (channel, message) => { const channelHandlers = handlers.get(channel) || []; channelHandlers.forEach((handler) => handler(message, channel)); }); export async function subscribe( channel: string, handler: MessageHandler, ): Promise<void> { const existing = handlers.get(channel) || []; handlers.set(channel, [...existing, handler]); if (existing.length === 0) { await subscriber.subscribe(channel); } } export async function publish( channel: string, message: unknown, ): Promise<number> { const serialized = typeof message === "string" ? message : JSON.stringify(message); return redis.publish(channel, serialized); } // ============================================ // Queue (Simple Job Queue) // ============================================ export interface Job<T = unknown> { id: string; type: string; data: T; createdAt: number; attempts: number; } const QUEUE_PREFIX = "queue:"; export const queue = { async push<T>(name: string, type: string, data: T): Promise<string> { const job: Job<T> = { id: `${Date.now()}-${Math.random().toString(36).substr(2, 9)}`, type, data, createdAt: Date.now(), attempts: 0, }; await redis.lpush(`${QUEUE_PREFIX}${name}`, JSON.stringify(job)); return job.id; }, async pop<T>(name: string, timeout = 0): Promise<Job<T> | null> { const result = await redis.brpop(`${QUEUE_PREFIX}${name}`, timeout); if (!result) return null; const job = JSON.parse(result[1]) as Job<T>; job.attempts++; return job; }, async size(name: string): Promise<number> { return redis.llen(`${QUEUE_PREFIX}${name}`); }, async clear(name: string): Promise<void> { await redis.del(`${QUEUE_PREFIX}${name}`); }, }; // ============================================ // Leaderboard // ============================================ export const leaderboard = { async addScore(name: string, memberId: string, score: number): Promise<void> { await redis.zadd(name, score, memberId); }, async incrementScore( name: string, memberId: string, increment: number, ): Promise<number> { return redis.zincrby(name, increment, memberId); }, async getTop( name: string, count = 10, ): Promise<Array<{ memberId: string; score: number }>> { const results = await redis.zrevrange(name, 0, count - 1, "WITHSCORES"); const entries: Array<{ memberId: string; score: number }> = []; for (let i = 0; i < results.length; i += 2) { entries.push({ memberId: results[i], score: parseFloat(results[i + 1]), }); } return entries; }, async getRank(name: string, memberId: string): Promise<number | null> { const rank = await redis.zrevrank(name, memberId); return rank !== null ? rank + 1 : null; }, async getScore(name: string, memberId: string): Promise<number | null> { const score = await redis.zscore(name, memberId); return score !== null ? parseFloat(score) : null; }, }; // ============================================ // Exports // ============================================ export { redis }; export default redis;