Adding check for external protocol in uri

bmaguireibm · bmaguireibm · commit 0939dbe91421 · 2018-08-10T19:07:17.000+01:00
diff --git a/src/commons/loadWeightMap.ts b/src/commons/loadWeightMap.ts
@@ -16,14 +16,16 @@ export function getModelUris(uri: string | undefined, defaultModelName: string)
       manifestUri: `/${defaultManifestFilename}`
     }
   }
+  const protocol = uri.startsWith('http://') ? 'http://' : uri.startsWith('https://') ? 'https://' : '';
+  uri = uri.replace(protocol, '');
 
   const parts = uri.split('/').filter(s => s)
 
   const manifestFile = uri.endsWith('.json')
     ? parts[parts.length - 1]
     : defaultManifestFilename
 
-  let modelBaseUri = (uri.endsWith('.json') ? parts.slice(0, parts.length - 1) : parts).join('/')
+  let modelBaseUri = protocol + (uri.endsWith('.json') ? parts.slice(0, parts.length - 1) : parts).join('/')
   modelBaseUri = uri.startsWith('/') ? `/${modelBaseUri}` : modelBaseUri
 
   return {

Original file line number	Diff line number	Diff line change
`@@ -16,14 +16,16 @@ export function getModelUris(uri: string \| undefined, defaultModelName: string)`
`16`	`16`	manifestUri: `/${defaultManifestFilename}`
`17`	`17`	`}`
`18`	`18`	`}`
	`19`	`+ const protocol = uri.startsWith('http://') ? 'http://' : uri.startsWith('https://') ? 'https://' : '';`
	`20`	`+ uri = uri.replace(protocol, '');`
`19`	`21`
`20`	`22`	`const parts = uri.split('/').filter(s => s)`
`21`	`23`
`22`	`24`	`const manifestFile = uri.endsWith('.json')`
`23`	`25`	`? parts[parts.length - 1]`
`24`	`26`	`: defaultManifestFilename`
`25`	`27`
`26`		`- let modelBaseUri = (uri.endsWith('.json') ? parts.slice(0, parts.length - 1) : parts).join('/')`
	`28`	`+ let modelBaseUri = protocol + (uri.endsWith('.json') ? parts.slice(0, parts.length - 1) : parts).join('/')`
`27`	`29`	modelBaseUri = uri.startsWith('/') ? `/${modelBaseUri}` : modelBaseUri
`28`	`30`
`29`	`31`	`return {`