vue-wp/vue-theme/src/store/modules/arrangements.js

import { sortTypes } from '../../utils/helpers'

const _insertInBetweens = bySort => {
    const postListWithInBetweens = []
    Object.keys(bySort).forEach(term => {
        if(term) {
            postListWithInBetweens.push({ 
                slug: term,
                title: term,
                inbetween: true
            })
        }
        bySort[term].forEach(post => postListWithInBetweens.push(post))
    })
    return postListWithInBetweens
}

const _arrangeByMaterial = postsList => {
    const byMaterial = {
        clay: []
    }
    postsList.forEach(post => {
        post.materials.forEach(mat => {
            if(!byMaterial[mat]) byMaterial[mat] = []
            byMaterial[mat].push(post)
        })
    })
    const flatPacked = _insertInBetweens(byMaterial)
    return flatPacked
}

const _arrangeByType = postsList => {
    const byType = {}
    postsList.forEach(post => {
        const subtypes = post.subtypes
        subtypes.forEach(type => {
            if(!byType[type]) byType[type] = []
            byType[type].push(post)
        })
    })
    const flatPacked = _insertInBetweens(byType)
    return flatPacked
}

const _arrangeByAlpha = (postsList, seenTitles) => {
    const alphabet = [...'9abcdefghijklmnopqrstuvwxyz']
    const flatPacked = []
    
    const storeTitle = letter => {
        if(seenTitles.includes(letter)) return
        flatPacked.push({ slug: letter, title: letter, inbetween: true })
        seenTitles.push(letter)
    }
    
    postsList.forEach(post => {
        const lastWord = post.slug.split('-').filter(c => c).pop()
        const firstCharaOflastWord = lastWord[0]
        const firstCharaOfSortWord = post.sortname ? post.sortname[0] : 'z'
        
        let charaIndex = null
        let otherIndex = null
        const lastWordAlphaIndex = alphabet.indexOf(firstCharaOflastWord)
        const altWordAlphaIndex = alphabet.indexOf(firstCharaOfSortWord)
        if(lastWordAlphaIndex < altWordAlphaIndex) {
            charaIndex = lastWordAlphaIndex
            otherIndex = altWordAlphaIndex
        } else if (lastWordAlphaIndex > altWordAlphaIndex) {
            otherIndex = lastWordAlphaIndex
            charaIndex = altWordAlphaIndex
        }
        if(seenTitles.includes(alphabet[charaIndex])) {
            charaIndex = otherIndex
        }
        storeTitle(alphabet[charaIndex])
        flatPacked.push(post)
    })
    return flatPacked
}

const _arrangeByEpisode = postsList => {
    const byEpisode = {}
    postsList.forEach(post => {
        const relatedEpisode = post.related_episode
        if(!byEpisode[relatedEpisode]) byEpisode[relatedEpisode] = []
        byEpisode[relatedEpisode].push(post)
    })
    const flatPacked = _insertInBetweens(byEpisode)
    return flatPacked
}

const repackBySort = (postsList, sortedBy, seenTitles) => {
    let repacked = postsList
    if(sortedBy == sortTypes.material) {
        repacked = _arrangeByMaterial(postsList)
    } else if(sortedBy == sortTypes.episode) {
        repacked = _arrangeByEpisode(postsList)
    } else if(sortedBy == sortTypes.subtype) {
        repacked = _arrangeByType(postsList)
    } else if(sortedBy == sortTypes.alpha) {
        repacked = _arrangeByAlpha(postsList, seenTitles)
    }
    return repacked
}

const allBySlug = state => {
    return Object.values(state.all).reduce((bySlug, post) => {
        bySlug[post.slug] = post
        return bySlug
    }, {})
}

export {
    allBySlug,
    repackBySort,
    _arrangeByMaterial,
    _arrangeByType,
    _arrangeByAlpha,
    _arrangeByEpisode
}