18160146944

const { resolve } = require('path')
const _ = require('lodash')
const frontMatter = require('front-matter')
const ImmutableStack = require('../../lib/ImmutableStack')
const { isTemplateFile } = require('./helpers')
const models = {
  homepage: require('./models/homepage'),
  subpage: require('./models/subpage'),
  collection: require('./models/collection'),
  asset: require('./models/asset')
}

const linkEntries = (contentModel) => {
  contentModel.collections.forEach(collection => {
    collection.posts.forEach(post => {
      const fields = Object.keys(post)
      const linkFields = fields
        .map(key => {
          const match = key.match(/(.+){(.+)}/)
          if (!match) {
            return
          }
          const [, entrySlug, categorySlug] = post[key].match(/([^(\s]+)(?:\s*\(([^)]+)\))?/)
          return {
            key: match[1],
            collectionSlug: match[2],
            categorySlug,
            entrySlug
          }
        })
        .filter(Boolean)
      linkFields.forEach(link => {
        const collection = contentModel.collections.find(c => c.slug.match(new RegExp(link.collectionSlug, 'i')))
        const container = link.categorySlug ? // TODO: Handle subcategories
          collection.categories.find(c => c.slug.match(new RegExp(link.categorySlug, 'i'))) || collection :
          collection
        const entry = container.posts.find(p => p.slug.match(new RegExp(link.entrySlug, 'i')))
        post[link.key] = Object.assign({}, entry)
        entry.links = entry.links || {}
        entry.links.relations = entry.links.relations || []
        const relation = entry.links.relations.find(r => r.key === link.key)
        if (relation) {
          relation.entries.push(post)
        } else {
          entry.links.relations.push({
            key: link.key,
            entries: [post]
          })
        }
      })
    })
  })
}

const defaultContentModelSettings = {
  permalinkPrefix: '/',
  out: resolve('.'),
  defaultCategoryName: 'Unclassified',
  assetsDirectory: 'assets',
  pagesDirectory: 'pages',
  homepageDirectory: 'homepage',
  debug: false,
  site: {
    title: '',
    description: ''
  }
}
class ContentModel {
  constructor(contentModelSettings = defaultContentModelSettings, contentTypes = []) {
    this.settings = {
      ...defaultContentModelSettings,
      ...contentModelSettings
    }
    this.contentTypes = contentTypes
  }

  create(fileSystemTree) {
    const indexFileNameOptions = ['root']

    const isRootIndexFile = (node) => {
      return isTemplateFile(node) && node.name.match(
        new RegExp(`^(${indexFileNameOptions.join('|')})\\..+$`)
      )
    }

    const indexFile = fileSystemTree.find(isRootIndexFile)
    const indexProps = indexFile ? frontMatter(indexFile.content) : {}

    const context = new ImmutableStack([{
      key: 'root',
      outputPath: this.settings.out,
      permalink: this.settings.permalinkPrefix
    }])

    this.models = {
      collection: models.collection({
        defaultCategoryName: this.settings.defaultCategoryName,
        collectionAliases: [
          ...this.contentTypes
            .filter(ct => ct.model === 'collection')
            .map(ct => ct.collectionAlias),
          ...(indexProps.attributes?.collectionAliases || [])
        ]
      }, this.contentTypes),

      subpage: models.subpage({
        pagesDirectory: this.settings.pagesDirectory
      }),

      homepage: models.homepage({
        homepageDirectory: this.settings.homepageDirectory
      }),

      asset: models.asset({
        assetsDirectory: this.settings.assetsDirectory
      })
    }

    this.contentModel = {
      homepage: this.models.homepage.create({
        name: 'index',
        extension: 'md',
        content: ''
      }, context),
      subpages: [],
      collections: [],
      assets: []
    }

    fileSystemTree.forEach(node => {
      if (this.models.homepage.match(node)) {
        this.contentModel.homepage = this.models.homepage.create(node, context)
        return
      }

      if (this.models.subpage.match(node)) {
        return this.contentModel.subpages.push(
          this.models.subpage.create(node, context)
        )
      }

      if (this.models.subpage.matchPagesDirectory(node)) {
        return node.children.forEach(childNode => {
          if (this.models.subpage.match(childNode)) {
            this.contentModel.subpages.push(
              this.models.subpage.create(childNode, context)
            )
          } else if (this.models.asset.match(childNode)) {
            this.contentModel.assets.push(
              this.models.asset.create(childNode, context)
            )
          }
        })
      }

      if (this.models.collection.match(node)) {
        return this.contentModel.collections.push(
          this.models.collection.create(node, context)
        )
      }

      if (this.models.asset.matchAssetsDirectory(node)) {
        return this.contentModel.assets.push(
          ...node.children.map(childNode => {
            return this.models.asset.create(childNode, context)
          })
        )
      }

      if (this.models.asset.match(node)) {
        return this.contentModel.assets.push(
          this.models.asset.create(node, context)
        )
      }
    })

    this.afterEffects()
    return this.contentModel
  }

