Support parsing remote sitemap indexes

This commit is contained in:
Sam Carlton 2022-06-13 13:22:44 -05:00
parent ee44d3b746
commit 50af45d99c
2 changed files with 81 additions and 8 deletions

View file

@ -7,12 +7,25 @@ import {
sitemapLocation,
sitemapIndexFileName,
} from '~/helpers/constants.js'
import { isValidHttpUrl } from '~/helpers/check-types.js'
const sitemapFilesToTry = [
sitemapIndexFileName,
'sitemap.xml'
]
function getDomainFromString ( urlString ) {
const url = new URL( urlString )
return url.hostname
}
function getPathFromUrl ( urlString ) {
const url = new URL( urlString )
return url.pathname
}
export function parseSitemapXml ( sitemapXml ) {
// Get URLs from index
const sitemapRoot = parse( sitemapXml )
@ -39,8 +52,23 @@ export function parseSitemapXml ( sitemapXml ) {
}
export async function getAllUrlsFromLocalSitemap ( sitemapPath ) {
return await getAllUrlsFromSitemap( sitemapPath, {
getMethod: async ( sitemapPath ) => {
const sitemapXml = await fs.readFile( sitemapPath, 'utf8' )
return sitemapXml
}
})
}
export async function getAllUrlsFromSitemap ( sitemapSource, { getMethod } = {} ) {
const sitemapPath = isValidHttpUrl( sitemapSource ) ? getPathFromUrl( sitemapSource ) : sitemapSource
// console.log('sitemapPath', sitemapPath)
// Get intial sitemap
const sitemapXml = await fs.readFile( sitemapPath, 'utf8' )
const sitemapXml = await getMethod( sitemapSource )// fs.readFile( sitemapPath, 'utf8' )
const sitemapDirectory = path.dirname( sitemapPath )
// Get URLs from index
@ -59,15 +87,58 @@ export async function getAllUrlsFromLocalSitemap ( sitemapPath ) {
const childSitemapPath = path.join( sitemapDirectory, sitemapUrl.pathname )
return await getAllUrlsFromLocalSitemap( childSitemapPath )
return await getAllUrlsFromSitemap( childSitemapPath, { getMethod } )
}))
// Flatten array
return sitemaps.flat()
}
export async function fetchParsedSitemapXmlForDomain ( domain ) {
export async function fetchAllUrlsFromSitemaps ( urlString ) {
// const domain = getDomainFromString( urlString )
const allUrls = new Map()
for ( const sitemapFile of sitemapFilesToTry ) {
const sitemapUrl = new URL( sitemapFile, urlString )
// console.log( 'sitemapUrl', sitemapUrl.href )
// Just do a quich HEAD request to see if the file exists with getting the whole body
const exists = await axios.head( sitemapUrl.href )
.catch( () => false )
.then( response => {
// console.log( 'response', response.status )
return response.status < 300
} )
// console.log( 'exists', exists )
// Skip if sitemap doesn't exist
if ( !exists ) continue
// Fetch the whole sitemap
const urls = await getAllUrlsFromSitemap( sitemapUrl.href , {
getMethod: async sitemapPath => {
const sitemapUrl = new URL( sitemapPath, urlString )
const sitemapXml = await axios.get( sitemapUrl.href )
.then( response => response.data )
return sitemapXml
}
})
// console.log( 'Sitemap urls', urls )
// Set our urls to the map
urls.forEach( tag => allUrls.set( tag.loc, new URL( tag.loc ) ) )
// Store sitemap urls to context
// const urlsMap = new Map( urls.map( tag => [ tag.loc, new URL( tag.loc )] ) )
}
return allUrls
}