fix(pencil): chinese letter loading

yjb94 · Jan 22, 2020 · b446528 · b446528
1 parent b836f53
commit b446528
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/index.js b/src/index.js
@@ -7,11 +7,15 @@ const ReactStoreBadges = ({
   width = 135,
   height = 40,
 }) => {
-  const shortLocale = locale.split(/[_-]/)[0];
+  const expeptionLocale = ["zh-cn", "zh-tw"];
+  let shortCode = locale;
+  if (expeptionLocale.indexOf(shortCode) === -1) {
+    shortCode = shortCode.split(/[_-]/)[0];
+  }
 
   const image = {
     ios:`https://linkmaker.itunes.apple.com/images/badges/${locale}/badge_appstore-lrg.svg`,
-    android: `https://raw.github.com/yjb94/google-play-badge-svg/master/img/${shortLocale}_get.svg?sanitize=true`
+    android: `https://raw.github.com/yjb94/google-play-badge-svg/master/img/${shortCode}_get.svg?sanitize=true`
   }
 
   return (