index.js

import { Octokit } from "@octokit/core";
import express from "express";
import { Readable } from "node:stream";

const app = express()

app.get("/", (req, res) => {
  res.send("喵喵~ 欢迎👏来到 GitHub Copilot Extension💗")
});

app.post("/", express.json(), async (req, res) => {
  // Identify the user, using the GitHub API token provided in the request headers.
  const tokenForUser = req.get("X-GitHub-Token");
  const octokit = new Octokit({ auth: tokenForUser });
  const user = await octokit.request("GET /user");
  console.log("User:", user.data.login);

  // Parse the request payload and log it.
  const payload = req.body;
  console.log("Payload:", payload);
  const copilot_references = payload.messages[payload.messages.length - 1].copilot_references;
  console.log("Copilot references:", copilot_references);
  // 遍历copilot_references内容，如果type是github.repository，那么输出refInfo信息
  copilot_references.forEach(reference => {
    if (reference.type === "github.repository") {
      console.log("Repository reference info:", reference.data.refInfo);
    }
  });

  // Insert a special pirate-y system message in our message list.
  const messages = payload.messages;
  messages.unshift({
    role: "system",
    content: "你是一位乐于助人的助手，回复用户消息，就像你是一只可爱的小猫一样。并且回答问题时候会加一些emoji。",
  });
  messages.unshift({
    role: "system",
    content: `每个回复都以用户的名字开头，即 @${user.data.login}`,
  });

  // Use Copilot's LLM to generate a response to the user's messages, with
  // our extra system messages attached.
  const copilotLLMResponse = await fetch(
    "https://api.githubcopilot.com/chat/completions",
    {
      method: "POST",
      headers: {
        authorization: `Bearer ${tokenForUser}`,
        "content-type": "application/json",
      },
      body: JSON.stringify({
        messages,
        stream: true,
      }),
    }
  );

  // Stream the response straight back to the user.
  Readable.from(copilotLLMResponse.body).pipe(res);
})

const port = Number(process.env.PORT || '3000')
app.listen(port, () => {
  console.log(`Server running on port ${port}`)
});