Ocelot-Social/backend/src/middleware/fixImageUrlsMiddleware.js

const legacyUrls = [
  'https://api-alpha.human-connection.org',
  'https://staging-api.human-connection.org',
  'http://localhost:3000',
]

export const fixUrl = url => {
  legacyUrls.forEach(legacyUrl => {
    url = url.replace(legacyUrl, '/api')
  })
  return url
}

const checkUrl = thing => {
  return (
    thing &&
    typeof thing === 'string' &&
    legacyUrls.find(legacyUrl => {
      return thing.indexOf(legacyUrl) === 0
    })
  )
}

export const fixImageURLs = (result, recursive) => {
  if (checkUrl(result)) {
    result = fixUrl(result)
  } else if (result && Array.isArray(result)) {
    result.forEach((res, index) => {
      result[index] = fixImageURLs(result[index], true)
    })
  } else if (result && typeof result === 'object') {
    Object.keys(result).forEach(key => {
      result[key] = fixImageURLs(result[key], true)
    })
  }
  return result
}

export default {
  Mutation: async (resolve, root, args, context, info) => {
    const result = await resolve(root, args, context, info)
    return fixImageURLs(result)
  },
  Query: async (resolve, root, args, context, info) => {
    let result = await resolve(root, args, context, info)
    return fixImageURLs(result)
  },
}