diff --git a/__tests__/Str-test.js b/__tests__/Str-test.js index e91c0d5c..463ee3df 100644 --- a/__tests__/Str-test.js +++ b/__tests__/Str-test.js @@ -102,7 +102,8 @@ describe('Str.sanitizeURL', () => { expect(Str.sanitizeURL('google.com')).toBe('https://google.com'); expect(Str.sanitizeURL('Https://google.com')).toBe('https://google.com'); expect(Str.sanitizeURL('https://GOOgle.com')).toBe('https://google.com'); - expect(Str.sanitizeURL('FOO.com/blah_BLAH')).toBe('https://foo.com/blah_BLAH'); + +expect(Str.sanitizeURL('FOO.com/blah_BLAH', 'http')).toBe('http://foo.com/blah_BLAH'); + +expect(Str.sanitizeURL('example.com', 'http')).toBe('http://example.com'); expect(Str.sanitizeURL('http://FOO.com/blah_BLAH')).toBe('http://foo.com/blah_BLAH'); expect(Str.sanitizeURL('HTtp://FOO.com/blah_BLAH')).toBe('http://foo.com/blah_BLAH'); }); diff --git a/lib/str.ts b/lib/str.ts index 2c367730..d32fbfa9 100644 --- a/lib/str.ts +++ b/lib/str.ts @@ -971,15 +971,16 @@ const Str = { * Formats a URL by converting the domain name to lowercase and adding the missing 'https://' protocol. * * @param url The URL to be formatted + * @param scheme The Scheme to use in the URL * @returns The formatted URL */ - sanitizeURL(url: string): string { + sanitizeURL(url: string, scheme = 'https'): string { const regex = new RegExp(`^${UrlPatterns.URL_REGEX}$`, 'i'); const match = regex.exec(url); if (!match) { return url; } - const website = match[3] ? match[2] : `https://${match[2]}`; + const website = match[3] ? match[2] : `${scheme}://${match[2]}`; return website.toLowerCase() + this.cutBefore(match[1], match[2]); },