pgmoon.crypto hmac_sha256 to support resty.openssl

Author: bungle

pgmoon/crypto.lua

@@ -29,11 +29,28 @@ else
   end
 end
 local hmac_sha256
-hmac_sha256 = function(key, str)
-  local openssl_hmac = require("openssl.hmac")
-  local hmac = assert(openssl_hmac.new(key, "sha256"))
-  hmac:update(str)
-  return assert(hmac:final())
+if pcall(function()
+  return require("openssl.hmac")
+end) then
+  hmac_sha256 = function(key, str)
+    local openssl_hmac = require("openssl.hmac")
+    local hmac = assert(openssl_hmac.new(key, "sha256"))
+    hmac:update(str)
+    return assert(hmac:final())
+  end
+elseif pcall(function()
+  return require("resty.openssl.hmac")
+end) then
+  hmac_sha256 = function(key, str)
+    local openssl_hmac = require("resty.openssl.hmac")
+    local hmac = assert(openssl_hmac.new(key, "sha256"))
+    hmac:update(str)
+    return assert(hmac:final())
+  end
+else
+  hmac_sha256 = function()
+    return error("Either luaossl or resty.openssl is required to calculate hmac sha256 digest")
+  end
 end
 local digest_sha256
 digest_sha256 = function(str)

pgmoon/crypto.moon

@@ -13,12 +13,24 @@ elseif pcall -> require "crypto"
 else
   -> error "Either luaossl (recommended) or LuaCrypto is required to calculate md5"
 
-hmac_sha256 = (key, str) ->
-  openssl_hmac = require("openssl.hmac")
-  hmac = assert openssl_hmac.new(key, "sha256")
 
-  hmac\update str
-  assert hmac\final!
+hmac_sha256 = if pcall -> require "openssl.hmac"
+  (key, str) ->
+    openssl_hmac = require("openssl.hmac")
+    hmac = assert openssl_hmac.new(key, "sha256")
+
+    hmac\update str
+    assert hmac\final!
+elseif pcall -> require "resty.openssl.hmac"
+  (key, str) ->
+    openssl_hmac = require("resty.openssl.hmac")
+    hmac = assert openssl_hmac.new(key, "sha256")
+
+    hmac\update str
+    assert hmac\final!
+else
+  -> error "Either luaossl or resty.openssl is required to calculate hmac sha256 digest"
+
 
 digest_sha256 = (str) ->
   digest = assert require("openssl.digest").new("sha256")