From b8482fa7a1e3351ef4af4d168c6c7be5eef1da57 Mon Sep 17 00:00:00 2001
From: lwolodkiewicz <l.wolodkiewicz@systemycallcenter.pl>
Date: Tue, 30 Mar 2021 08:55:02 +0200
Subject: [PATCH] Allow defaultCountry prop to accept both iso2 and dialCode

---
 docs/documentation/v3.md             |  2 +-
 src/components/vue-tel-input.test.js | 11 +++++++++++
 src/components/vue-tel-input.vue     | 21 +++++++++++++++++----
 3 files changed, 29 insertions(+), 5 deletions(-)

diff --git a/docs/documentation/v3.md b/docs/documentation/v3.md
index 1d2ebac8..2471c097 100644
--- a/docs/documentation/v3.md
+++ b/docs/documentation/v3.md
@@ -107,7 +107,7 @@ Read more on `vue-form-generator`'s [instruction page](https://icebob.gitbooks.i
   | -------- | ---- | ------------- | ----------- |
   | `autocomplete` | `String` | `'on'` | Native input 'autocomplete' attribute |
   | `autofocus` | `Boolean` | `false` | Native input 'autofocus' attribute |
-  | `defaultCountry` | `String` | `''` | Default country, will override the country fetched from IP address of user |
+  | `defaultCountry` | `String`  &#124; `Number` | `''` | Default country (by iso2 or dialCode), will override the country fetched from IP address of user |
   | `disabled` | `Boolean` | `false` | Disable input field |
   | `disabledFetchingCountry` | `Boolean` | `false` | Disable fetching current country based on IP address of user |
   | `dropdownOptions` | `Object` | `{ disabledDialCode: false, tabindex: 0 }` | Options for dropdown, supporting `disabledDialCode` and `tabindex`| 
diff --git a/src/components/vue-tel-input.test.js b/src/components/vue-tel-input.test.js
index 35c9f3e9..1db6dc17 100644
--- a/src/components/vue-tel-input.test.js
+++ b/src/components/vue-tel-input.test.js
@@ -58,6 +58,17 @@ describe('Props', () => {
       expect(wrapper.find('.vti__selection > .vti__flag').classes()).toContain('au');
     });
   });
+  describe(':defaultCountryByDialCode', () => {
+    it('shows correct default country by dial code', async () => {
+      const wrapper = shallowMount(VueTelInput, {
+        propsData: {
+          defaultCountry: 48,
+        },
+      });
+      await Vue.nextTick();
+      expect(wrapper.find('.vti__selection > .vti__flag').classes()).toContain('pl');
+    });
+  });
   describe(':disabled', () => {
     it('adds disabled class to component', () => {
       const wrapper = shallowMount(VueTelInput, {
diff --git a/src/components/vue-tel-input.vue b/src/components/vue-tel-input.vue
index ef907c54..754a4821 100644
--- a/src/components/vue-tel-input.vue
+++ b/src/components/vue-tel-input.vue
@@ -109,7 +109,7 @@ export default {
     defaultCountry: {
       // Default country code, ie: 'AU'
       // Will override the current country of user
-      type: String,
+      type: [String, Number],
       default: () => getDefault('defaultCountry'),
     },
     disabled: {
@@ -350,9 +350,19 @@ export default {
          * 2. Use default country if passed from parent
          */
         if (this.defaultCountry) {
-          this.choose(this.defaultCountry);
-          resolve();
-          return;
+          if (typeof this.defaultCountry === 'string') {
+            this.choose(this.defaultCountry);
+            resolve();
+            return;
+          }
+          if (typeof this.defaultCountry === 'number') {
+            const country = this.findCountryByDialCode(this.defaultCountry);
+            if (country) {
+              this.choose(country.iso2);
+              resolve();
+              return;
+            }
+          }
         }
 
         const fallbackCountry = this.preferredCountries[0] || this.filteredCountries[0];
@@ -394,6 +404,9 @@ export default {
     findCountry(iso = '') {
       return this.filteredCountries.find((country) => country.iso2 === iso.toUpperCase());
     },
+    findCountryByDialCode(dialCode) {
+      return this.filteredCountries.find((country) => Number(country.dialCode) === dialCode);
+    },
     getItemClass(index, iso2) {
       const highlighted = this.selectedIndex === index;
       const lastPreferred = index === this.preferredCountries.length - 1;