125 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			125 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
| const visit = require('unist-util-visit');
 | |
| const { selectAll, select } = require('hast-util-select');
 | |
| 
 | |
| const { sectionFilter } = require('./utils');
 | |
| 
 | |
| const seedRE = /(.+)-seed$/;
 | |
| const headRE = /(.+)-setup$/;
 | |
| const tailRE = /(.+)-teardown$/;
 | |
| 
 | |
| const editableRegionMarker = '--fcc-editable-region--';
 | |
| 
 | |
| function defaultFile(lang) {
 | |
|   return {
 | |
|     key: `index${lang}`,
 | |
|     ext: lang,
 | |
|     name: 'index',
 | |
|     contents: '',
 | |
|     head: '',
 | |
|     tail: ''
 | |
|   };
 | |
| }
 | |
| function createCodeGetter(codeKey, regEx, seeds) {
 | |
|   return container => {
 | |
|     const {
 | |
|       properties: { id }
 | |
|     } = container;
 | |
|     const lang = id.match(regEx)[1];
 | |
|     const key = `index${lang}`;
 | |
|     const code = select('code', container).children[0].value;
 | |
|     if (key in seeds) {
 | |
|       seeds[key] = {
 | |
|         ...seeds[key],
 | |
|         [codeKey]: code
 | |
|       };
 | |
|     } else {
 | |
|       seeds[key] = {
 | |
|         ...defaultFile(lang),
 | |
|         [codeKey]: code
 | |
|       };
 | |
|     }
 | |
|   };
 | |
| }
 | |
| 
 | |
| // TODO: any reason to worry about CRLF?
 | |
| 
 | |
| function findRegionMarkers(file) {
 | |
|   const lines = file.contents.split('\n');
 | |
|   const editableLines = lines
 | |
|     .map((line, id) => (line.trim() === editableRegionMarker ? id : -1))
 | |
|     .filter(id => id >= 0);
 | |
| 
 | |
|   if (editableLines.length > 2) {
 | |
|     throw Error('Editable region has too many markers' + editableLines);
 | |
|   }
 | |
| 
 | |
|   if (editableLines.length === 0) {
 | |
|     return null;
 | |
|   } else if (editableLines.length === 1) {
 | |
|     throw Error(`Editable region not closed`);
 | |
|   } else {
 | |
|     return editableLines;
 | |
|   }
 | |
| }
 | |
| 
 | |
| function removeLines(contents, toRemove) {
 | |
|   const lines = contents.split('\n');
 | |
|   return lines.filter((_, id) => !toRemove.includes(id)).join('\n');
 | |
| }
 | |
| 
 | |
| function createPlugin() {
 | |
|   return function transformer(tree, file) {
 | |
|     function visitor(node) {
 | |
|       if (sectionFilter(node, 'challengeSeed')) {
 | |
|         let seeds = {};
 | |
|         const codeDivs = selectAll('div', node);
 | |
|         const seedContainers = codeDivs.filter(({ properties: { id } }) =>
 | |
|           seedRE.test(id)
 | |
|         );
 | |
|         seedContainers.forEach(createCodeGetter('contents', seedRE, seeds));
 | |
| 
 | |
|         const headContainers = codeDivs.filter(({ properties: { id } }) =>
 | |
|           headRE.test(id)
 | |
|         );
 | |
|         headContainers.forEach(createCodeGetter('head', headRE, seeds));
 | |
| 
 | |
|         const tailContainers = codeDivs.filter(({ properties: { id } }) =>
 | |
|           tailRE.test(id)
 | |
|         );
 | |
|         tailContainers.forEach(createCodeGetter('tail', tailRE, seeds));
 | |
| 
 | |
|         file.data = {
 | |
|           ...file.data,
 | |
|           files: seeds
 | |
|         };
 | |
| 
 | |
|         // TODO: make this readable.
 | |
| 
 | |
|         Object.keys(seeds).forEach(key => {
 | |
|           const fileData = seeds[key];
 | |
|           const editRegionMarkers = findRegionMarkers(fileData);
 | |
|           if (editRegionMarkers) {
 | |
|             fileData.contents = removeLines(
 | |
|               fileData.contents,
 | |
|               editRegionMarkers
 | |
|             );
 | |
| 
 | |
|             if (editRegionMarkers[1] <= editRegionMarkers[0]) {
 | |
|               throw Error('Editable region must be non zero');
 | |
|             }
 | |
|             fileData.editableRegionBoundaries = editRegionMarkers;
 | |
|           } else {
 | |
|             fileData.editableRegionBoundaries = [];
 | |
|           }
 | |
|         });
 | |
|         // TODO: TESTS!
 | |
|       }
 | |
|     }
 | |
|     visit(tree, 'element', visitor);
 | |
|   };
 | |
| }
 | |
| 
 | |
| exports.challengeSeedToData = createPlugin;
 | |
| exports.createCodeGetter = createCodeGetter;
 | |
| exports.defaultFile = defaultFile;
 |