  afterEffects() {
    linkEntries(this.contentModel)

    this.contentModel.collections.forEach(collection => {
      this.models.collection.afterEffects(this.contentModel, collection)
    })

    this.contentModel.subpages.forEach(subpage => {
      this.models.subpage.afterEffects(this.contentModel, subpage)
    })

    this.models.homepage.afterEffects(this.contentModel, this.contentModel.homepage)

    this.contentModel.assets.forEach(asset => {
      this.models.asset.afterEffects(this.contentModel, asset)
    })
  }

  render(renderer) {
    const renderHomepage = () => {
      return this.models.homepage.render(renderer, this.contentModel.homepage, {
        contentModel: this.contentModel,
        settings: this.settings,
        debug: this.settings.debug
      })
    }

    const renderCollections = () => {
      return Promise.all(
        this.contentModel.collections.map(collection => {
          return this.models.collection.render( renderer, collection, {
            contentModel: this.contentModel,
            settings: this.settings,
            debug: this.settings.debug
          })
        })
      )
    }

    const renderSubpages = () => {
      return Promise.all(
        this.contentModel.subpages.map(subpage => {
          return this.models.subpage.render(renderer, subpage, {
            contentModel: this.contentModel,
            settings: this.settings,
            debug: this.settings.debug
          })
        })
      )
    }

    const renderAssets = () => {
      return Promise.all(
        this.contentModel.assets.map(asset => {
          return this.models.asset.render(renderer, asset, {
            contentModel: this.contentModel,
            settings: this.settings,
            debug: this.settings.debug
          })
        })
      )
    }

    return renderHomepage()
      .then(() =>
        Promise.all([
          renderCollections(),
          renderSubpages(),
          renderAssets()
        ])
      )
  }
}

module.exports = ContentModel

