mirror of https://github.com/jkjoy/sunpeiwen.git
59 lines
1.8 KiB
JavaScript
59 lines
1.8 KiB
JavaScript
|
hexo.extend.helper.register('findArchiveLength', function (func) {
|
||
|
|
const allPostsLength = this.site.posts.length
|
||
|
|
if (hexo.config.archive_generator && hexo.config.archive_generator.enable === false) return allPostsLength
|
||
|
|
const { yearly, monthly, daily } = hexo.config.archive_generator
|
||
|
|
const { year, month, day } = this.page
|
||
|
|
if (yearly === false || !year) return allPostsLength
|
||
|
|
|
||
|
|
const posts = this.site.posts.sort('date')
|
||
|
|
|
||
|
|
const compareFunc = (type, y1, m1, d1, y2, m2, d2) => {
|
||
|
|
if (type === 'year') {
|
||
|
|
return y1 === y2
|
||
|
|
} else if (type === 'month') {
|
||
|
|
return y1 === y2 && m1 === m2
|
||
|
|
} else if (type === 'day') {
|
||
|
|
return y1 === y2 && m1 === m2 && d1 === d2
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
const generateDateObj = (type) => {
|
||
|
|
const dateObj = []
|
||
|
|
let length = 0
|
||
|
|
|
||
|
|
posts.forEach(post => {
|
||
|
|
const date = post.date.clone()
|
||
|
|
const year = date.year()
|
||
|
|
const month = date.month() + 1
|
||
|
|
const day = date.date()
|
||
|
|
const lastData = dateObj[length - 1]
|
||
|
|
|
||
|
|
if (!lastData || !compareFunc(type, lastData.year, lastData.month, lastData.day, year, month, day)) {
|
||
|
|
const name = type === 'year' ? year : type === 'month' ? `${year}-${month}` : `${year}-${month}-${day}`
|
||
|
|
length = dateObj.push({
|
||
|
|
name,
|
||
|
|
year,
|
||
|
|
month,
|
||
|
|
day,
|
||
|
|
count: 1
|
||
|
|
})
|
||
|
|
} else {
|
||
|
|
lastData.count++
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
return dateObj
|
||
|
|
}
|
||
|
|
|
||
|
|
const data = func('createArchiveObj', () => {
|
||
|
|
const yearObj = yearly ? generateDateObj('year') : []
|
||
|
|
const monthObj = monthly ? generateDateObj('month') : []
|
||
|
|
const dayObj = daily ? generateDateObj('day') : []
|
||
|
|
const fullObj = [...yearObj, ...monthObj, ...dayObj]
|
||
|
|
return fullObj
|
||
|
|
})
|
||
|
|
|
||
|
|
const name = month ? day ? `${year}-${month}-${day}` : `${year}-${month}` : year
|
||
|
|
return data.find(item => item.name === name).count
|
||
|
|
})
|