From 47e8df758bb8b8f655f8a2b82603785306a992f9 Mon Sep 17 00:00:00 2001
From: bitsnaps <bitsnaps@yahoo.fr>
Date: Sat, 16 Feb 2019 21:53:35 +0100
Subject: [PATCH] Create an enum for WooCommerce REST-api version

---
 .../java/me/gilo/woodroid/Woocommerce.java    | 26 ++++++++++++++++---
 1 file changed, 22 insertions(+), 4 deletions(-)

diff --git a/woodroid/src/main/java/me/gilo/woodroid/Woocommerce.java b/woodroid/src/main/java/me/gilo/woodroid/Woocommerce.java
index 4a3daad..5f1c862 100644
--- a/woodroid/src/main/java/me/gilo/woodroid/Woocommerce.java
+++ b/woodroid/src/main/java/me/gilo/woodroid/Woocommerce.java
@@ -9,16 +9,34 @@ import java.util.List;
 
 public class Woocommerce {
 
+    public static final ApiVersion API_V1 = ApiVersion.API_VERSION1;
+    public static final ApiVersion API_V2 = ApiVersion.API_VERSION2;
+
     private String baseUrl;
     private String siteUrl;
-    private String apiVerion;
+    private ApiVersion apiVerion;
     private String consumerKey;
     private String consumerSecret;
 
     final ProductRepository productRepository;
     final CouponService couponService;
 
-    public Woocommerce(String siteUrl, String apiVerion, String consumerKey, String consumerSecret) {
+    enum ApiVersion {
+        API_VERSION1{
+            @Override
+            public String toString() {
+                return "1";
+            }
+        },
+        API_VERSION2{
+            @Override
+            public String toString() {
+                return "2";
+            }
+        };
+    }
+
+    public Woocommerce(String siteUrl, ApiVersion apiVerion, String consumerKey, String consumerSecret) {
         this.siteUrl = siteUrl;
         this.apiVerion = apiVerion;
         this.consumerKey = consumerKey;
@@ -33,7 +51,7 @@ public class Woocommerce {
 
     public static class Builder {
         private String siteUrl;
-        private String apiVerion;
+        private ApiVersion apiVerion;
         private String consumerKey;
         private String consumerSecret;
 
@@ -46,7 +64,7 @@ public class Woocommerce {
             return this;
         }
 
-        public Builder setApiVersion(String apiVerion) {
+        public Builder setApiVersion(ApiVersion apiVerion) {
             this.apiVerion = apiVerion;
             return this;
         }