1	const { resolve } = require('path')	7✔
2	const _ = require('lodash')	7✔
3	const frontMatter = require('front-matter')	7✔
4	const ImmutableStack = require('../../lib/ImmutableStack')	7✔
5	const { isTemplateFile } = require('./helpers')	7✔
6	const models = {	7✔
7	homepage: require('./models/homepage'),
8	subpage: require('./models/subpage'),
9	collection: require('./models/collection'),
10	asset: require('./models/asset')
11	}
12
13	const linkEntries = (contentModel) => {	7✔
14	contentModel.collections.forEach(collection => {	7✔
15	collection.posts.forEach(post => {	7✔
16	const fields = Object.keys(post)	49✔
17	const linkFields = fields	49✔
18	.map(key => {
19	const match = key.match(/(.+){(.+)}/)	924✔
20	if (!match) {	924!
21	return	924✔
22	}
23	const [, entrySlug, categorySlug] = post[key].match(/([^(\s]+)(?:\s*\(([^)]+)\))?/)	×
24	return {	×
25	key: match[1],
26	collectionSlug: match[2],
27	categorySlug,
28	entrySlug
29	}
30	})
31	.filter(Boolean)
32	linkFields.forEach(link => {	49✔
33	const collection = contentModel.collections.find(c => c.slug.match(new RegExp(link.collectionSlug, 'i')))	×
NEW 34	const container = link.categorySlug ? // TODO: Handle subcategories	×
35	collection.categories.find(c => c.slug.match(new RegExp(link.categorySlug, 'i'))) \|\| collection :	×
36	collection
37	const entry = container.posts.find(p => p.slug.match(new RegExp(link.entrySlug, 'i')))	×
NEW 38	post[link.key] = Object.assign({}, entry)	×
39	entry.links = entry.links \|\| {}	×
40	entry.links.relations = entry.links.relations \|\| []	×
41	const relation = entry.links.relations.find(r => r.key === link.key)	×
42	if (relation) {	×
43	relation.entries.push(post)	×
44	} else {
45	entry.links.relations.push({	×
46	key: link.key,
47	entries: [post]
48	})
49	}
50	})
51	})
52	})
53	}
54
55	const defaultContentModelSettings = {	7✔
56	permalinkPrefix: '/',
57	out: resolve('.'),
58	defaultCategoryName: 'Unclassified',
59	assetsDirectory: 'assets',
60	pagesDirectory: 'pages',
61	homepageDirectory: 'homepage',
62	debug: false,
63	site: {
64	title: '',
65	description: ''
66	}
67	}
68	class ContentModel {
69	constructor(contentModelSettings = defaultContentModelSettings, contentTypes = []) {	7!
70	this.settings = {	7✔
71	...defaultContentModelSettings,
72	...contentModelSettings
73	}
74	this.contentTypes = contentTypes	7✔
75	}
76
77	create(fileSystemTree) {
78	const indexFileNameOptions = ['root']	7✔
79
80	const isRootIndexFile = (node) => {	7✔
81	return isTemplateFile(node) && node.name.match(	14✔
82	new RegExp(`^(${indexFileNameOptions.join('\|')})\\..+$`)
83	)
84	}
85
86	const indexFile = fileSystemTree.find(isRootIndexFile)	7✔
87	const indexProps = indexFile ? frontMatter(indexFile.content) : {}	7!
88
89	const context = new ImmutableStack([{	7✔
90	key: 'root',
91	outputPath: this.settings.out,
92	permalink: this.settings.permalinkPrefix
93	}])
94
95	this.models = {	7✔
96	collection: models.collection({
97	defaultCategoryName: this.settings.defaultCategoryName,
98	collectionAliases: [
99	...this.contentTypes
100	.filter(ct => ct.model === 'collection')	×
101	.map(ct => ct.collectionAlias),	×
102	...(indexProps.attributes?.collectionAliases \|\| [])	14✔
103	]
104	}, this.contentTypes),
105
106	subpage: models.subpage({
107	pagesDirectory: this.settings.pagesDirectory
108	}),
109
110	homepage: models.homepage({
111	homepageDirectory: this.settings.homepageDirectory
112	}),
113
114	asset: models.asset({
115	assetsDirectory: this.settings.assetsDirectory
116	})
117	}
118
119	this.contentModel = {	7✔
120	homepage: this.models.homepage.create({
121	name: 'index',
122	extension: 'md',
123	content: ''
124	}, context),
125	subpages: [],
126	collections: [],
127	assets: []
128	}
129
130	fileSystemTree.forEach(node => {	7✔
131	if (this.models.homepage.match(node)) {	14✔
132	this.contentModel.homepage = this.models.homepage.create(node, context)	7✔
133	return	7✔
134	}
135
136	if (this.models.subpage.match(node)) {	7!
137	return this.contentModel.subpages.push(	×
138	this.models.subpage.create(node, context)
139	)
140	}
141
142	if (this.models.subpage.matchPagesDirectory(node)) {	7!
143	return node.children.forEach(childNode => {	×
144	if (this.models.subpage.match(childNode)) {	×
145	this.contentModel.subpages.push(	×
146	this.models.subpage.create(childNode, context)
147	)
148	} else if (this.models.asset.match(childNode)) {	×
149	this.contentModel.assets.push(	×
150	this.models.asset.create(childNode, context)
151	)
152	}
153	})
154	}
155
156	if (this.models.collection.match(node)) {	7!
157	return this.contentModel.collections.push(	7✔
158	this.models.collection.create(node, context)
159	)
160	}
161
162	if (this.models.asset.matchAssetsDirectory(node)) {	×
163	return this.contentModel.assets.push(	×
164	...node.children.map(childNode => {
165	return this.models.asset.create(childNode, context)	×
166	})
167	)
168	}
169
170	if (this.models.asset.match(node)) {	×
171	return this.contentModel.assets.push(	×
172	this.models.asset.create(node, context)
173	)
174	}
175	})
176
177	this.afterEffects()	7✔
178	return this.contentModel	7✔
179	}
180
181	afterEffects() {
182	linkEntries(this.contentModel)	7✔
183
184	this.contentModel.collections.forEach(collection => {	7✔
185	this.models.collection.afterEffects(this.contentModel, collection)	7✔
186	})
187
188	this.contentModel.subpages.forEach(subpage => {	7✔
NEW 189	this.models.subpage.afterEffects(this.contentModel, subpage)	×
190	})
191
192	this.models.homepage.afterEffects(this.contentModel, this.contentModel.homepage)	7✔
193
194	this.contentModel.assets.forEach(asset => {	7✔
NEW 195	this.models.asset.afterEffects(this.contentModel, asset)	×
196	})
197	}
198
199	render(renderer) {
200	const renderHomepage = () => {	×
201	return this.models.homepage.render(renderer, this.contentModel.homepage, {	×
202	contentModel: this.contentModel,
203	settings: this.settings,
204	debug: this.settings.debug
205	})
206	}
207
208	const renderCollections = () => {	×
209	return Promise.all(	×
210	this.contentModel.collections.map(collection => {
211	return this.models.collection.render( renderer, collection, {	×
212	contentModel: this.contentModel,
213	settings: this.settings,
214	debug: this.settings.debug
215	})
216	})
217	)
218	}
219
220	const renderSubpages = () => {	×
221	return Promise.all(	×
222	this.contentModel.subpages.map(subpage => {
223	return this.models.subpage.render(renderer, subpage, {	×
224	contentModel: this.contentModel,
225	settings: this.settings,
226	debug: this.settings.debug
227	})
228	})
229	)
230	}
231
232	const renderAssets = () => {	×
233	return Promise.all(	×
234	this.contentModel.assets.map(asset => {
235	return this.models.asset.render(renderer, asset, {	×
236	contentModel: this.contentModel,
237	settings: this.settings,
238	debug: this.settings.debug
239	})
240	})
241	)
242	}
243
244	return renderHomepage()	×
245	.then(() =>
246	Promise.all([	×
247	renderCollections(),
248	renderSubpages(),
249	renderAssets()
250	])
251	)
252	}
253	}
254
255	module.exports = ContentModel	7✔

scriptype / writ-cms / 18160146944

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous