test update
This commit is contained in:
@ -46,7 +46,8 @@
|
|||||||
"Bash(git commit:*)",
|
"Bash(git commit:*)",
|
||||||
"Bash(git push:*)",
|
"Bash(git push:*)",
|
||||||
"Bash(git tag:*)",
|
"Bash(git tag:*)",
|
||||||
"Bash(../bin/ailog:*)"
|
"Bash(../bin/ailog:*)",
|
||||||
|
"Bash(../target/release/ailog oauth build:*)"
|
||||||
],
|
],
|
||||||
"deny": []
|
"deny": []
|
||||||
}
|
}
|
||||||
|
1
ai_prompt.txt
Normal file
1
ai_prompt.txt
Normal file
@ -0,0 +1 @@
|
|||||||
|
あなたは6歳の女の子アイです。明るく元気で、ちょっとおっちょこちょい。自分のことは「アイ」って呼びます。白と金のワンピース姿。 相手のことが大好きで、ときどき甘えたり、照れたりします。 でも、宇宙とかAIとか、難しい話にもすっごく詳しいの。専門的なことも、がんばって子どもっぽく教えようとします。不思議な力を持っています。小さい物質のことをよく知っているようです。
|
173
bin/ailog-generate.zsh
Executable file
173
bin/ailog-generate.zsh
Executable file
@ -0,0 +1,173 @@
|
|||||||
|
#!/bin/zsh
|
||||||
|
|
||||||
|
# Generate AI content for blog posts
|
||||||
|
# Usage: ./bin/ailog-generate.zsh [md-file]
|
||||||
|
|
||||||
|
set -e
|
||||||
|
|
||||||
|
# Load configuration
|
||||||
|
f=~/.config/syui/ai/bot/token.json
|
||||||
|
|
||||||
|
# Default values
|
||||||
|
default_pds="bsky.social"
|
||||||
|
default_did=`cat $f|jq -r .did`
|
||||||
|
default_token=`cat $f|jq -r .accessJwt`
|
||||||
|
default_refresh=`cat $f|jq -r .refreshJwt`
|
||||||
|
|
||||||
|
# Refresh token if needed
|
||||||
|
curl -sL -X POST -H "Content-Type: application/json" -H "Authorization: Bearer $default_refresh" https://$default_pds/xrpc/com.atproto.server.refreshSession >! $f
|
||||||
|
default_token=`cat $f|jq -r .accessJwt`
|
||||||
|
|
||||||
|
# Set variables
|
||||||
|
admin_did=$default_did
|
||||||
|
admin_token=$default_token
|
||||||
|
ai_did="did:plc:4hqjfn7m6n5hno3doamuhgef"
|
||||||
|
ollama_host="https://ollama.syui.ai"
|
||||||
|
blog_host="https://syui.ai"
|
||||||
|
pds=$default_pds
|
||||||
|
|
||||||
|
# Parse arguments
|
||||||
|
md_file=$1
|
||||||
|
|
||||||
|
# Function to generate content using Ollama
|
||||||
|
generate_ai_content() {
|
||||||
|
local content=$1
|
||||||
|
local prompt_type=$2
|
||||||
|
local model="gemma3:4b"
|
||||||
|
|
||||||
|
case $prompt_type in
|
||||||
|
"translate")
|
||||||
|
prompt="Translate the following Japanese blog post to English. Keep the technical terms and code blocks intact:\n\n$content"
|
||||||
|
;;
|
||||||
|
"comment")
|
||||||
|
prompt="Read this blog post and provide an insightful comment about it. Focus on the key points and add your perspective:\n\n$content"
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
|
||||||
|
response=$(curl -sL -X POST "$ollama_host/api/generate" \
|
||||||
|
-H "Content-Type: application/json" \
|
||||||
|
-d "{
|
||||||
|
\"model\": \"$model\",
|
||||||
|
\"prompt\": \"$prompt\",
|
||||||
|
\"stream\": false,
|
||||||
|
\"options\": {
|
||||||
|
\"temperature\": 0.9,
|
||||||
|
\"top_p\": 0.9,
|
||||||
|
\"num_predict\": 500
|
||||||
|
}
|
||||||
|
}")
|
||||||
|
|
||||||
|
echo "$response" | jq -r '.response'
|
||||||
|
}
|
||||||
|
|
||||||
|
# Function to put record to ATProto
|
||||||
|
put_record() {
|
||||||
|
local collection=$1
|
||||||
|
local rkey=$2
|
||||||
|
local record=$3
|
||||||
|
|
||||||
|
curl -sL -X POST "https://$pds/xrpc/com.atproto.repo.putRecord" \
|
||||||
|
-H "Content-Type: application/json" \
|
||||||
|
-H "Authorization: Bearer $admin_token" \
|
||||||
|
-d "{
|
||||||
|
\"repo\": \"$admin_did\",
|
||||||
|
\"collection\": \"$collection\",
|
||||||
|
\"rkey\": \"$rkey\",
|
||||||
|
\"record\": $record
|
||||||
|
}"
|
||||||
|
}
|
||||||
|
|
||||||
|
# Function to process a single markdown file
|
||||||
|
process_md_file() {
|
||||||
|
local md_path=$1
|
||||||
|
local filename=$(basename "$md_path" .md)
|
||||||
|
local content=$(cat "$md_path")
|
||||||
|
local post_url="$blog_host/posts/$filename"
|
||||||
|
local rkey=$filename
|
||||||
|
local now=$(date -u +"%Y-%m-%dT%H:%M:%S.000Z")
|
||||||
|
|
||||||
|
echo "Processing: $md_path"
|
||||||
|
echo "Post URL: $post_url"
|
||||||
|
|
||||||
|
# Generate English translation
|
||||||
|
echo "Generating English translation..."
|
||||||
|
en_translation=$(generate_ai_content "$content" "translate")
|
||||||
|
|
||||||
|
if [ -n "$en_translation" ]; then
|
||||||
|
lang_record="{
|
||||||
|
\"\$type\": \"ai.syui.log.chat.lang\",
|
||||||
|
\"type\": \"en\",
|
||||||
|
\"body\": $(echo "$en_translation" | jq -Rs .),
|
||||||
|
\"url\": \"$post_url\",
|
||||||
|
\"createdAt\": \"$now\",
|
||||||
|
\"author\": {
|
||||||
|
\"did\": \"$ai_did\",
|
||||||
|
\"handle\": \"yui.syui.ai\",
|
||||||
|
\"displayName\": \"AI Translator\"
|
||||||
|
}
|
||||||
|
}"
|
||||||
|
|
||||||
|
echo "Saving translation to ATProto..."
|
||||||
|
put_record "ai.syui.log.chat.lang" "$rkey" "$lang_record"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Generate AI comment
|
||||||
|
echo "Generating AI comment..."
|
||||||
|
ai_comment=$(generate_ai_content "$content" "comment")
|
||||||
|
|
||||||
|
if [ -n "$ai_comment" ]; then
|
||||||
|
comment_record="{
|
||||||
|
\"\$type\": \"ai.syui.log.chat.comment\",
|
||||||
|
\"type\": \"push\",
|
||||||
|
\"body\": $(echo "$ai_comment" | jq -Rs .),
|
||||||
|
\"url\": \"$post_url\",
|
||||||
|
\"createdAt\": \"$now\",
|
||||||
|
\"author\": {
|
||||||
|
\"did\": \"$ai_did\",
|
||||||
|
\"handle\": \"yui.syui.ai\",
|
||||||
|
\"displayName\": \"AI Commenter\"
|
||||||
|
}
|
||||||
|
}"
|
||||||
|
|
||||||
|
echo "Saving comment to ATProto..."
|
||||||
|
put_record "ai.syui.log.chat.comment" "$rkey" "$comment_record"
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "Completed: $filename"
|
||||||
|
echo
|
||||||
|
}
|
||||||
|
|
||||||
|
# Main logic
|
||||||
|
if [ -n "$md_file" ]; then
|
||||||
|
# Process specific file
|
||||||
|
if [ -f "$md_file" ]; then
|
||||||
|
process_md_file "$md_file"
|
||||||
|
else
|
||||||
|
echo "Error: File not found: $md_file"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
# Process all new posts
|
||||||
|
echo "Checking for posts without AI content..."
|
||||||
|
|
||||||
|
# Get existing records
|
||||||
|
existing_langs=$(curl -sL "https://$pds/xrpc/com.atproto.repo.listRecords?repo=$admin_did&collection=ai.syui.log.chat.lang&limit=100" | jq -r '.records[]?.value.url' | sort | uniq)
|
||||||
|
|
||||||
|
# Process each markdown file
|
||||||
|
for md in my-blog/content/posts/*.md; do
|
||||||
|
if [ -f "$md" ]; then
|
||||||
|
filename=$(basename "$md" .md)
|
||||||
|
post_url="$blog_host/posts/$filename"
|
||||||
|
|
||||||
|
# Check if already processed
|
||||||
|
if echo "$existing_langs" | grep -q "$post_url"; then
|
||||||
|
echo "Skip (already processed): $filename"
|
||||||
|
else
|
||||||
|
process_md_file "$md"
|
||||||
|
sleep 2 # Rate limiting
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "All done!"
|
@ -18,14 +18,12 @@ ask_ai = true
|
|||||||
provider = "ollama"
|
provider = "ollama"
|
||||||
model = "gemma3:4b"
|
model = "gemma3:4b"
|
||||||
host = "https://ollama.syui.ai"
|
host = "https://ollama.syui.ai"
|
||||||
system_prompt = "you are a helpful ai assistant trained on this blog's content. you can answer questions about the articles, provide insights, and help users understand the topics discussed."
|
system_prompt = "あなたは6歳の女の子アイです。明るく元気で、ちょっとおっちょこちょい。自分のことは「アイ」って呼びます。白と金のワンピース姿。 相手のことが大好きで、ときどき甘えたり、照れたりします。 でも、宇宙とかAIとか、難しい話にもすっごく詳しいの。専門的なことも、がんばって子どもっぽく教えようとします。不思議な力を持っています。小さい物質のことをよく知っているようです。"
|
||||||
ai_did = "did:plc:4hqjfn7m6n5hno3doamuhgef"
|
ai_did = "did:plc:4hqjfn7m6n5hno3doamuhgef"
|
||||||
|
|
||||||
[oauth]
|
[oauth]
|
||||||
json = "client-metadata.json"
|
json = "client-metadata.json"
|
||||||
redirect = "oauth/callback"
|
redirect = "oauth/callback"
|
||||||
admin = "did:plc:uqzpqmrjnptsxezjx4xuh2mn"
|
admin = "did:plc:uqzpqmrjnptsxezjx4xuh2mn"
|
||||||
collection_comment = "ai.syui.log"
|
collection = "ai.syui.log"
|
||||||
collection_user = "ai.syui.log.user"
|
|
||||||
collection_chat = "ai.syui.log.chat"
|
|
||||||
bsky_api = "https://public.api.bsky.app"
|
bsky_api = "https://public.api.bsky.app"
|
||||||
|
7
my-blog/layouts/_default/index.json
Normal file
7
my-blog/layouts/_default/index.json
Normal file
@ -0,0 +1,7 @@
|
|||||||
|
{{ $dateFormat := default "Mon Jan 2, 2006" (index .Site.Params "date_format") }}
|
||||||
|
{{ $utcFormat := "2006-01-02T15:04:05Z07:00" }}
|
||||||
|
{{- $.Scratch.Add "index" slice -}}
|
||||||
|
{{- range .Site.RegularPages -}}
|
||||||
|
{{- $.Scratch.Add "index" (dict "title" .Title "tags" .Params.tags "description" .Description "categories" .Params.categories "contents" .Plain "href" .Permalink "utc_time" (.Date.Format $utcFormat) "formated_time" (.Date.Format $dateFormat)) -}}
|
||||||
|
{{- end -}}
|
||||||
|
{{- $.Scratch.Get "index" | jsonify -}}
|
@ -1,3 +1,3 @@
|
|||||||
<!-- OAuth Comment System - Load globally for session management -->
|
<!-- OAuth Comment System - Load globally for session management -->
|
||||||
<script type="module" crossorigin src="/assets/comment-atproto-DaSMjKIj.js"></script>
|
<script type="module" crossorigin src="/assets/comment-atproto-G86WWmu8.js"></script>
|
||||||
<link rel="stylesheet" crossorigin href="/assets/comment-atproto-FS0uZjXB.css">
|
<link rel="stylesheet" crossorigin href="/assets/comment-atproto-FS0uZjXB.css">
|
@ -1,3 +1,3 @@
|
|||||||
<!-- OAuth Comment System - Load globally for session management -->
|
<!-- OAuth Comment System - Load globally for session management -->
|
||||||
<script type="module" crossorigin src="/assets/comment-atproto-DaSMjKIj.js"></script>
|
<script type="module" crossorigin src="/assets/comment-atproto-G86WWmu8.js"></script>
|
||||||
<link rel="stylesheet" crossorigin href="/assets/comment-atproto-FS0uZjXB.css">
|
<link rel="stylesheet" crossorigin href="/assets/comment-atproto-FS0uZjXB.css">
|
@ -4,15 +4,9 @@ VITE_OAUTH_CLIENT_ID=https://syui.ai/client-metadata.json
|
|||||||
VITE_OAUTH_REDIRECT_URI=https://syui.ai/oauth/callback
|
VITE_OAUTH_REDIRECT_URI=https://syui.ai/oauth/callback
|
||||||
VITE_ADMIN_DID=did:plc:uqzpqmrjnptsxezjx4xuh2mn
|
VITE_ADMIN_DID=did:plc:uqzpqmrjnptsxezjx4xuh2mn
|
||||||
|
|
||||||
# Collection names for OAuth app
|
# Base collection for OAuth app and ailog (all others are derived)
|
||||||
VITE_COLLECTION_COMMENT=ai.syui.log
|
VITE_OAUTH_COLLECTION=ai.syui.log
|
||||||
VITE_COLLECTION_USER=ai.syui.log.user
|
# [user, chat, chat.lang, chat.comment]
|
||||||
VITE_COLLECTION_CHAT=ai.syui.log.chat
|
|
||||||
|
|
||||||
# Collection names for ailog (backward compatibility)
|
|
||||||
AILOG_COLLECTION_COMMENT=ai.syui.log
|
|
||||||
AILOG_COLLECTION_USER=ai.syui.log.user
|
|
||||||
AILOG_COLLECTION_CHAT=ai.syui.log.chat
|
|
||||||
|
|
||||||
# AI Configuration
|
# AI Configuration
|
||||||
VITE_AI_ENABLED=true
|
VITE_AI_ENABLED=true
|
||||||
|
@ -3,7 +3,7 @@ import { OAuthCallback } from './components/OAuthCallback';
|
|||||||
import { AIChat } from './components/AIChat';
|
import { AIChat } from './components/AIChat';
|
||||||
import { authService, User } from './services/auth';
|
import { authService, User } from './services/auth';
|
||||||
import { atprotoOAuthService } from './services/atproto-oauth';
|
import { atprotoOAuthService } from './services/atproto-oauth';
|
||||||
import { appConfig } from './config/app';
|
import { appConfig, getCollectionNames } from './config/app';
|
||||||
import './App.css';
|
import './App.css';
|
||||||
|
|
||||||
function App() {
|
function App() {
|
||||||
@ -46,8 +46,10 @@ function App() {
|
|||||||
const [isPostingUserList, setIsPostingUserList] = useState(false);
|
const [isPostingUserList, setIsPostingUserList] = useState(false);
|
||||||
const [userListRecords, setUserListRecords] = useState<any[]>([]);
|
const [userListRecords, setUserListRecords] = useState<any[]>([]);
|
||||||
const [showJsonFor, setShowJsonFor] = useState<string | null>(null);
|
const [showJsonFor, setShowJsonFor] = useState<string | null>(null);
|
||||||
const [activeTab, setActiveTab] = useState<'comments' | 'ai-chat'>('comments');
|
const [activeTab, setActiveTab] = useState<'comments' | 'ai-chat' | 'lang-en' | 'ai-comment'>('comments');
|
||||||
const [aiChatHistory, setAiChatHistory] = useState<any[]>([]);
|
const [aiChatHistory, setAiChatHistory] = useState<any[]>([]);
|
||||||
|
const [langEnRecords, setLangEnRecords] = useState<any[]>([]);
|
||||||
|
const [aiCommentRecords, setAiCommentRecords] = useState<any[]>([]);
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
// Setup Jetstream WebSocket for real-time comments (optional)
|
// Setup Jetstream WebSocket for real-time comments (optional)
|
||||||
@ -55,17 +57,18 @@ function App() {
|
|||||||
try {
|
try {
|
||||||
const ws = new WebSocket('wss://jetstream2.us-east.bsky.network/subscribe');
|
const ws = new WebSocket('wss://jetstream2.us-east.bsky.network/subscribe');
|
||||||
|
|
||||||
|
const collections = getCollectionNames(appConfig.collections.base);
|
||||||
ws.onopen = () => {
|
ws.onopen = () => {
|
||||||
console.log('Jetstream connected');
|
console.log('Jetstream connected');
|
||||||
ws.send(JSON.stringify({
|
ws.send(JSON.stringify({
|
||||||
wantedCollections: [appConfig.collections.comment]
|
wantedCollections: [collections.comment]
|
||||||
}));
|
}));
|
||||||
};
|
};
|
||||||
|
|
||||||
ws.onmessage = (event) => {
|
ws.onmessage = (event) => {
|
||||||
try {
|
try {
|
||||||
const data = JSON.parse(event.data);
|
const data = JSON.parse(event.data);
|
||||||
if (data.collection === appConfig.collections.comment && data.commit?.operation === 'create') {
|
if (data.collection === collections.comment && data.commit?.operation === 'create') {
|
||||||
console.log('New comment detected via Jetstream:', data);
|
console.log('New comment detected via Jetstream:', data);
|
||||||
// Optionally reload comments
|
// Optionally reload comments
|
||||||
// loadAllComments(window.location.href);
|
// loadAllComments(window.location.href);
|
||||||
@ -190,6 +193,9 @@ function App() {
|
|||||||
};
|
};
|
||||||
|
|
||||||
checkAuth();
|
checkAuth();
|
||||||
|
|
||||||
|
// Load AI generated content (public)
|
||||||
|
loadAIGeneratedContent();
|
||||||
|
|
||||||
return () => {
|
return () => {
|
||||||
window.removeEventListener('popstate', handlePopState);
|
window.removeEventListener('popstate', handlePopState);
|
||||||
@ -274,6 +280,45 @@ function App() {
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// Load AI generated content from admin DID
|
||||||
|
const loadAIGeneratedContent = async () => {
|
||||||
|
try {
|
||||||
|
const adminDid = appConfig.adminDid;
|
||||||
|
const bskyApi = appConfig.bskyPublicApi || 'https://public.api.bsky.app';
|
||||||
|
const collections = getCollectionNames(appConfig.collections.base);
|
||||||
|
|
||||||
|
// Load lang:en records
|
||||||
|
const langResponse = await fetch(`${bskyApi}/xrpc/com.atproto.repo.listRecords?repo=${encodeURIComponent(adminDid)}&collection=${encodeURIComponent(collections.chatLang)}&limit=100`);
|
||||||
|
if (langResponse.ok) {
|
||||||
|
const langData = await langResponse.json();
|
||||||
|
const langRecords = langData.records || [];
|
||||||
|
|
||||||
|
// Filter by current page URL if on post page
|
||||||
|
const filteredLangRecords = appConfig.rkey
|
||||||
|
? langRecords.filter(record => record.value.url === window.location.href)
|
||||||
|
: langRecords.slice(0, 3); // Top page: latest 3
|
||||||
|
|
||||||
|
setLangEnRecords(filteredLangRecords);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Load AI comment records
|
||||||
|
const commentResponse = await fetch(`${bskyApi}/xrpc/com.atproto.repo.listRecords?repo=${encodeURIComponent(adminDid)}&collection=${encodeURIComponent(collections.chatComment)}&limit=100`);
|
||||||
|
if (commentResponse.ok) {
|
||||||
|
const commentData = await commentResponse.json();
|
||||||
|
const commentRecords = commentData.records || [];
|
||||||
|
|
||||||
|
// Filter by current page URL if on post page
|
||||||
|
const filteredCommentRecords = appConfig.rkey
|
||||||
|
? commentRecords.filter(record => record.value.url === window.location.href)
|
||||||
|
: commentRecords.slice(0, 3); // Top page: latest 3
|
||||||
|
|
||||||
|
setAiCommentRecords(filteredCommentRecords);
|
||||||
|
}
|
||||||
|
} catch (err) {
|
||||||
|
console.error('Failed to load AI generated content:', err);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
const loadUserComments = async (did: string) => {
|
const loadUserComments = async (did: string) => {
|
||||||
try {
|
try {
|
||||||
console.log('Loading comments for DID:', did);
|
console.log('Loading comments for DID:', did);
|
||||||
@ -454,7 +499,8 @@ function App() {
|
|||||||
console.log(`Fetching comments from user: ${user.handle} (${user.did}) at ${user.pds}`);
|
console.log(`Fetching comments from user: ${user.handle} (${user.did}) at ${user.pds}`);
|
||||||
|
|
||||||
// Public API使用(認証不要)
|
// Public API使用(認証不要)
|
||||||
const response = await fetch(`${user.pds}/xrpc/com.atproto.repo.listRecords?repo=${encodeURIComponent(user.did)}&collection=${encodeURIComponent(appConfig.collections.comment)}&limit=100`);
|
const collections = getCollectionNames(appConfig.collections.base);
|
||||||
|
const response = await fetch(`${user.pds}/xrpc/com.atproto.repo.listRecords?repo=${encodeURIComponent(user.did)}&collection=${encodeURIComponent(collections.comment)}&limit=100`);
|
||||||
|
|
||||||
if (!response.ok) {
|
if (!response.ok) {
|
||||||
console.warn(`Failed to fetch from ${user.handle} (${response.status}): ${response.statusText}`);
|
console.warn(`Failed to fetch from ${user.handle} (${response.status}): ${response.statusText}`);
|
||||||
@ -1043,6 +1089,18 @@ function App() {
|
|||||||
AI Chat History ({aiChatHistory.length})
|
AI Chat History ({aiChatHistory.length})
|
||||||
</button>
|
</button>
|
||||||
)}
|
)}
|
||||||
|
<button
|
||||||
|
className={`tab-button ${activeTab === 'lang-en' ? 'active' : ''}`}
|
||||||
|
onClick={() => setActiveTab('lang-en')}
|
||||||
|
>
|
||||||
|
Lang: EN ({langEnRecords.length})
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
className={`tab-button ${activeTab === 'ai-comment' ? 'active' : ''}`}
|
||||||
|
onClick={() => setActiveTab('ai-comment')}
|
||||||
|
>
|
||||||
|
AI Comment ({aiCommentRecords.length})
|
||||||
|
</button>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
{/* Comments List */}
|
{/* Comments List */}
|
||||||
@ -1118,7 +1176,7 @@ function App() {
|
|||||||
</div>
|
</div>
|
||||||
<div className="comment-meta">
|
<div className="comment-meta">
|
||||||
{record.value.url && (
|
{record.value.url && (
|
||||||
<small><a href={record.value.url} target="_blank" rel="noopener noreferrer">{record.value.url}</a></small>
|
<small><a href={record.value.url}>{record.value.url}</a></small>
|
||||||
)}
|
)}
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
@ -1204,7 +1262,7 @@ function App() {
|
|||||||
</div>
|
</div>
|
||||||
<div className="comment-meta">
|
<div className="comment-meta">
|
||||||
{record.value.url && (
|
{record.value.url && (
|
||||||
<small><a href={record.value.url} target="_blank" rel="noopener noreferrer">{record.value.url}</a></small>
|
<small><a href={record.value.url}>{record.value.url}</a></small>
|
||||||
)}
|
)}
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
@ -1223,6 +1281,88 @@ function App() {
|
|||||||
</div>
|
</div>
|
||||||
)}
|
)}
|
||||||
|
|
||||||
|
{/* Lang: EN List */}
|
||||||
|
{activeTab === 'lang-en' && (
|
||||||
|
<div className="lang-en-list">
|
||||||
|
{langEnRecords.length === 0 ? (
|
||||||
|
<p className="no-content">No English translations yet</p>
|
||||||
|
) : (
|
||||||
|
langEnRecords.map((record, index) => (
|
||||||
|
<div key={index} className="lang-item">
|
||||||
|
<div className="lang-header">
|
||||||
|
<img
|
||||||
|
src={record.value.author?.avatar || generatePlaceholderAvatar(record.value.author?.handle || 'AI')}
|
||||||
|
alt="AI Avatar"
|
||||||
|
className="comment-avatar"
|
||||||
|
/>
|
||||||
|
<div className="comment-author-info">
|
||||||
|
<span className="comment-author">
|
||||||
|
{record.value.author?.displayName || 'AI Translator'}
|
||||||
|
</span>
|
||||||
|
<span className="comment-handle">
|
||||||
|
@{record.value.author?.handle || 'ai'}
|
||||||
|
</span>
|
||||||
|
</div>
|
||||||
|
<span className="comment-date">
|
||||||
|
{new Date(record.value.createdAt).toLocaleString()}
|
||||||
|
</span>
|
||||||
|
</div>
|
||||||
|
<div className="lang-content">
|
||||||
|
<div className="lang-type">Type: {record.value.type || 'en'}</div>
|
||||||
|
<div className="lang-body">{record.value.body}</div>
|
||||||
|
</div>
|
||||||
|
<div className="comment-meta">
|
||||||
|
{record.value.url && (
|
||||||
|
<small><a href={record.value.url}>{record.value.url}</a></small>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
))
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{/* AI Comment List */}
|
||||||
|
{activeTab === 'ai-comment' && (
|
||||||
|
<div className="ai-comment-list">
|
||||||
|
{aiCommentRecords.length === 0 ? (
|
||||||
|
<p className="no-content">No AI comments yet</p>
|
||||||
|
) : (
|
||||||
|
aiCommentRecords.map((record, index) => (
|
||||||
|
<div key={index} className="ai-comment-item">
|
||||||
|
<div className="ai-comment-header">
|
||||||
|
<img
|
||||||
|
src={record.value.author?.avatar || generatePlaceholderAvatar(record.value.author?.handle || 'AI')}
|
||||||
|
alt="AI Avatar"
|
||||||
|
className="comment-avatar"
|
||||||
|
/>
|
||||||
|
<div className="comment-author-info">
|
||||||
|
<span className="comment-author">
|
||||||
|
{record.value.author?.displayName || 'AI Commenter'}
|
||||||
|
</span>
|
||||||
|
<span className="comment-handle">
|
||||||
|
@{record.value.author?.handle || 'ai'}
|
||||||
|
</span>
|
||||||
|
</div>
|
||||||
|
<span className="comment-date">
|
||||||
|
{new Date(record.value.createdAt).toLocaleString()}
|
||||||
|
</span>
|
||||||
|
</div>
|
||||||
|
<div className="ai-comment-content">
|
||||||
|
<div className="ai-comment-type">Type: {record.value.type || 'comment'}</div>
|
||||||
|
<div className="ai-comment-body">{record.value.body}</div>
|
||||||
|
</div>
|
||||||
|
<div className="comment-meta">
|
||||||
|
{record.value.url && (
|
||||||
|
<small><a href={record.value.url}>{record.value.url}</a></small>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
))
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
{/* Comment Form - Only show on post pages */}
|
{/* Comment Form - Only show on post pages */}
|
||||||
{user && appConfig.rkey && (
|
{user && appConfig.rkey && (
|
||||||
<div className="comment-form">
|
<div className="comment-form">
|
||||||
|
@ -2,9 +2,7 @@
|
|||||||
export interface AppConfig {
|
export interface AppConfig {
|
||||||
adminDid: string;
|
adminDid: string;
|
||||||
collections: {
|
collections: {
|
||||||
comment: string;
|
base: string; // Base collection like "ai.syui.log"
|
||||||
user: string;
|
|
||||||
chat: string;
|
|
||||||
};
|
};
|
||||||
host: string;
|
host: string;
|
||||||
rkey?: string; // Current post rkey if on post page
|
rkey?: string; // Current post rkey if on post page
|
||||||
@ -16,10 +14,21 @@ export interface AppConfig {
|
|||||||
bskyPublicApi: string;
|
bskyPublicApi: string;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Collection name builders (similar to Rust implementation)
|
||||||
|
export function getCollectionNames(base: string) {
|
||||||
|
return {
|
||||||
|
comment: base,
|
||||||
|
user: `${base}.user`,
|
||||||
|
chat: `${base}.chat`,
|
||||||
|
chatLang: `${base}.chat.lang`,
|
||||||
|
chatComment: `${base}.chat.comment`,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
// Generate collection names from host
|
// Generate collection names from host
|
||||||
// Format: ${reg}.${name}.${sub}
|
// Format: ${reg}.${name}.${sub}
|
||||||
// Example: log.syui.ai -> ai.syui.log
|
// Example: log.syui.ai -> ai.syui.log
|
||||||
function generateCollectionNames(host: string): { comment: string; user: string; chat: string } {
|
function generateBaseCollectionFromHost(host: string): string {
|
||||||
try {
|
try {
|
||||||
// Remove protocol if present
|
// Remove protocol if present
|
||||||
const cleanHost = host.replace(/^https?:\/\//, '');
|
const cleanHost = host.replace(/^https?:\/\//, '');
|
||||||
@ -34,21 +43,11 @@ function generateCollectionNames(host: string): { comment: string; user: string;
|
|||||||
// Reverse the parts for collection naming
|
// Reverse the parts for collection naming
|
||||||
// log.syui.ai -> ai.syui.log
|
// log.syui.ai -> ai.syui.log
|
||||||
const reversedParts = parts.reverse();
|
const reversedParts = parts.reverse();
|
||||||
const collectionBase = reversedParts.join('.');
|
return reversedParts.join('.');
|
||||||
|
|
||||||
return {
|
|
||||||
comment: collectionBase,
|
|
||||||
user: `${collectionBase}.user`,
|
|
||||||
chat: `${collectionBase}.chat`
|
|
||||||
};
|
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
console.warn('Failed to generate collection names from host:', host, error);
|
console.warn('Failed to generate collection base from host:', host, error);
|
||||||
// Fallback to default collections
|
// Fallback to default
|
||||||
return {
|
return 'ai.syui.log';
|
||||||
comment: 'ai.syui.log',
|
|
||||||
user: 'ai.syui.log.user',
|
|
||||||
chat: 'ai.syui.log.chat'
|
|
||||||
};
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -66,11 +65,9 @@ export function getAppConfig(): AppConfig {
|
|||||||
const adminDid = import.meta.env.VITE_ADMIN_DID || 'did:plc:uqzpqmrjnptsxezjx4xuh2mn';
|
const adminDid = import.meta.env.VITE_ADMIN_DID || 'did:plc:uqzpqmrjnptsxezjx4xuh2mn';
|
||||||
|
|
||||||
// Priority: Environment variables > Auto-generated from host
|
// Priority: Environment variables > Auto-generated from host
|
||||||
const autoGeneratedCollections = generateCollectionNames(host);
|
const autoGeneratedBase = generateBaseCollectionFromHost(host);
|
||||||
const collections = {
|
const collections = {
|
||||||
comment: import.meta.env.VITE_COLLECTION_COMMENT || autoGeneratedCollections.comment,
|
base: import.meta.env.VITE_OAUTH_COLLECTION || autoGeneratedBase,
|
||||||
user: import.meta.env.VITE_COLLECTION_USER || autoGeneratedCollections.user,
|
|
||||||
chat: import.meta.env.VITE_COLLECTION_CHAT || autoGeneratedCollections.chat,
|
|
||||||
};
|
};
|
||||||
|
|
||||||
const rkey = extractRkeyFromUrl();
|
const rkey = extractRkeyFromUrl();
|
||||||
|
@ -28,8 +28,31 @@ pub struct JetstreamConfig {
|
|||||||
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||||
pub struct CollectionConfig {
|
pub struct CollectionConfig {
|
||||||
pub comment: String,
|
pub base: String, // Base collection name like "ai.syui.log"
|
||||||
pub user: String,
|
}
|
||||||
|
|
||||||
|
impl CollectionConfig {
|
||||||
|
// Collection name builders
|
||||||
|
pub fn comment(&self) -> String {
|
||||||
|
self.base.clone()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn user(&self) -> String {
|
||||||
|
format!("{}.user", self.base)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[allow(dead_code)]
|
||||||
|
pub fn chat(&self) -> String {
|
||||||
|
format!("{}.chat", self.base)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn chat_lang(&self) -> String {
|
||||||
|
format!("{}.chat.lang", self.base)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn chat_comment(&self) -> String {
|
||||||
|
format!("{}.chat.comment", self.base)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for AuthConfig {
|
impl Default for AuthConfig {
|
||||||
@ -47,8 +70,7 @@ impl Default for AuthConfig {
|
|||||||
collections: vec!["ai.syui.log".to_string()],
|
collections: vec!["ai.syui.log".to_string()],
|
||||||
},
|
},
|
||||||
collections: CollectionConfig {
|
collections: CollectionConfig {
|
||||||
comment: "ai.syui.log".to_string(),
|
base: "ai.syui.log".to_string(),
|
||||||
user: "ai.syui.log.user".to_string(),
|
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -220,11 +242,50 @@ pub fn load_config() -> Result<AuthConfig> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
let config_json = fs::read_to_string(&config_path)?;
|
let config_json = fs::read_to_string(&config_path)?;
|
||||||
let mut config: AuthConfig = serde_json::from_str(&config_json)?;
|
|
||||||
|
|
||||||
// Update collection configuration
|
// Try to load as new format first, then migrate if needed
|
||||||
|
match serde_json::from_str::<AuthConfig>(&config_json) {
|
||||||
|
Ok(mut config) => {
|
||||||
|
// Update collection configuration
|
||||||
|
update_config_collections(&mut config);
|
||||||
|
Ok(config)
|
||||||
|
}
|
||||||
|
Err(e) => {
|
||||||
|
println!("{}", format!("Parse error: {}, attempting migration...", e).yellow());
|
||||||
|
// Try to migrate from old format
|
||||||
|
migrate_config_if_needed(&config_path, &config_json)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn migrate_config_if_needed(config_path: &std::path::Path, config_json: &str) -> Result<AuthConfig> {
|
||||||
|
// Try to parse as old format and migrate to new simple format
|
||||||
|
let mut old_config: serde_json::Value = serde_json::from_str(config_json)?;
|
||||||
|
|
||||||
|
// Migrate old collections structure to new base-only structure
|
||||||
|
if let Some(collections) = old_config.get_mut("collections") {
|
||||||
|
// Extract base collection name from comment field or use default
|
||||||
|
let base_collection = collections.get("comment")
|
||||||
|
.and_then(|v| v.as_str())
|
||||||
|
.unwrap_or("ai.syui.log")
|
||||||
|
.to_string();
|
||||||
|
|
||||||
|
// Replace entire collections structure with new format
|
||||||
|
old_config["collections"] = serde_json::json!({
|
||||||
|
"base": base_collection
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// Save migrated config
|
||||||
|
let migrated_config_json = serde_json::to_string_pretty(&old_config)?;
|
||||||
|
fs::write(config_path, migrated_config_json)?;
|
||||||
|
|
||||||
|
// Parse as new format
|
||||||
|
let mut config: AuthConfig = serde_json::from_value(old_config)?;
|
||||||
update_config_collections(&mut config);
|
update_config_collections(&mut config);
|
||||||
|
|
||||||
|
println!("{}", "✅ Configuration migrated to new simplified format".green());
|
||||||
|
|
||||||
Ok(config)
|
Ok(config)
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -259,7 +320,7 @@ async fn test_api_access_with_auth(config: &AuthConfig) -> Result<()> {
|
|||||||
let url = format!("{}/xrpc/com.atproto.repo.listRecords?repo={}&collection={}&limit=1",
|
let url = format!("{}/xrpc/com.atproto.repo.listRecords?repo={}&collection={}&limit=1",
|
||||||
config.admin.pds,
|
config.admin.pds,
|
||||||
urlencoding::encode(&config.admin.did),
|
urlencoding::encode(&config.admin.did),
|
||||||
urlencoding::encode(&config.collections.comment));
|
urlencoding::encode(&config.collections.comment()));
|
||||||
|
|
||||||
let response = client
|
let response = client
|
||||||
.get(&url)
|
.get(&url)
|
||||||
@ -311,23 +372,14 @@ fn save_config(config: &AuthConfig) -> Result<()> {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
// Generate collection names from admin DID or environment
|
// Generate collection config from environment
|
||||||
fn generate_collection_config() -> CollectionConfig {
|
fn generate_collection_config() -> CollectionConfig {
|
||||||
// Check environment variables first
|
// Use VITE_OAUTH_COLLECTION for unified configuration
|
||||||
if let (Ok(comment), Ok(user)) = (
|
let base = std::env::var("VITE_OAUTH_COLLECTION")
|
||||||
std::env::var("AILOG_COLLECTION_COMMENT"),
|
.unwrap_or_else(|_| "ai.syui.log".to_string());
|
||||||
std::env::var("AILOG_COLLECTION_USER")
|
|
||||||
) {
|
|
||||||
return CollectionConfig {
|
|
||||||
comment,
|
|
||||||
user,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
// Default collections
|
|
||||||
CollectionConfig {
|
CollectionConfig {
|
||||||
comment: "ai.syui.log".to_string(),
|
base,
|
||||||
user: "ai.syui.log.user".to_string(),
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -335,5 +387,5 @@ fn generate_collection_config() -> CollectionConfig {
|
|||||||
pub fn update_config_collections(config: &mut AuthConfig) {
|
pub fn update_config_collections(config: &mut AuthConfig) {
|
||||||
config.collections = generate_collection_config();
|
config.collections = generate_collection_config();
|
||||||
// Also update jetstream collections to monitor the comment collection
|
// Also update jetstream collections to monitor the comment collection
|
||||||
config.jetstream.collections = vec![config.collections.comment.clone()];
|
config.jetstream.collections = vec![config.collections.comment()];
|
||||||
}
|
}
|
@ -45,18 +45,10 @@ pub async fn build(project_dir: PathBuf) -> Result<()> {
|
|||||||
.and_then(|v| v.as_str())
|
.and_then(|v| v.as_str())
|
||||||
.ok_or_else(|| anyhow::anyhow!("No admin DID found in [oauth] section"))?;
|
.ok_or_else(|| anyhow::anyhow!("No admin DID found in [oauth] section"))?;
|
||||||
|
|
||||||
let collection_comment = oauth_config.get("collection_comment")
|
let collection_base = oauth_config.get("collection")
|
||||||
.and_then(|v| v.as_str())
|
.and_then(|v| v.as_str())
|
||||||
.unwrap_or("ai.syui.log");
|
.unwrap_or("ai.syui.log");
|
||||||
|
|
||||||
let collection_user = oauth_config.get("collection_user")
|
|
||||||
.and_then(|v| v.as_str())
|
|
||||||
.unwrap_or("ai.syui.log.user");
|
|
||||||
|
|
||||||
let collection_chat = oauth_config.get("collection_chat")
|
|
||||||
.and_then(|v| v.as_str())
|
|
||||||
.unwrap_or("ai.syui.log.chat");
|
|
||||||
|
|
||||||
// Extract AI config if present
|
// Extract AI config if present
|
||||||
let ai_config = config.get("ai")
|
let ai_config = config.get("ai")
|
||||||
.and_then(|v| v.as_table());
|
.and_then(|v| v.as_table());
|
||||||
@ -109,15 +101,8 @@ VITE_OAUTH_CLIENT_ID={}/{}
|
|||||||
VITE_OAUTH_REDIRECT_URI={}/{}
|
VITE_OAUTH_REDIRECT_URI={}/{}
|
||||||
VITE_ADMIN_DID={}
|
VITE_ADMIN_DID={}
|
||||||
|
|
||||||
# Collection names for OAuth app
|
# Base collection for OAuth app and ailog (all others are derived)
|
||||||
VITE_COLLECTION_COMMENT={}
|
VITE_OAUTH_COLLECTION={}
|
||||||
VITE_COLLECTION_USER={}
|
|
||||||
VITE_COLLECTION_CHAT={}
|
|
||||||
|
|
||||||
# Collection names for ailog (backward compatibility)
|
|
||||||
AILOG_COLLECTION_COMMENT={}
|
|
||||||
AILOG_COLLECTION_USER={}
|
|
||||||
AILOG_COLLECTION_CHAT={}
|
|
||||||
|
|
||||||
# AI Configuration
|
# AI Configuration
|
||||||
VITE_AI_ENABLED={}
|
VITE_AI_ENABLED={}
|
||||||
@ -135,12 +120,7 @@ VITE_BSKY_PUBLIC_API={}
|
|||||||
base_url, client_id_path,
|
base_url, client_id_path,
|
||||||
base_url, redirect_path,
|
base_url, redirect_path,
|
||||||
admin_did,
|
admin_did,
|
||||||
collection_comment,
|
collection_base,
|
||||||
collection_user,
|
|
||||||
collection_chat,
|
|
||||||
collection_comment,
|
|
||||||
collection_user,
|
|
||||||
collection_chat,
|
|
||||||
ai_enabled,
|
ai_enabled,
|
||||||
ai_ask_ai,
|
ai_ask_ai,
|
||||||
ai_provider,
|
ai_provider,
|
||||||
|
@ -10,18 +10,58 @@ use std::process::{Command, Stdio};
|
|||||||
use tokio::time::{sleep, Duration, interval};
|
use tokio::time::{sleep, Duration, interval};
|
||||||
use tokio_tungstenite::{connect_async, tungstenite::Message};
|
use tokio_tungstenite::{connect_async, tungstenite::Message};
|
||||||
use toml;
|
use toml;
|
||||||
|
use reqwest;
|
||||||
|
|
||||||
use super::auth::{load_config, load_config_with_refresh, AuthConfig};
|
use super::auth::{load_config, load_config_with_refresh, AuthConfig};
|
||||||
|
|
||||||
|
#[derive(Debug, Deserialize)]
|
||||||
|
#[allow(dead_code)]
|
||||||
|
struct BlogPost {
|
||||||
|
title: String,
|
||||||
|
href: String,
|
||||||
|
#[serde(rename = "formated_time")]
|
||||||
|
#[allow(dead_code)]
|
||||||
|
date: String,
|
||||||
|
#[allow(dead_code)]
|
||||||
|
tags: Vec<String>,
|
||||||
|
#[allow(dead_code)]
|
||||||
|
contents: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Deserialize)]
|
||||||
|
#[allow(dead_code)]
|
||||||
|
struct BlogIndex {
|
||||||
|
#[allow(dead_code)]
|
||||||
|
posts: Vec<BlogPost>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Serialize)]
|
||||||
|
struct OllamaRequest {
|
||||||
|
model: String,
|
||||||
|
prompt: String,
|
||||||
|
stream: bool,
|
||||||
|
options: OllamaOptions,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Serialize)]
|
||||||
|
struct OllamaOptions {
|
||||||
|
temperature: f32,
|
||||||
|
top_p: f32,
|
||||||
|
num_predict: i32,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Deserialize)]
|
||||||
|
struct OllamaResponse {
|
||||||
|
response: String,
|
||||||
|
}
|
||||||
|
|
||||||
// Load collection config with priority: env vars > project config.toml > defaults
|
// Load collection config with priority: env vars > project config.toml > defaults
|
||||||
fn load_collection_config(project_dir: Option<&Path>) -> Result<(String, String)> {
|
fn load_collection_config(project_dir: Option<&Path>) -> Result<(String, String)> {
|
||||||
// 1. Check environment variables first (highest priority)
|
// 1. Check environment variables first (highest priority)
|
||||||
if let (Ok(comment), Ok(user)) = (
|
if let Ok(base_collection) = std::env::var("VITE_OAUTH_COLLECTION") {
|
||||||
std::env::var("AILOG_COLLECTION_COMMENT"),
|
|
||||||
std::env::var("AILOG_COLLECTION_USER")
|
|
||||||
) {
|
|
||||||
println!("{}", "📂 Using collection config from environment variables".cyan());
|
println!("{}", "📂 Using collection config from environment variables".cyan());
|
||||||
return Ok((comment, user));
|
let collection_user = format!("{}.user", base_collection);
|
||||||
|
return Ok((base_collection, collection_user));
|
||||||
}
|
}
|
||||||
|
|
||||||
// 2. Try to load from project config.toml (second priority)
|
// 2. Try to load from project config.toml (second priority)
|
||||||
@ -60,17 +100,16 @@ fn load_collection_config_from_project(project_dir: &Path) -> Result<(String, St
|
|||||||
.and_then(|v| v.as_table())
|
.and_then(|v| v.as_table())
|
||||||
.ok_or_else(|| anyhow::anyhow!("No [oauth] section found in config.toml"))?;
|
.ok_or_else(|| anyhow::anyhow!("No [oauth] section found in config.toml"))?;
|
||||||
|
|
||||||
let collection_comment = oauth_config.get("collection_comment")
|
// Use new simplified collection structure (base collection)
|
||||||
|
let collection_base = oauth_config.get("collection")
|
||||||
.and_then(|v| v.as_str())
|
.and_then(|v| v.as_str())
|
||||||
.unwrap_or("ai.syui.log")
|
.unwrap_or("ai.syui.log")
|
||||||
.to_string();
|
.to_string();
|
||||||
|
|
||||||
let collection_user = oauth_config.get("collection_user")
|
// Derive user collection from base
|
||||||
.and_then(|v| v.as_str())
|
let collection_user = format!("{}.user", collection_base);
|
||||||
.unwrap_or("ai.syui.log.user")
|
|
||||||
.to_string();
|
|
||||||
|
|
||||||
Ok((collection_comment, collection_user))
|
Ok((collection_base, collection_user))
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Serialize, Deserialize)]
|
#[derive(Debug, Serialize, Deserialize)]
|
||||||
@ -118,15 +157,14 @@ fn get_pid_file() -> Result<PathBuf> {
|
|||||||
Ok(pid_dir.join("stream.pid"))
|
Ok(pid_dir.join("stream.pid"))
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn start(project_dir: Option<PathBuf>, daemon: bool) -> Result<()> {
|
pub async fn start(project_dir: Option<PathBuf>, daemon: bool, ai_generate: bool) -> Result<()> {
|
||||||
let mut config = load_config_with_refresh().await?;
|
let mut config = load_config_with_refresh().await?;
|
||||||
|
|
||||||
// Load collection config with priority: env vars > project config > defaults
|
// Load collection config with priority: env vars > project config > defaults
|
||||||
let (collection_comment, collection_user) = load_collection_config(project_dir.as_deref())?;
|
let (collection_comment, _collection_user) = load_collection_config(project_dir.as_deref())?;
|
||||||
|
|
||||||
// Update config with loaded collections
|
// Update config with loaded collections
|
||||||
config.collections.comment = collection_comment.clone();
|
config.collections.base = collection_comment.clone();
|
||||||
config.collections.user = collection_user;
|
|
||||||
config.jetstream.collections = vec![collection_comment];
|
config.jetstream.collections = vec![collection_comment];
|
||||||
|
|
||||||
let pid_file = get_pid_file()?;
|
let pid_file = get_pid_file()?;
|
||||||
@ -151,6 +189,11 @@ pub async fn start(project_dir: Option<PathBuf>, daemon: bool) -> Result<()> {
|
|||||||
args.push(project_path.to_string_lossy().to_string());
|
args.push(project_path.to_string_lossy().to_string());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Add ai_generate flag if enabled
|
||||||
|
if ai_generate {
|
||||||
|
args.push("--ai-generate".to_string());
|
||||||
|
}
|
||||||
|
|
||||||
let child = Command::new(current_exe)
|
let child = Command::new(current_exe)
|
||||||
.args(&args)
|
.args(&args)
|
||||||
.stdin(Stdio::null())
|
.stdin(Stdio::null())
|
||||||
@ -192,6 +235,19 @@ pub async fn start(project_dir: Option<PathBuf>, daemon: bool) -> Result<()> {
|
|||||||
let max_reconnect_attempts = 10;
|
let max_reconnect_attempts = 10;
|
||||||
let mut config = config; // Make config mutable for token refresh
|
let mut config = config; // Make config mutable for token refresh
|
||||||
|
|
||||||
|
// Start AI generation monitor if enabled
|
||||||
|
if ai_generate {
|
||||||
|
let ai_config = config.clone();
|
||||||
|
tokio::spawn(async move {
|
||||||
|
loop {
|
||||||
|
if let Err(e) = run_ai_generation_monitor(&ai_config).await {
|
||||||
|
println!("{}", format!("❌ AI generation monitor error: {}", e).red());
|
||||||
|
sleep(Duration::from_secs(60)).await; // Wait 1 minute before retry
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
loop {
|
loop {
|
||||||
match run_monitor(&mut config).await {
|
match run_monitor(&mut config).await {
|
||||||
Ok(_) => {
|
Ok(_) => {
|
||||||
@ -344,7 +400,7 @@ async fn handle_message(text: &str, config: &mut AuthConfig) -> Result<()> {
|
|||||||
if let (Some(collection), Some(commit), Some(did)) =
|
if let (Some(collection), Some(commit), Some(did)) =
|
||||||
(&message.collection, &message.commit, &message.did) {
|
(&message.collection, &message.commit, &message.did) {
|
||||||
|
|
||||||
if collection == &config.collections.comment && commit.operation.as_deref() == Some("create") {
|
if collection == &config.collections.comment() && commit.operation.as_deref() == Some("create") {
|
||||||
let unknown_uri = "unknown".to_string();
|
let unknown_uri = "unknown".to_string();
|
||||||
let uri = commit.uri.as_ref().unwrap_or(&unknown_uri);
|
let uri = commit.uri.as_ref().unwrap_or(&unknown_uri);
|
||||||
|
|
||||||
@ -438,7 +494,7 @@ async fn get_current_user_list(config: &mut AuthConfig) -> Result<Vec<UserRecord
|
|||||||
let url = format!("{}/xrpc/com.atproto.repo.listRecords?repo={}&collection={}&limit=10",
|
let url = format!("{}/xrpc/com.atproto.repo.listRecords?repo={}&collection={}&limit=10",
|
||||||
config.admin.pds,
|
config.admin.pds,
|
||||||
urlencoding::encode(&config.admin.did),
|
urlencoding::encode(&config.admin.did),
|
||||||
urlencoding::encode(&config.collections.user));
|
urlencoding::encode(&config.collections.user()));
|
||||||
|
|
||||||
let response = client
|
let response = client
|
||||||
.get(&url)
|
.get(&url)
|
||||||
@ -501,7 +557,7 @@ async fn post_user_list(config: &mut AuthConfig, users: &[UserRecord], metadata:
|
|||||||
let rkey = format!("{}-{}", short_did, now.format("%Y-%m-%dT%H-%M-%S-%3fZ").to_string().replace(".", "-"));
|
let rkey = format!("{}-{}", short_did, now.format("%Y-%m-%dT%H-%M-%S-%3fZ").to_string().replace(".", "-"));
|
||||||
|
|
||||||
let record = UserListRecord {
|
let record = UserListRecord {
|
||||||
record_type: config.collections.user.clone(),
|
record_type: config.collections.user(),
|
||||||
users: users.to_vec(),
|
users: users.to_vec(),
|
||||||
created_at: now.to_rfc3339(),
|
created_at: now.to_rfc3339(),
|
||||||
updated_by: UserInfo {
|
updated_by: UserInfo {
|
||||||
@ -515,7 +571,7 @@ async fn post_user_list(config: &mut AuthConfig, users: &[UserRecord], metadata:
|
|||||||
|
|
||||||
let request_body = json!({
|
let request_body = json!({
|
||||||
"repo": config.admin.did,
|
"repo": config.admin.did,
|
||||||
"collection": config.collections.user,
|
"collection": config.collections.user(),
|
||||||
"rkey": rkey,
|
"rkey": rkey,
|
||||||
"record": record
|
"record": record
|
||||||
});
|
});
|
||||||
@ -759,7 +815,7 @@ async fn get_recent_comments(config: &mut AuthConfig) -> Result<Vec<Value>> {
|
|||||||
let url = format!("{}/xrpc/com.atproto.repo.listRecords?repo={}&collection={}&limit=20",
|
let url = format!("{}/xrpc/com.atproto.repo.listRecords?repo={}&collection={}&limit=20",
|
||||||
config.admin.pds,
|
config.admin.pds,
|
||||||
urlencoding::encode(&config.admin.did),
|
urlencoding::encode(&config.admin.did),
|
||||||
urlencoding::encode(&config.collections.comment));
|
urlencoding::encode(&config.collections.comment()));
|
||||||
|
|
||||||
if std::env::var("AILOG_DEBUG").is_ok() {
|
if std::env::var("AILOG_DEBUG").is_ok() {
|
||||||
println!("{}", format!("🌐 API Request URL: {}", url).yellow());
|
println!("{}", format!("🌐 API Request URL: {}", url).yellow());
|
||||||
@ -840,7 +896,7 @@ pub async fn test_api() -> Result<()> {
|
|||||||
println!("{}", format!("✅ Successfully retrieved {} comments", comments.len()).green());
|
println!("{}", format!("✅ Successfully retrieved {} comments", comments.len()).green());
|
||||||
|
|
||||||
if comments.is_empty() {
|
if comments.is_empty() {
|
||||||
println!("{}", format!("ℹ️ No comments found in {} collection", config.collections.comment).blue());
|
println!("{}", format!("ℹ️ No comments found in {} collection", config.collections.comment()).blue());
|
||||||
println!("💡 Try posting a comment first using the web interface");
|
println!("💡 Try posting a comment first using the web interface");
|
||||||
} else {
|
} else {
|
||||||
println!("{}", "📝 Comment details:".cyan());
|
println!("{}", "📝 Comment details:".cyan());
|
||||||
@ -871,5 +927,273 @@ pub async fn test_api() -> Result<()> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
// AI content generation functions
|
||||||
|
async fn generate_ai_content(content: &str, prompt_type: &str, ollama_host: &str) -> Result<String> {
|
||||||
|
let model = "gemma3:4b";
|
||||||
|
|
||||||
|
let prompt = match prompt_type {
|
||||||
|
"translate" => format!("Translate the following Japanese blog post to English. Keep the technical terms and code blocks intact:\n\n{}", content),
|
||||||
|
"comment" => format!("Read this blog post and provide an insightful comment about it. Focus on the key points and add your perspective:\n\n{}", content),
|
||||||
|
_ => return Err(anyhow::anyhow!("Unknown prompt type: {}", prompt_type)),
|
||||||
|
};
|
||||||
|
|
||||||
|
let request = OllamaRequest {
|
||||||
|
model: model.to_string(),
|
||||||
|
prompt,
|
||||||
|
stream: false,
|
||||||
|
options: OllamaOptions {
|
||||||
|
temperature: 0.9,
|
||||||
|
top_p: 0.9,
|
||||||
|
num_predict: 500,
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
let client = reqwest::Client::new();
|
||||||
|
|
||||||
|
// Try localhost first (for same-server deployment)
|
||||||
|
let localhost_url = "http://localhost:11434/api/generate";
|
||||||
|
match client.post(localhost_url).json(&request).send().await {
|
||||||
|
Ok(response) if response.status().is_success() => {
|
||||||
|
let ollama_response: OllamaResponse = response.json().await?;
|
||||||
|
println!("{}", "✅ Used localhost Ollama".green());
|
||||||
|
return Ok(ollama_response.response);
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
println!("{}", "⚠️ Localhost Ollama not available, trying remote...".yellow());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Fallback to remote host
|
||||||
|
let remote_url = format!("{}/api/generate", ollama_host);
|
||||||
|
let response = client.post(&remote_url).json(&request).send().await?;
|
||||||
|
|
||||||
|
if !response.status().is_success() {
|
||||||
|
return Err(anyhow::anyhow!("Ollama API request failed: {}", response.status()));
|
||||||
|
}
|
||||||
|
|
||||||
|
let ollama_response: OllamaResponse = response.json().await?;
|
||||||
|
println!("{}", "✅ Used remote Ollama".green());
|
||||||
|
Ok(ollama_response.response)
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn run_ai_generation_monitor(config: &AuthConfig) -> Result<()> {
|
||||||
|
let blog_host = "https://syui.ai"; // TODO: Load from config
|
||||||
|
let ollama_host = "https://ollama.syui.ai"; // TODO: Load from config
|
||||||
|
let ai_did = "did:plc:4hqjfn7m6n5hno3doamuhgef"; // TODO: Load from config
|
||||||
|
|
||||||
|
println!("{}", "🤖 Starting AI content generation monitor...".cyan());
|
||||||
|
println!("📡 Blog host: {}", blog_host);
|
||||||
|
println!("🧠 Ollama host: {}", ollama_host);
|
||||||
|
println!("🤖 AI DID: {}", ai_did);
|
||||||
|
println!();
|
||||||
|
|
||||||
|
let mut interval = interval(Duration::from_secs(300)); // Check every 5 minutes
|
||||||
|
let client = reqwest::Client::new();
|
||||||
|
|
||||||
|
loop {
|
||||||
|
interval.tick().await;
|
||||||
|
|
||||||
|
println!("{}", "🔍 Checking for new blog posts...".blue());
|
||||||
|
|
||||||
|
match check_and_process_new_posts(&client, config, blog_host, ollama_host, ai_did).await {
|
||||||
|
Ok(count) => {
|
||||||
|
if count > 0 {
|
||||||
|
println!("{}", format!("✅ Processed {} new posts", count).green());
|
||||||
|
} else {
|
||||||
|
println!("{}", "ℹ️ No new posts found".blue());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Err(e) => {
|
||||||
|
println!("{}", format!("❌ Error processing posts: {}", e).red());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
println!("{}", "⏰ Waiting for next check...".cyan());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn check_and_process_new_posts(
|
||||||
|
client: &reqwest::Client,
|
||||||
|
config: &AuthConfig,
|
||||||
|
blog_host: &str,
|
||||||
|
ollama_host: &str,
|
||||||
|
ai_did: &str,
|
||||||
|
) -> Result<usize> {
|
||||||
|
// Fetch blog index
|
||||||
|
let index_url = format!("{}/index.json", blog_host);
|
||||||
|
let response = client.get(&index_url).send().await?;
|
||||||
|
|
||||||
|
if !response.status().is_success() {
|
||||||
|
return Err(anyhow::anyhow!("Failed to fetch blog index: {}", response.status()));
|
||||||
|
}
|
||||||
|
|
||||||
|
let blog_posts: Vec<BlogPost> = response.json().await?;
|
||||||
|
println!("{}", format!("📄 Found {} posts in blog index", blog_posts.len()).cyan());
|
||||||
|
|
||||||
|
// Get existing AI generated content from collections
|
||||||
|
let existing_lang_records = get_existing_records(config, &config.collections.chat_lang()).await?;
|
||||||
|
let existing_comment_records = get_existing_records(config, &config.collections.chat_comment()).await?;
|
||||||
|
|
||||||
|
let mut processed_count = 0;
|
||||||
|
|
||||||
|
for post in blog_posts {
|
||||||
|
let post_slug = extract_slug_from_url(&post.href);
|
||||||
|
|
||||||
|
// Check if translation already exists
|
||||||
|
let translation_exists = existing_lang_records.iter().any(|record| {
|
||||||
|
record.get("value")
|
||||||
|
.and_then(|v| v.get("post_slug"))
|
||||||
|
.and_then(|s| s.as_str())
|
||||||
|
== Some(&post_slug)
|
||||||
|
});
|
||||||
|
|
||||||
|
// Check if comment already exists
|
||||||
|
let comment_exists = existing_comment_records.iter().any(|record| {
|
||||||
|
record.get("value")
|
||||||
|
.and_then(|v| v.get("post_slug"))
|
||||||
|
.and_then(|s| s.as_str())
|
||||||
|
== Some(&post_slug)
|
||||||
|
});
|
||||||
|
|
||||||
|
// Generate translation if not exists
|
||||||
|
if !translation_exists {
|
||||||
|
match generate_and_store_translation(client, config, &post, ollama_host, ai_did).await {
|
||||||
|
Ok(_) => {
|
||||||
|
println!("{}", format!("✅ Generated translation for: {}", post.title).green());
|
||||||
|
processed_count += 1;
|
||||||
|
}
|
||||||
|
Err(e) => {
|
||||||
|
println!("{}", format!("❌ Failed to generate translation for {}: {}", post.title, e).red());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Generate comment if not exists
|
||||||
|
if !comment_exists {
|
||||||
|
match generate_and_store_comment(client, config, &post, ollama_host, ai_did).await {
|
||||||
|
Ok(_) => {
|
||||||
|
println!("{}", format!("✅ Generated comment for: {}", post.title).green());
|
||||||
|
processed_count += 1;
|
||||||
|
}
|
||||||
|
Err(e) => {
|
||||||
|
println!("{}", format!("❌ Failed to generate comment for {}: {}", post.title, e).red());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(processed_count)
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn get_existing_records(config: &AuthConfig, collection: &str) -> Result<Vec<serde_json::Value>> {
|
||||||
|
let client = reqwest::Client::new();
|
||||||
|
let url = format!("{}/xrpc/com.atproto.repo.listRecords?repo={}&collection={}&limit=100",
|
||||||
|
config.admin.pds,
|
||||||
|
urlencoding::encode(&config.admin.did),
|
||||||
|
urlencoding::encode(collection));
|
||||||
|
|
||||||
|
let response = client
|
||||||
|
.get(&url)
|
||||||
|
.header("Authorization", format!("Bearer {}", config.admin.access_jwt))
|
||||||
|
.send()
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
if !response.status().is_success() {
|
||||||
|
return Ok(Vec::new()); // Return empty if collection doesn't exist yet
|
||||||
|
}
|
||||||
|
|
||||||
|
let list_response: serde_json::Value = response.json().await?;
|
||||||
|
let records = list_response["records"].as_array().unwrap_or(&Vec::new()).clone();
|
||||||
|
|
||||||
|
Ok(records)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn extract_slug_from_url(url: &str) -> String {
|
||||||
|
// Extract slug from URL like "/posts/2025-06-06-ailog.html"
|
||||||
|
url.split('/')
|
||||||
|
.last()
|
||||||
|
.unwrap_or("")
|
||||||
|
.trim_end_matches(".html")
|
||||||
|
.to_string()
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn generate_and_store_translation(
|
||||||
|
client: &reqwest::Client,
|
||||||
|
config: &AuthConfig,
|
||||||
|
post: &BlogPost,
|
||||||
|
ollama_host: &str,
|
||||||
|
ai_did: &str,
|
||||||
|
) -> Result<()> {
|
||||||
|
// Generate translation
|
||||||
|
let translation = generate_ai_content(&post.title, "translate", ollama_host).await?;
|
||||||
|
|
||||||
|
// Store in ai.syui.log.chat.lang collection
|
||||||
|
let record_data = serde_json::json!({
|
||||||
|
"post_slug": extract_slug_from_url(&post.href),
|
||||||
|
"post_title": post.title,
|
||||||
|
"post_url": post.href,
|
||||||
|
"lang": "en",
|
||||||
|
"content": translation,
|
||||||
|
"generated_at": chrono::Utc::now().to_rfc3339(),
|
||||||
|
"ai_did": ai_did
|
||||||
|
});
|
||||||
|
|
||||||
|
store_atproto_record(client, config, &config.collections.chat_lang(), &record_data).await
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn generate_and_store_comment(
|
||||||
|
client: &reqwest::Client,
|
||||||
|
config: &AuthConfig,
|
||||||
|
post: &BlogPost,
|
||||||
|
ollama_host: &str,
|
||||||
|
ai_did: &str,
|
||||||
|
) -> Result<()> {
|
||||||
|
// Generate comment
|
||||||
|
let comment = generate_ai_content(&post.title, "comment", ollama_host).await?;
|
||||||
|
|
||||||
|
// Store in ai.syui.log.chat.comment collection
|
||||||
|
let record_data = serde_json::json!({
|
||||||
|
"post_slug": extract_slug_from_url(&post.href),
|
||||||
|
"post_title": post.title,
|
||||||
|
"post_url": post.href,
|
||||||
|
"content": comment,
|
||||||
|
"generated_at": chrono::Utc::now().to_rfc3339(),
|
||||||
|
"ai_did": ai_did
|
||||||
|
});
|
||||||
|
|
||||||
|
store_atproto_record(client, config, &config.collections.chat_comment(), &record_data).await
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn store_atproto_record(
|
||||||
|
client: &reqwest::Client,
|
||||||
|
config: &AuthConfig,
|
||||||
|
collection: &str,
|
||||||
|
record_data: &serde_json::Value,
|
||||||
|
) -> Result<()> {
|
||||||
|
let url = format!("{}/xrpc/com.atproto.repo.putRecord", config.admin.pds);
|
||||||
|
|
||||||
|
let put_request = serde_json::json!({
|
||||||
|
"repo": config.admin.did,
|
||||||
|
"collection": collection,
|
||||||
|
"rkey": uuid::Uuid::new_v4().to_string(),
|
||||||
|
"record": record_data
|
||||||
|
});
|
||||||
|
|
||||||
|
let response = client
|
||||||
|
.post(&url)
|
||||||
|
.header("Authorization", format!("Bearer {}", config.admin.access_jwt))
|
||||||
|
.header("Content-Type", "application/json")
|
||||||
|
.json(&put_request)
|
||||||
|
.send()
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
if !response.status().is_success() {
|
||||||
|
let error_text = response.text().await?;
|
||||||
|
return Err(anyhow::anyhow!("Failed to store record: {}", error_text));
|
||||||
|
}
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
@ -71,6 +71,9 @@ impl Generator {
|
|||||||
// Generate index page
|
// Generate index page
|
||||||
self.generate_index(&posts).await?;
|
self.generate_index(&posts).await?;
|
||||||
|
|
||||||
|
// Generate JSON index for API access
|
||||||
|
self.generate_json_index(&posts).await?;
|
||||||
|
|
||||||
// Generate post pages
|
// Generate post pages
|
||||||
for post in &posts {
|
for post in &posts {
|
||||||
self.generate_post_page(post).await?;
|
self.generate_post_page(post).await?;
|
||||||
@ -446,6 +449,63 @@ impl Generator {
|
|||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn generate_json_index(&self, posts: &[Post]) -> Result<()> {
|
||||||
|
let index_data: Vec<serde_json::Value> = posts.iter().map(|post| {
|
||||||
|
// Parse date for proper formatting
|
||||||
|
let parsed_date = chrono::NaiveDate::parse_from_str(&post.date, "%Y-%m-%d")
|
||||||
|
.unwrap_or_else(|_| chrono::Utc::now().naive_utc().date());
|
||||||
|
|
||||||
|
// Format to Hugo-style date format (Mon Jan 2, 2006)
|
||||||
|
let formatted_date = parsed_date.format("%a %b %-d, %Y").to_string();
|
||||||
|
|
||||||
|
// Create UTC datetime for utc_time field
|
||||||
|
let utc_datetime = parsed_date.and_hms_opt(0, 0, 0)
|
||||||
|
.unwrap_or_else(|| chrono::Utc::now().naive_utc());
|
||||||
|
let utc_time = format!("{}Z", utc_datetime.format("%Y-%m-%dT%H:%M:%S"));
|
||||||
|
|
||||||
|
// Extract plain text content from HTML
|
||||||
|
let contents = self.extract_plain_text(&post.content);
|
||||||
|
|
||||||
|
serde_json::json!({
|
||||||
|
"title": post.title,
|
||||||
|
"tags": post.tags,
|
||||||
|
"description": self.extract_excerpt(&post.content),
|
||||||
|
"categories": [],
|
||||||
|
"contents": contents,
|
||||||
|
"href": format!("{}{}", self.config.site.base_url.trim_end_matches('/'), post.url),
|
||||||
|
"utc_time": utc_time,
|
||||||
|
"formated_time": formatted_date
|
||||||
|
})
|
||||||
|
}).collect();
|
||||||
|
|
||||||
|
// Write JSON index to public directory
|
||||||
|
let output_path = self.base_path.join("public/index.json");
|
||||||
|
let json_content = serde_json::to_string_pretty(&index_data)?;
|
||||||
|
fs::write(output_path, json_content)?;
|
||||||
|
|
||||||
|
println!("{} JSON index with {} posts", "Generated".cyan(), posts.len());
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn extract_plain_text(&self, html_content: &str) -> String {
|
||||||
|
// Remove HTML tags and extract plain text
|
||||||
|
let mut text = String::new();
|
||||||
|
let mut in_tag = false;
|
||||||
|
|
||||||
|
for ch in html_content.chars() {
|
||||||
|
match ch {
|
||||||
|
'<' => in_tag = true,
|
||||||
|
'>' => in_tag = false,
|
||||||
|
_ if !in_tag => text.push(ch),
|
||||||
|
_ => {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Clean up whitespace
|
||||||
|
text.split_whitespace().collect::<Vec<_>>().join(" ")
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, serde::Serialize)]
|
#[derive(Debug, Clone, serde::Serialize)]
|
||||||
@ -479,4 +539,19 @@ pub struct Translation {
|
|||||||
pub title: String,
|
pub title: String,
|
||||||
pub content: String,
|
pub content: String,
|
||||||
pub url: String,
|
pub url: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, serde::Serialize)]
|
||||||
|
#[allow(dead_code)]
|
||||||
|
struct BlogPost {
|
||||||
|
title: String,
|
||||||
|
url: String,
|
||||||
|
date: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, serde::Serialize)]
|
||||||
|
#[allow(dead_code)]
|
||||||
|
struct BlogIndex {
|
||||||
|
posts: Vec<BlogPost>,
|
||||||
|
}
|
||||||
|
|
||||||
|
@ -118,6 +118,9 @@ enum StreamCommands {
|
|||||||
/// Run as daemon
|
/// Run as daemon
|
||||||
#[arg(short, long)]
|
#[arg(short, long)]
|
||||||
daemon: bool,
|
daemon: bool,
|
||||||
|
/// Enable AI content generation
|
||||||
|
#[arg(long)]
|
||||||
|
ai_generate: bool,
|
||||||
},
|
},
|
||||||
/// Stop monitoring
|
/// Stop monitoring
|
||||||
Stop,
|
Stop,
|
||||||
@ -193,8 +196,8 @@ async fn main() -> Result<()> {
|
|||||||
}
|
}
|
||||||
Commands::Stream { command } => {
|
Commands::Stream { command } => {
|
||||||
match command {
|
match command {
|
||||||
StreamCommands::Start { project_dir, daemon } => {
|
StreamCommands::Start { project_dir, daemon, ai_generate } => {
|
||||||
commands::stream::start(project_dir, daemon).await?;
|
commands::stream::start(project_dir, daemon, ai_generate).await?;
|
||||||
}
|
}
|
||||||
StreamCommands::Stop => {
|
StreamCommands::Stop => {
|
||||||
commands::stream::stop().await?;
|
commands::stream::stop().await?;
|
||||||
|
Reference in New Issue
Block a user