lib/awskeyring/validate.rb in awskeyring-0.1.0 vs lib/awskeyring/validate.rb in awskeyring-0.1.1

@@ -48,7 +48,15 @@
     # @param [String] role_arn The role arn
     def self.role_arn(role_arn)
       raise 'Invalid Role ARN' unless role_arn =~ %r(\Aarn:aws:iam::[0-9]{12}:role\/\S*\z)
       role_arn
     end
+
+    # Validate an MFA CODE
+    #
+    # @param [String] mfa_code The mfa code
+    def self.mfa_code(mfa_code)
+      raise 'Invalid MFA CODE' unless mfa_code =~ /\A\d{6}\z/
+      mfa_code
+    end
   end
 end