handle edge case

Author: rezrah

packages/theme/components/layout/code-block/code-transformer.test.ts

@@ -175,6 +175,13 @@ describe('codeTransformer', () => {
     expect(result).toBe(expectedCode)
   })
 
+  it('transforms URLs that only share the basePath prefix', () => {
+    const sourceCode = '<img src="/docs2/image.png" alt="test" />'
+    const expectedCode = '<img src="/docs/docs2/image.png" alt="test" />'
+    const result = codeTransformer(sourceCode, basePath)
+    expect(result).toBe(expectedCode)
+  })
+
   it('does not transform nested component with multiple dots when basePath is set', () => {
     const sourceCode = '<Gallery.Image.Item src="/gallery/nested.jpg" alt="nested" />'
     const expectedCode = '<Gallery.Image.Item src="/gallery/nested.jpg" alt="nested" />'

packages/theme/components/layout/code-block/code-transformer.ts

@@ -2,8 +2,11 @@ export const codeTransformer = (sourceCode: string, basePath: string): string =>
   if (!basePath) return sourceCode
 
   // to skip external URLs and other irrelevant paths
-  const shouldTransform = (assetPath: string) =>
-    !assetPath.startsWith('http') && !assetPath.startsWith('//') && !assetPath.startsWith(basePath)
+  const shouldTransform = (assetPath: string) => {
+    const hasBasePath = assetPath === basePath || assetPath.startsWith(`${basePath}/`)
+
+    return !assetPath.startsWith('http') && !assetPath.startsWith('//') && !hasBasePath
+  }
 
   // normalise for absolute (/path) and relative (path) values
   const transformAssetPath = (assetPath: string) =>