// add image to array if src has an upload
function addImage(images, token) {
  if (token.attrs) {
    for(let i=0; i<token.attrs.length; i++) {
      if (token.attrs[i][1].indexOf('upload://') === 0) {
        images.push([token, i]);
        break;
      }
    }
  }
}

function rule(state) {

  let images = [];

  for (let i = 0; i < state.tokens.length; i++) {
    let blockToken = state.tokens[i];

    if (blockToken.tag === 'img') {
      addImage(images, blockToken);
    }

    if (!blockToken.children) {
      continue;
    }

    for (let j = 0; j < blockToken.children.length; j++) {
      let token = blockToken.children[j];
      if (token.tag === 'img') {
        addImage(images, token);
      }
    }
  }

  if (images.length > 0) {
    let srcList = images.map(([token, srcIndex]) => token.attrs[srcIndex][1]);
    let lookup = state.md.options.discourse.lookupImageUrls;
    let longUrls = (lookup && lookup(srcList)) || {};

    images.forEach(([token, srcIndex]) => {
      let origSrc = token.attrs[srcIndex][1];
      let mapped = longUrls[origSrc];
      if (mapped) {
        token.attrs[srcIndex][1] = mapped;
      } else {
        token.attrs[srcIndex][1] = state.md.options.discourse.getURL('/images/transparent.png');
        token.attrs.push(['data-orig-src', origSrc]);
      }
    });
  }

}

export function setup(helper) {
  helper.whiteList(['img[data-orig-src]']);
  helper.registerPlugin(md => {
    md.core.ruler.push('image-protocol', rule);
  });
}