8 Commits

Author SHA1 Message Date
IndieKKY
1a475d1f13 chore: release 1.14.1 2025-06-24 13:37:41 +08:00
IndieKKY
4a1ea0dbbe 章节模式配置项 2025-06-24 13:37:19 +08:00
IndieKKY
af80b8a51c chore: release 1.14.0 2025-06-23 16:29:16 +08:00
IndieKKY
404ab904cc 章节显示 2025-06-23 16:28:59 +08:00
IndieKKY
d53884269a 获取章节列表 2025-06-23 16:04:00 +08:00
IndieKKY
e4693cde6a chore: release 1.13.1 2025-06-02 09:31:24 +08:00
IndieKKY
fea22fae61 优化性能 2025-06-02 09:26:38 +08:00
IndieKKY
d5b5b5d374 优化性能 2025-06-02 09:18:02 +08:00
11 changed files with 185 additions and 42 deletions

View File

@@ -1,7 +1,7 @@
{
"private": true,
"name": "bilibili-subtitle",
"version": "1.13.0",
"version": "1.14.1",
"type": "module",
"description": "哔哩哔哩字幕列表",
"main": "index.js",

View File

@@ -1,6 +1,6 @@
import {MutableRefObject, useCallback, useEffect, useMemo, useRef} from 'react'
import {useAppDispatch, useAppSelector} from '../hooks/redux'
import {setFloatKeyPointsSegIdx, setSegmentFold, setTempData} from '../redux/envReducer'
import {setFloatKeyPointsSegIdx, setNeedScroll, setSegmentFold, setTempData} from '../redux/envReducer'
import classNames from 'classnames'
import {FaClipboardList, FaComments} from 'react-icons/fa'
import {SUMMARIZE_THRESHOLD, SUMMARIZE_TYPES} from '../consts/const'
@@ -13,6 +13,7 @@ import SegmentItem from './SegmentItem'
import {stopPopFunc} from '../utils/util'
import useSubtitle from '../hooks/useSubtitle'
import DebateChat from './DebateChat'
import { RootState } from '../store'
const SummarizeItemOverview = (props: {
segment: Segment
@@ -25,7 +26,7 @@ const SummarizeItemOverview = (props: {
const {move} = useSubtitle()
const time = parseStrTimeToSeconds(overviewItem.time)
const currentTime = useAppSelector(state => state.env.currentTime)
const currentTime = useAppSelector((state: RootState) => state.currentTime.currentTime)
const isIn = useMemo(() => {
if (currentTime != null) {
// check in current segment
@@ -158,6 +159,7 @@ const SegmentCard = (props: {
}
return undefined
}, [curSummaryType, segment.summaries])
const {move} = useSubtitle()
const onFold = useCallback(() => {
dispatch(setSegmentFold({
@@ -166,6 +168,21 @@ const SegmentCard = (props: {
}))
}, [dispatch, segment.fold, segment.startIdx])
const onChapterClick = useCallback(() => {
if (segment.items && segment.items.length > 0) {
// 展开当前segment
if (segment.fold) {
dispatch(setSegmentFold({
segmentStartIdx: segment.startIdx,
fold: false
}))
}
const firstItem = segment.items[0]
move(firstItem.from, false)
}
}, [dispatch, move, segment.fold, segment.items, segment.startIdx])
// 检测设置floatKeyPointsSegIdx
useEffect(() => {
if (summarizeFloat) { // 已启用
@@ -211,6 +228,10 @@ const SegmentCard = (props: {
return <div
className={classNames('border border-base-300 bg-base-200/25 rounded flex flex-col m-1.5 p-1.5 gap-1', showCurrent && 'shadow shadow-md')}>
{/* 章节标题 */}
{segment.chapterTitle && <div className='text-center py-1 px-2 bg-primary/10 rounded text-sm font-semibold text-primary border-b border-primary/20 cursor-pointer hover:bg-primary/20 transition-colors' onClick={onChapterClick}>
{segment.chapterTitle}
</div>}
<div className='relative flex justify-center min-h-[20px]'>
{segments != null && segments.length > 0 &&
<div className='absolute left-0 top-0 bottom-0 text-xs select-none flex-center desc'>

View File

@@ -1,4 +1,4 @@
import { setAuthor, setCtime, setCurFetched, setCurInfo, setData, setInfos, setTitle, setUrl } from '@/redux/envReducer'
import { setAuthor, setChapters, setCtime, setCurFetched, setCurInfo, setData, setInfos, setTitle, setUrl } from '@/redux/envReducer'
import { useAppDispatch, useAppSelector } from './redux'
import { AllAPPMessages, AllExtensionMessages, AllInjectMessages } from '@/message-typings'
import { useMessaging, useMessagingService } from '@kky002/kky-message'
@@ -19,6 +19,7 @@ const useMessageService = () => {
dispatch(setData(undefined))
},
SET_VIDEO_INFO: async (params, context: MethodContext) => {
dispatch(setChapters(params.chapters))
dispatch(setInfos(params.infos))
dispatch(setUrl(params.url))
dispatch(setTitle(params.title))

View File

@@ -4,15 +4,11 @@ import {
setCurFetched,
setCurIdx,
setCurInfo,
setCurrentTime,
setData,
setInfos,
setNoVideo,
setSegmentFold,
setSegments,
setTitle,
setTotalHeight,
setUrl,
setTempData,
} from '../redux/envReducer'
import {EventBusContext} from '../Router'
@@ -20,6 +16,8 @@ import {EVENT_EXPAND, TOTAL_HEIGHT_MAX, TOTAL_HEIGHT_MIN, WORDS_MIN, WORDS_RATE}
import {useAsyncEffect, useInterval} from 'ahooks'
import {getModelMaxTokens, getWholeText} from '../utils/bizUtil'
import { useMessage } from './useMessageService'
import { setCurrentTime } from '../redux/currentTimeReducer'
import { RootState } from '../store'
/**
* Service是单例类似后端的服务概念
@@ -31,10 +29,11 @@ const useSubtitleService = () => {
const curFetched = useAppSelector(state => state.env.curFetched)
const fold = useAppSelector(state => state.env.fold)
const envReady = useAppSelector(state => state.env.envReady)
const envData = useAppSelector(state => state.env.envData)
const data = useAppSelector(state => state.env.data)
const currentTime = useAppSelector(state => state.env.currentTime)
const curIdx = useAppSelector(state => state.env.curIdx)
const envData = useAppSelector((state: RootState) => state.env.envData)
const data = useAppSelector((state: RootState) => state.env.data)
const chapters = useAppSelector((state: RootState) => state.env.chapters)
const currentTime = useAppSelector((state: RootState) => state.currentTime.currentTime)
const curIdx = useAppSelector((state: RootState) => state.env.curIdx)
const eventBus = useContext(EventBusContext)
const needScroll = useAppSelector(state => state.env.needScroll)
const segments = useAppSelector(state => state.env.segments)
@@ -113,19 +112,22 @@ const useSubtitleService = () => {
// 更新当前位置
useEffect(() => {
let curIdx
let newCurIdx
if (((data?.body) != null) && currentTime) {
for (let i=0; i<data.body.length; i++) {
const item = data.body[i]
if (item.from && currentTime < item.from) {
break
} else {
curIdx = i
newCurIdx = i
}
}
}
dispatch(setCurIdx(curIdx))
}, [currentTime, data?.body, dispatch])
// 只有当索引发生变化时才更新状态
if (newCurIdx !== curIdx) {
dispatch(setCurIdx(newCurIdx))
}
}, [currentTime, data?.body, dispatch, curIdx])
// 需要滚动 => segment自动展开
useEffect(() => {
@@ -157,6 +159,59 @@ const useSubtitleService = () => {
size = Math.max(size, WORDS_MIN)
segments = []
// 如果启用章节模式且有章节信息,按章节分割
if ((envData.chapterMode ?? true) && chapters && chapters.length > 0) {
for (let chapterIdx = 0; chapterIdx < chapters.length; chapterIdx++) {
const chapter = chapters[chapterIdx]
const nextChapter = chapters[chapterIdx + 1]
// 找到属于当前章节的字幕项
const chapterItems = items.filter(item => {
const itemTime = item.from
return itemTime >= chapter.from && (nextChapter ? itemTime < nextChapter.from : true)
})
if (chapterItems.length === 0) continue
// 如果章节内容过长,需要进一步分割
const chapterText = getWholeText(chapterItems.map(item => item.content))
if (chapterText.length <= size) {
// 章节内容不长作为一个segment
segments.push({
items: chapterItems,
startIdx: chapterItems[0].idx,
endIdx: chapterItems[chapterItems.length - 1].idx,
text: chapterText,
chapterTitle: chapter.content,
summaries: {},
})
} else {
// 章节内容过长需要分割成多个segment
let transcriptItems: TranscriptItem[] = []
let totalLength = 0
for (let i = 0; i < chapterItems.length; i++) {
const item = chapterItems[i]
transcriptItems.push(item)
totalLength += item.content.length
if (totalLength >= size || i === chapterItems.length - 1) {
segments.push({
items: transcriptItems,
startIdx: transcriptItems[0].idx,
endIdx: transcriptItems[transcriptItems.length - 1].idx,
text: getWholeText(transcriptItems.map(item => item.content)),
chapterTitle: chapter.content,
summaries: {},
})
// reset
transcriptItems = []
totalLength = 0
}
}
}
}
} else {
// 没有章节信息,按原来的逻辑分割
let transcriptItems: TranscriptItem[] = []
let totalLength = 0
for (let i = 0; i < items.length; i++) {
@@ -177,6 +232,7 @@ const useSubtitleService = () => {
totalLength = 0
}
}
}
} else { // 都放一个分段
segments = [{
items,
@@ -188,12 +244,15 @@ const useSubtitleService = () => {
}
}
dispatch(setSegments(segments))
}, [data?.body, dispatch, envData])
}, [data?.body, dispatch, envData, chapters])
// 每0.5秒更新当前视频时间
useInterval(() => {
sendInject(null, 'GET_VIDEO_STATUS', {}).then(status => {
// 只有当时间发生显著变化时才更新状态差异大于0.1秒),避免不必要的重新渲染
if (currentTime == null || Math.abs(status.currentTime - currentTime) > 0.1) {
dispatch(setCurrentTime(status.currentTime))
}
})
}, 500)

View File

@@ -161,6 +161,21 @@ const debug = (...args: any[]) => {
if (aidOrBvid) {
// aid,pages
let cid: string | undefined
/**
* [
{
"type": 2,
"from": 0,
"to": 152, //单位秒
"content": "发现美",
"imgUrl": "http://i0.hdslb.com/bfs/vchapter/29168372111_0.jpg",
"logoUrl": "",
"team_type": "",
"team_name": ""
}
]
*/
let chapters: any[] = []
let subtitles
if (aidOrBvid.toLowerCase().startsWith('av')) { // avxxx
aid = parseInt(aidOrBvid.slice(2))
@@ -170,6 +185,7 @@ const debug = (...args: any[]) => {
author = pages[0].owner?.name
title = pages[0].part
await fetch(`https://api.bilibili.com/x/player/wbi/v2?aid=${aid}&cid=${cid!}`, { credentials: 'include' }).then(async res => await res.json()).then(res => {
chapters = res.data.view_points ?? []
subtitles = res.data.subtitle.subtitles
})
} else { // bvxxx
@@ -182,10 +198,14 @@ const debug = (...args: any[]) => {
pages = res.data.pages
})
await fetch(`https://api.bilibili.com/x/player/wbi/v2?aid=${aid!}&cid=${cid!}`, { credentials: 'include' }).then(async res => await res.json()).then(res => {
chapters = res.data.view_points ?? []
subtitles = res.data.subtitle.subtitles
})
}
//筛选chapters里type为2的
chapters = chapters.filter(chapter => chapter.type === 2)
// pagesMap
pagesMap = {}
pages.forEach(page => {
@@ -202,6 +222,7 @@ const debug = (...args: any[]) => {
ctime,
author,
pages,
chapters,
infos: subtitles,
})
}

View File

@@ -80,7 +80,7 @@ interface AppSetInfosMessage extends AppMessage<{ infos: any }> {
method: 'SET_INFOS'
}
interface AppSetVideoInfoMessage extends AppMessage<{ url: string, title: string, aid: number | null, ctime: number | null, author?: string, pages: any, infos: any }> {
interface AppSetVideoInfoMessage extends AppMessage<{ url: string, title: string, aid: number | null, ctime: number | null, author?: string, pages: any, chapters: any, infos: any }> {
method: 'SET_VIDEO_INFO'
}

View File

@@ -77,6 +77,7 @@ const OptionsPage = () => {
const {value: askEnabledValue, onChange: setAskEnabledValue} = useEventChecked(envData.askEnabled??ASK_ENABLED_DEFAULT)
const {value: cnSearchEnabledValue, onChange: setCnSearchEnabledValue} = useEventChecked(envData.cnSearchEnabled)
const {value: summarizeFloatValue, onChange: setSummarizeFloatValue} = useEventChecked(envData.summarizeFloat)
const {value: chapterModeValue, onChange: setChapterModeValue} = useEventChecked(envData.chapterMode ?? true)
const [apiKeyValue, { onChange: onChangeApiKeyValue }] = useEventTarget({initialValue: envData.apiKey??''})
const [serverUrlValue, setServerUrlValue] = useState(envData.serverUrl)
const [languageValue, { onChange: onChangeLanguageValue }] = useEventTarget({initialValue: envData.language??LANGUAGE_DEFAULT})
@@ -139,6 +140,7 @@ const OptionsPage = () => {
searchEnabled: searchEnabledValue,
cnSearchEnabled: cnSearchEnabledValue,
askEnabled: askEnabledValue,
chapterMode: chapterModeValue,
}))
toast.success('保存成功')
sendExtension(null, 'CLOSE_SIDE_PANEL')
@@ -146,7 +148,7 @@ const OptionsPage = () => {
setTimeout(() => {
window.close()
}, 3000)
}, [dispatch, sendExtension, sidePanelValue, autoInsertValue, autoExpandValue, apiKeyValue, serverUrlValue, modelValue, customModelValue, customModelTokensValue, translateEnableValue, languageValue, hideOnDisableAutoTranslateValue, themeValue, transDisplayValue, summarizeEnableValue, summarizeFloatValue, summarizeLanguageValue, wordsValue, fetchAmountValue, fontSizeValue, promptsValue, searchEnabledValue, cnSearchEnabledValue, askEnabledValue])
}, [dispatch, sendExtension, sidePanelValue, autoInsertValue, autoExpandValue, apiKeyValue, serverUrlValue, modelValue, customModelValue, customModelTokensValue, translateEnableValue, languageValue, hideOnDisableAutoTranslateValue, themeValue, transDisplayValue, summarizeEnableValue, summarizeFloatValue, summarizeLanguageValue, wordsValue, fetchAmountValue, fontSizeValue, promptsValue, searchEnabledValue, cnSearchEnabledValue, askEnabledValue, chapterModeValue])
const onCancel = useCallback(() => {
window.close()
@@ -207,6 +209,10 @@ const OptionsPage = () => {
<input id='autoExpand' type='checkbox' className='toggle toggle-primary' checked={autoExpandValue}
onChange={setAutoExpandValue}/>
</FormItem>}
<FormItem title='章节模式' htmlFor='chapterMode' tip='如果视频包含章节,则会按章节分割(会导致总结只能按章节来)'>
<input id='chapterMode' type='checkbox' className='toggle toggle-primary' checked={chapterModeValue}
onChange={setChapterModeValue}/>
</FormItem>
<FormItem title='主题'>
<div className="btn-group">
<button onClick={onSelTheme1} className={classNames('btn btn-sm no-animation', (!themeValue || themeValue === 'system')?'btn-active':'')}></button>

View File

@@ -0,0 +1,23 @@
import { createSlice, PayloadAction } from '@reduxjs/toolkit'
interface CurrentTimeState {
currentTime?: number
}
const initialState: CurrentTimeState = {
currentTime: undefined
}
export const slice = createSlice({
name: 'currentTime',
initialState,
reducers: {
setCurrentTime: (state, action: PayloadAction<number | undefined>) => {
state.currentTime = action.payload
}
}
})
export const { setCurrentTime } = slice.actions
export default slice.reducer

View File

@@ -23,7 +23,7 @@ interface EnvState {
totalHeight: number
curIdx?: number // 从0开始
needScroll?: boolean
currentTime?: number
chapters?: Chapter[]
infos?: any[]
curInfo?: any
curFetched?: boolean
@@ -263,9 +263,6 @@ export const slice = createSlice({
setNeedScroll: (state, action: PayloadAction<boolean>) => {
state.needScroll = action.payload
},
setCurrentTime: (state, action: PayloadAction<number | undefined>) => {
state.currentTime = action.payload
},
setUrl: (state, action: PayloadAction<string | undefined>) => {
state.url = action.payload
},
@@ -278,6 +275,9 @@ export const slice = createSlice({
setAuthor: (state, action: PayloadAction<string | undefined>) => {
state.author = action.payload
},
setChapters: (state, action: PayloadAction<Chapter[]>) => {
state.chapters = action.payload
},
setInfos: (state, action: PayloadAction<any[]>) => {
state.infos = action.payload
},
@@ -337,7 +337,6 @@ export const {
setCurIdx,
setEnvData,
setEnvReady,
setCurrentTime,
setInfos,
setCurInfo,
setCurFetched,
@@ -351,6 +350,7 @@ export const {
mergeAskInfo,
setCtime,
setAuthor,
setChapters,
} = slice.actions
export default slice.reducer

View File

@@ -1,9 +1,11 @@
import {configureStore} from '@reduxjs/toolkit'
import envReducer from './redux/envReducer'
import currentTimeReducer from './redux/currentTimeReducer'
const store = configureStore({
reducer: {
env: envReducer,
currentTime: currentTimeReducer,
},
})

10
src/typings.d.ts vendored
View File

@@ -30,6 +30,9 @@ interface EnvData {
theme?: 'system' | 'light' | 'dark'
fontSize?: 'normal' | 'large'
// chapter
chapterMode?: boolean // 是否启用章节模式undefined/null/true表示启用false表示禁用
// search
searchEnabled?: boolean
cnSearchEnabled?: boolean
@@ -88,12 +91,19 @@ interface TranscriptItem {
idx: number
}
interface Chapter {
from: number
to: number
content: string // 标题
}
interface Segment {
items: TranscriptItem[]
startIdx: number // 从1开始
endIdx: number
text: string
fold?: boolean
chapterTitle?: string // 章节标题
summaries: {
[type: string]: Summary
}