chore: add crontab ai gen

Author: sunls24

internal/api/api.go

@@ -1 +1,36 @@
 package api
+
+import (
+	"devtools/internal/openai"
+	"strings"
+)
+
+func Crontab(ctx *Context) error {
+	desc := ctx.QueryParam("desc")
+	if strings.TrimSpace(desc) == "" {
+		return nil
+	}
+	resp, err := ctx.oai.Chat(ctx.Request().Context(), openai.ReqChat{
+		Stream:      false,
+		Model:       ctx.cfg.OpenAI.Model,
+		Messages:    buildPrompt(desc),
+		Temperature: 0,
+	})
+	if err != nil {
+		return ctx.ErrMsg(err, "调用 AI 生成异常")
+	}
+	return ctx.Data(resp.Message.Content)
+}
+
+func buildPrompt(desc string) []openai.Message {
+	return []openai.Message{
+		{
+			Role:    openai.RSystem,
+			Content: "You are a professional Crontab generator. Don't do any interpretation, just reply to the expression. If the user sends a crontab, it is returned as is.",
+		},
+		{
+			Role:    openai.RUser,
+			Content: desc,
+		},
+	}
+}

internal/api/context.go

@@ -11,8 +11,8 @@ import (
 
 type Context struct {
 	echo.Context
-	*config.Config
-	*openai.OpenAI
+	cfg *config.Config
+	oai *openai.OpenAI
 }
 
 func Middleware(cfg *config.Config) echo.MiddlewareFunc {

internal/route/route.go

@@ -1,10 +1,11 @@
 package route
 
 import (
+	"devtools/internal/api"
 	"github.com/labstack/echo/v4"
 )
 
 func Register(e *echo.Echo) {
 	g := e.Group("/api")
-	_ = g
+	g.GET("/crontab", api.Wrap(api.Crontab))
 }

internal/utils/utils.go

@@ -9,7 +9,7 @@ import (
 
 func Md5(str string) string {
 	h := md5.New()
-	h.Write([]byte(str))
+	h.Write(Str2Bytes(str))
 	return hex.EncodeToString(h.Sum(nil))
 }
 

web/astro.config.mjs

@@ -11,5 +11,10 @@ export default defineConfig({
   integrations: [react()],
   vite: {
     plugins: [tailwindcss()],
+    server: {
+      proxy: {
+        "/api": "http://127.0.0.1:3000",
+      },
+    },
   },
 })

web/bun.lock

@@ -20,7 +20,7 @@
     "class-variance-authority": "^0.7.1",
     "clsx": "^2.1.1",
     "cron-parser": "^5.3.0",
-    "lucide-react": "^0.518.0",
+    "lucide-react": "^0.522.0",
     "nanostores": "^1.0.1",
     "prettier": "^3.5.3",
     "prettier-plugin-astro": "^0.14.1",

web/package.json

@@ -1,16 +1,19 @@
 import { Button } from "@/components/ui/button.tsx"
 import { Input } from "@/components/ui/input.tsx"
+import { respData } from "@/lib/utils"
 import { CronExpressionParser } from "cron-parser"
-import { Calculator, Timer } from "lucide-react"
+import { Atom, Calculator, Timer } from "lucide-react"
 import React, { useState } from "react"
 import { toast } from "sonner"
 
 function TCrontab() {
   const [input, setInput] = useState("")
   const [list, setList] = useState<string[]>([])
 
-  function onClick() {
-    const expr = input.trim()
+  const [loading, setLoading] = useState(false)
+
+  function onClick(data?: string) {
+    const expr = (data ?? input).trim()
     if (!expr) {
       return
     }
@@ -24,6 +27,22 @@ function TCrontab() {
     }
   }
 
+  function onAIClick() {
+    const desc = input.trim()
+    if (!desc) {
+      return
+    }
+    setLoading(true)
+    fetch(`/api/crontab?desc=${desc}`)
+      .then(respData)
+      .then((data) => {
+        setInput(data)
+        onClick(data)
+      })
+      .catch((err) => toast.error(err.message ?? err))
+      .finally(() => setLoading(false))
+  }
+
   function onKeyDown(e: React.KeyboardEvent<HTMLInputElement>) {
     if (e.key !== "Enter" || e.nativeEvent.isComposing) {
       return
@@ -37,15 +56,19 @@ function TCrontab() {
       <div className="flex flex-col gap-3 sm:flex-row">
         <Input
           value={input}
+          disabled={loading}
           onKeyDown={onKeyDown}
           className="font-mono"
           onChange={(e) => setInput(e.currentTarget.value)}
-          placeholder="请输入 Crontab 表达式"
-        ></Input>
-        <Button variant="secondary" onClick={onClick}>
+          placeholder="Crontab 表达式 / 每两小时执行一次"
+        />
+        <Button onClick={() => onClick()} disabled={loading}>
           <Calculator />
           计算执行时间
         </Button>
+        <Button variant="secondary" onClick={onAIClick} disabled={loading}>
+          <Atom />让 AI 生成
+        </Button>
       </div>
       {list.length > 0 && (
         <div className="flex flex-col gap-2">

web/src/components/tsx/TCrontab.tsx

@@ -12,3 +12,14 @@ export function copyToClipboard(value: string) {
     .then(() => toast.success("已拷贝至剪贴板"))
     .catch((err) => toast.error(err.message ?? err))
 }
+
+export async function respData(resp: Response): Promise<any> {
+  const res = await resp.json()
+  if (!resp.ok) {
+    return Promise.reject(`${resp.status} ${res.message}`)
+  }
+  if (res.code !== 0) {
+    return Promise.reject(res.message)
+  }
+  return res.data
+}