add credentials_for

spec/http_auth_store_spec.cr

@@ -6,8 +6,19 @@ describe "Mechanize AuthStore test" do
     url = URI.parse("http://example.com/")
     user = "kanezoh"
     password = "password"
+    realm = ""
     auth_store.add_auth(url, user, password)
-    auth_store.auth_accounts.size.should eq 1
-    auth_store.auth_accounts[url].should eq({"kanezoh", "password"})
+    auth_store.auth_accounts[url].size.should eq 1
+    auth_store.auth_accounts[url][realm].should eq(["kanezoh", "password", nil])
+  end
+
+  it "credentials_for" do
+    auth_store = Mechanize::HTTP::AuthStore.new
+    url = URI.parse("http://example.com/")
+    user = "kanezoh"
+    password = "password"
+    realm = ""
+    auth_store.add_auth(url, user, password)
+    auth_store.credentials_for(url, realm).should eq ["kanezoh", "password", nil]
   end
 end

src/mechanize/http/auth_store.cr

@@ -2,25 +2,28 @@ class Mechanize
   module HTTP
     # This class store info for HTTP Authentication.
     class AuthStore
-      getter auth_accounts : Hash(URI, Tuple(String, String))
+      getter auth_accounts : Hash(URI, Hash(String, Array(String?)))
 
       def initialize
-        @auth_accounts = Hash(URI, Tuple(String, String)).new
+        @auth_accounts = Hash(URI, Hash(String, Array(String?))).new
       end
 
-      def add_auth(uri, user, pass)
+      def add_auth(uri : String | URI, user : String, pass : String, realm : String? = nil, domain : String? = nil)
         unless uri.is_a?(URI)
           uri = URI.new(uri)
         end
         # uri += '/'
         uri.user = nil
         uri.password = nil
-
-        auth_accounts[uri] = {user, pass}
-      end ##
+        if realm.nil?
+          realm = ""
+        end
+        realm_hash = {realm => [user, pass, domain]}
+        auth_accounts[uri] = realm_hash
+      end
 
       # Retrieves credentials for +realm+ on the server at +uri+.
-      def credentials_for(uri, realm) : Tuple(String)
+      def credentials_for(uri : String | URI, realm : String) : Array(String?)
         unless uri.is_a?(URI)
           uri = URI.new(uri)
         end
@@ -28,7 +31,9 @@ class Mechanize
         uri.user = nil
         uri.password = nil
 
-        auth_accounts[uri]
+        realms = auth_accounts[uri]
+
+        realms[realm]
       end
     end
   end