edit short-url code

short-url/src/index.ts

@@ -37,9 +37,15 @@ export default {
 		console.log(`search: ${search}`);
 
 		if (search[0] === '?') {
-				let key = generateRandomString(6);
+				let key;
+				if (pathname.length >= 2) {
+					key = pathname.substr(1);
+				}
+				else {
+					key = generateRandomString(6);
 					while (await env.KV_STORAGE.get(key) !== null)
 						key = generateRandomString(6);
+				}
 
 				await env.KV_STORAGE.put(key, search.substr(1));
 				return new Response(`<a href=${protocol}//${host}/${key}>${protocol}//${host}/${key}</a>`, { headers: { 'Content-Type': 'text/html'}});