index.js

/**
 * Copyright 2020 The AMP HTML Authors. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS-IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

const {readdirSync} = require('fs');

const addShortCodes = (eleventyConfig) => {
  const filesInDir = readdirSync(__dirname);

  for (const file of filesInDir) {
    if (!isShortcodeFile(file)) {
      continue;
    }
    const shortcodeName = file.replace('.js', '');
    const shortcodeImpl = require(`./${file}`);
    eleventyConfig.addShortcode(shortcodeName, shortcodeImpl);
  }
};

const isShortcodeFile = (fileName) => {
  if (!fileName.endsWith('.js')) {
    return false;
  }
  if (fileName.endsWith('index.js')) {
    return false;
  }
  if (fileName.endsWith('test.js')) {
    return false;
  }
  return true;
};

module.exports = addShortCodes;