Telegram Bot

export const handleStart = (api: Api) => async (ctx: Context) => {
  const chatId = ctx.chat?.id;
  const userId = ctx.from?.id;
  
  if (!chatId || !userId) {
    await ctx.reply("Unable to start: missing chat information.");
    return;
  }
  
  const chatIdStr = String(chatId);
  
  // Check if user already exists
  const userResult = await Effect.runPromise(
    Effect.catchAll(api.getUserByChatId(chatIdStr), (error) => {
      if (isApiError(error) && error.type === "not_found") {
        return Effect.succeed(null);  // User doesn't exist
      }
      return Effect.succeed(null);
    })
  );
  
  // Register new user if not found
  if (!userResult) {
    const registerResult = await Effect.runPromise(
      api.registerTelegramUser(chatIdStr, userId)
    );
    
    if (!registerResult) {
      await ctx.reply("❌ Registration failed. Please try again.");
      return;
    }
  }
  
  const welcomeMsg =
    "🚀 Welcome to Celebrum AI!\n\n" +
    "✅ You're now registered and ready to receive arbitrage alerts!\n\n" +
    "Use /opportunities to see current opportunities.\n" +
    "Use /help to see available commands.";
  
  await ctx.reply(welcomeMsg);
};

export const formatOpportunitiesMessage = (opps: any[]) => {
  if (!opps || opps.length === 0) {
    return "📊 No arbitrage opportunities found right now.";
  }
  
  const top = opps.slice(0, 5);
  const lines = ["⚡ Top Arbitrage Opportunities", ""];
  
  top.forEach((opp, index) => {
    lines.push(`${index + 1}. ${opp.symbol}`);
    lines.push(`   Buy: ${opp.buy_exchange} @ ${opp.buy_price}`);
    lines.push(`   Sell: ${opp.sell_exchange} @ ${opp.sell_price}`);
    lines.push(`   Profit: ${Number(opp.profit_percent).toFixed(2)}%`);
    lines.push("");
  });
  
  return lines.join("\n");
};

export const handleStatus = (api: Api) => async (ctx: Context) => {
  const chatId = ctx.chat?.id;
  
  // Fetch user info
  const userResult = await Effect.runPromise(
    Effect.catchAll(api.getUserByChatId(String(chatId)), (error) => {
      if (error.type === "not_found") {
        return Effect.succeed(null);
      }
      return Effect.succeed(null);
    })
  );
  
  if (!userResult) {
    await ctx.reply(
      "👤 User not found in our system.\n\n" +
      "Use /start to register."
    );
    return;
  }
  
  // Get notification preferences
  const preference = await Effect.runPromise(
    Effect.catchAll(api.getNotificationPreference(String(userId)), () =>
      Effect.succeed({ enabled: true, profit_threshold: 0.5 })
    )
  );
  
  const msg =
    "📊 Account Status:\n\n" +
    `💰 Subscription: ${userResult.user.subscription_tier}\n` +
    `📅 Member since: ${createdAt}\n` +
    `🔔 Notifications: ${preference.enabled ? "Active" : "Paused"}`;
  
  await ctx.reply(msg);
};

export const handleMode = (api: Api) => async (ctx: Context) => {
  const chatId = ctx.chat?.id;
  
  // Get current mode from API
  const modeResult = await Effect.runPromise(
    Effect.catchAll(api.getTradingMode(String(chatId)), (error) => {
      return Effect.succeed({
        mode: "dry",
        confirmations: 0,
        required_confirmations: 2,
      });
    })
  );
  
  const mode = modeResult.mode || "dry";
  const confirmations = modeResult.confirmations || 0;
  const required = modeResult.required_confirmations || 2;
  
  let msg: string;
  if (mode === "dry") {
    msg =
      "🧪 Current Mode: DRY (Paper Trading)\n\n" +
      "• No real orders executed\n" +
      "• Safe for testing\n\n" +
      `Confirmations: ${confirmations}/${required}\n\n` +
      "Commands:\n" +
      "/mode confirm - Add confirmation\n" +
      "/mode live - Switch to live (requires confirmations)\n" +
      "/mode dry - Switch to dry mode";
  } else {
    msg =
      "🔴 Current Mode: LIVE (Real Trading)\n\n" +
      "⚠️ REAL ORDERS WILL BE EXECUTED\n" +
      "⚠️ REAL MONEY IS AT RISK\n\n" +
      "Commands:\n" +
      "/mode dry - Switch to safe dry mode";
  }
  
  await ctx.reply(msg);
};

else if (action === "confirm") {
  const result = await Effect.runPromise(
    Effect.catchAll(api.addTradingModeConfirmation(String(chatId)), (error) => {
      return Effect.succeed({ confirmations: 0, required: 2 });
    })
  );
  
  if (result.confirmations >= result.required) {
    await ctx.reply(
      `✅ Confirmation ${result.confirmations}/${result.required}\n\n` +
      "You have enough confirmations!\n" +
      "Use /mode live to switch to live trading.\n\n" +
      "⚠️ Remember: Live mode uses real money!"
    );
  } else {
    await ctx.reply(
      `✅ Confirmation ${result.confirmations}/${result.required}\n\n` +
      "More confirmations needed before live trading.\n" +
      "Use /mode confirm again to add another confirmation."
    );
  }
}

const ADMIN_API_KEY = process.env.ADMIN_API_KEY;
const API_BASE_URL = process.env.TELEGRAM_API_BASE_URL;

const headers = {
  "Content-Type": "application/json",
  "X-Admin-Key": ADMIN_API_KEY,
};

service TelegramService {
  rpc SendMessage(SendMessageRequest) returns (SendMessageResponse);
  rpc Health(HealthRequest) returns (HealthResponse);
}

message SendMessageRequest {
  int64 chat_id = 1;
  string message = 2;
  string parse_mode = 3;  // "Markdown" or "HTML"
}

try {
  const response = await Effect.runPromise(api.getOpportunities());
  await ctx.reply(formatOpportunitiesMessage(response.opportunities));
} catch (error) {
  let errorMessage = "An unexpected error occurred.";
  
  const apiError = extractApiError(error);
  if (apiError) {
    errorMessage =
      apiError.type === "network_error"
        ? "Unable to connect to the server."
        : apiError.type === "server_error"
        ? "Server is temporarily unavailable."
        : apiError.message;
  }
  
  await ctx.reply(
    `❌ Failed to fetch opportunities.\n\n${errorMessage}\n\nPlease try again later.`
  );
}