From 445617624a531737a9906de58754f5f22bf66c4e Mon Sep 17 00:00:00 2001 From: Nam Huynh Date: Mon, 9 Aug 2021 11:41:57 +0700 Subject: [PATCH] init module --- .gitignore | 2 + README.md | 2 +- constant.go | 10 +++++ go.mod | 8 ++++ go.sum | 120 ++++++++++++++++++++++++++++++++++++++++++++++++++ helper.go | 72 ++++++++++++++++++++++++++++++ log.go | 33 ++++++++++++++ random_otp.go | 30 +++++++++++++ send_otp.go | 106 ++++++++++++++++++++++++++++++++++++++++++++ time.go | 32 ++++++++++++++ verify_otp.go | 11 +++++ vietguys.go | 78 ++++++++++++++++++++++++++++++++ 12 files changed, 503 insertions(+), 1 deletion(-) create mode 100644 constant.go create mode 100644 go.mod create mode 100644 go.sum create mode 100644 helper.go create mode 100644 log.go create mode 100644 random_otp.go create mode 100644 send_otp.go create mode 100644 time.go create mode 100644 verify_otp.go create mode 100644 vietguys.go diff --git a/.gitignore b/.gitignore index 66fd13c..bbfd2cb 100644 --- a/.gitignore +++ b/.gitignore @@ -13,3 +13,5 @@ # Dependency directories (remove the comment below to include it) # vendor/ + +.idea \ No newline at end of file diff --git a/README.md b/README.md index 3e9bdff..27b650b 100644 --- a/README.md +++ b/README.md @@ -1 +1 @@ -# vietguys \ No newline at end of file +# vietguys diff --git a/constant.go b/constant.go new file mode 100644 index 0000000..09f9393 --- /dev/null +++ b/constant.go @@ -0,0 +1,10 @@ +package vietguys + +// Constant +const ( + logCollectionName = "log_sms_vietguys" + + SMSTypeOTP = "otp" + + SellyOTPContent = "Selly - Ma xac nhan cua ban la %s" +) diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..8b1a999 --- /dev/null +++ b/go.mod @@ -0,0 +1,8 @@ +module github.com/Selly-Modules/vietguys + +go 1.16 + +require ( + github.com/Selly-Modules/mongodb v0.0.0-20210809042025-449f377c954a + go.mongodb.org/mongo-driver v1.7.1 +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..a8b42f2 --- /dev/null +++ b/go.sum @@ -0,0 +1,120 @@ +github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= +github.com/Selly-Modules/mongodb v0.0.0-20210809042025-449f377c954a h1:ecCTXyFY+N138g5o0sXDYpCuEjAreN01kCvkCQ16j9I= +github.com/Selly-Modules/mongodb v0.0.0-20210809042025-449f377c954a/go.mod h1:hhzPI2DO0ghSTlfy5ddgBqIs9H3wtksSHLOWvDHp74M= +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/go-stack/stack v1.8.0 h1:5SgMzNM5HxrEjV0ww2lTmX6E2Izsfxas4+YHWRs3Lsk= +github.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY= +github.com/gobuffalo/attrs v0.0.0-20190224210810-a9411de4debd/go.mod h1:4duuawTqi2wkkpB4ePgWMaai6/Kc6WEz83bhFwpHzj0= +github.com/gobuffalo/depgen v0.0.0-20190329151759-d478694a28d3/go.mod h1:3STtPUQYuzV0gBVOY3vy6CfMm/ljR4pABfrTeHNLHUY= +github.com/gobuffalo/depgen v0.1.0/go.mod h1:+ifsuy7fhi15RWncXQQKjWS9JPkdah5sZvtHc2RXGlg= +github.com/gobuffalo/envy v1.6.15/go.mod h1:n7DRkBerg/aorDM8kbduw5dN3oXGswK5liaSCx4T5NI= +github.com/gobuffalo/envy v1.7.0/go.mod h1:n7DRkBerg/aorDM8kbduw5dN3oXGswK5liaSCx4T5NI= +github.com/gobuffalo/flect v0.1.0/go.mod h1:d2ehjJqGOH/Kjqcoz+F7jHTBbmDb38yXA598Hb50EGs= +github.com/gobuffalo/flect v0.1.1/go.mod h1:8JCgGVbRjJhVgD6399mQr4fx5rRfGKVzFjbj6RE/9UI= +github.com/gobuffalo/flect v0.1.3/go.mod h1:8JCgGVbRjJhVgD6399mQr4fx5rRfGKVzFjbj6RE/9UI= +github.com/gobuffalo/genny v0.0.0-20190329151137-27723ad26ef9/go.mod h1:rWs4Z12d1Zbf19rlsn0nurr75KqhYp52EAGGxTbBhNk= +github.com/gobuffalo/genny v0.0.0-20190403191548-3ca520ef0d9e/go.mod h1:80lIj3kVJWwOrXWWMRzzdhW3DsrdjILVil/SFKBzF28= +github.com/gobuffalo/genny v0.1.0/go.mod h1:XidbUqzak3lHdS//TPu2OgiFB+51Ur5f7CSnXZ/JDvo= +github.com/gobuffalo/genny v0.1.1/go.mod h1:5TExbEyY48pfunL4QSXxlDOmdsD44RRq4mVZ0Ex28Xk= +github.com/gobuffalo/gitgen v0.0.0-20190315122116-cc086187d211/go.mod h1:vEHJk/E9DmhejeLeNt7UVvlSGv3ziL+djtTr3yyzcOw= +github.com/gobuffalo/gogen v0.0.0-20190315121717-8f38393713f5/go.mod h1:V9QVDIxsgKNZs6L2IYiGR8datgMhB577vzTDqypH360= +github.com/gobuffalo/gogen v0.1.0/go.mod h1:8NTelM5qd8RZ15VjQTFkAW6qOMx5wBbW4dSCS3BY8gg= +github.com/gobuffalo/gogen v0.1.1/go.mod h1:y8iBtmHmGc4qa3urIyo1shvOD8JftTtfcKi+71xfDNE= +github.com/gobuffalo/logger v0.0.0-20190315122211-86e12af44bc2/go.mod h1:QdxcLw541hSGtBnhUc4gaNIXRjiDppFGaDqzbrBd3v8= +github.com/gobuffalo/mapi v1.0.1/go.mod h1:4VAGh89y6rVOvm5A8fKFxYG+wIW6LO1FMTG9hnKStFc= +github.com/gobuffalo/mapi v1.0.2/go.mod h1:4VAGh89y6rVOvm5A8fKFxYG+wIW6LO1FMTG9hnKStFc= +github.com/gobuffalo/packd v0.0.0-20190315124812-a385830c7fc0/go.mod h1:M2Juc+hhDXf/PnmBANFCqx4DM3wRbgDvnVWeG2RIxq4= +github.com/gobuffalo/packd v0.1.0/go.mod h1:M2Juc+hhDXf/PnmBANFCqx4DM3wRbgDvnVWeG2RIxq4= +github.com/gobuffalo/packr/v2 v2.0.9/go.mod h1:emmyGweYTm6Kdper+iywB6YK5YzuKchGtJQZ0Odn4pQ= +github.com/gobuffalo/packr/v2 v2.2.0/go.mod h1:CaAwI0GPIAv+5wKLtv8Afwl+Cm78K/I/VCm/3ptBN+0= +github.com/gobuffalo/syncx v0.0.0-20190224160051-33c29581e754/go.mod h1:HhnNqWY95UYwwW3uSASeV7vtgYkT2t16hJgV3AEPUpw= +github.com/golang/snappy v0.0.1 h1:Qgr9rKW7uDUkrbSmQeiDsGa8SjGyCOGtuasMWwvp2P4= +github.com/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q= +github.com/google/go-cmp v0.5.2 h1:X2ev0eStA3AbceY54o37/0PQ/UWqKEiiO2dKL5OPaFM= +github.com/google/go-cmp v0.5.2/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/inconshreveable/mousetrap v1.0.0/go.mod h1:PxqpIevigyE2G7u3NXJIT2ANytuPF1OarO4DADm73n8= +github.com/joho/godotenv v1.3.0/go.mod h1:7hK45KPybAkOC6peb+G5yklZfMxEjkZhHbwpqxOKXbg= +github.com/karrick/godirwalk v1.8.0/go.mod h1:H5KPZjojv4lE+QYImBI8xVtrBRgYrIVsaRPx4tDPEn4= +github.com/karrick/godirwalk v1.10.3/go.mod h1:RoGL9dQei4vP9ilrpETWE8CLOZ1kiN0LhBygSwrAsHA= +github.com/klauspost/compress v1.9.5 h1:U+CaK85mrNNb4k8BNOfgJtJ/gr6kswUCFj6miSzVC6M= +github.com/klauspost/compress v1.9.5/go.mod h1:RyIbtBH6LamlWaDj8nUwkbUhJ87Yi3uG0guNDohfE1A= +github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ= +github.com/konsorten/go-windows-terminal-sequences v1.0.2/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ= +github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo= +github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= +github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= +github.com/logrusorgru/aurora v2.0.3+incompatible h1:tOpm7WcpBTn4fjmVfgpQq0EfczGlG91VSDkswnjF5A8= +github.com/logrusorgru/aurora v2.0.3+incompatible/go.mod h1:7rIyQOR62GCctdiQpZ/zOJlFyk6y+94wXzv6RNZgaR4= +github.com/markbates/oncer v0.0.0-20181203154359-bf2de49a0be2/go.mod h1:Ld9puTsIW75CHf65OeIOkyKbteujpZVXDpWK6YGZbxE= +github.com/markbates/safe v1.0.1/go.mod h1:nAqgmRi7cY2nqMc92/bSEeQA+R4OheNU2T1kNSCBdG0= +github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe/go.mod h1:wL8QJuTMNUDYhXwkmfOly8iTdp5TEcJFWZD2D7SIkUc= +github.com/pelletier/go-toml v1.7.0/go.mod h1:vwGMzjaWMwyfHwgIBhI2YUM4fB6nL6lVAvS1LBMMhTE= +github.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= +github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= +github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4= +github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= +github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/rogpeppe/go-internal v1.1.0/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4= +github.com/rogpeppe/go-internal v1.2.2/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4= +github.com/rogpeppe/go-internal v1.3.0/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4= +github.com/sirupsen/logrus v1.4.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPxbbu5VWo= +github.com/sirupsen/logrus v1.4.1/go.mod h1:ni0Sbl8bgC9z8RoU9G6nDWqqs/fq4eDPysMBDgk/93Q= +github.com/sirupsen/logrus v1.4.2/go.mod h1:tLMulIdttU9McNUspp0xgXVQah82FyeX6MwdIuYE2rE= +github.com/spf13/cobra v0.0.3/go.mod h1:1l0Ry5zgKvJasoi3XT1TypsSe7PqH0Sj9dhYf7v3XqQ= +github.com/spf13/pflag v1.0.3/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnInEg4= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs= +github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= +github.com/stretchr/testify v1.6.1 h1:hDPOHmpOpP40lSULcqw7IrRb/u7w6RpDC9399XyoNd0= +github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/tidwall/pretty v1.0.0 h1:HsD+QiTn7sK6flMKIvNmpqz1qrpP3Ps6jOKIKMooyg4= +github.com/tidwall/pretty v1.0.0/go.mod h1:XNkn88O1ChpSDQmQeStsy+sBenx6DDtFZJxhVysOjyk= +github.com/xdg-go/pbkdf2 v1.0.0 h1:Su7DPu48wXMwC3bs7MCNG+z4FhcyEuz5dlvchbq0B0c= +github.com/xdg-go/pbkdf2 v1.0.0/go.mod h1:jrpuAogTd400dnrH08LKmI/xc1MbPOebTwRqcT5RDeI= +github.com/xdg-go/scram v1.0.2 h1:akYIkZ28e6A96dkWNJQu3nmCzH3YfwMPQExUYDaRv7w= +github.com/xdg-go/scram v1.0.2/go.mod h1:1WAq6h33pAW+iRreB34OORO2Nf7qel3VV3fjBj+hCSs= +github.com/xdg-go/stringprep v1.0.2 h1:6iq84/ryjjeRmMJwxutI51F2GIPlP5BfTvXHeYjyhBc= +github.com/xdg-go/stringprep v1.0.2/go.mod h1:8F9zXuvzgwmyT5DUm4GUfZGDdT3W+LCvS6+da4O5kxM= +github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d h1:splanxYIlg+5LfHAM6xpdFEAYOk8iySO56hMFq6uLyA= +github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d/go.mod h1:rHwXgn7JulP+udvsHwJoVG1YGAP6VLg4y9I5dyZdqmA= +go.mongodb.org/mongo-driver v1.7.1 h1:jwqTeEM3x6L9xDXrCxN0Hbg7vdGfPBOTIkr0+/LYZDA= +go.mongodb.org/mongo-driver v1.7.1/go.mod h1:Q4oFMbo1+MSNqICAdYMlC/zSTrwCogR4R8NzkI+yfU8= +golang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= +golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= +golang.org/x/crypto v0.0.0-20190422162423-af44ce270edf/go.mod h1:WFFai1msRO1wXaEeE5yQxYXgSfI8pQAWXbQop6sCtWE= +golang.org/x/crypto v0.0.0-20200302210943-78000ba7a073 h1:xMPOj6Pz6UipU1wXLkrtqpHbR0AVFnyPEQq/wRWz9lM= +golang.org/x/crypto v0.0.0-20200302210943-78000ba7a073/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= +golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/sync v0.0.0-20190227155943-e225da77a7e6/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20190412183630-56d357773e84/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e h1:vcxGaoTs7kV8m5Np9uUNQin4BrLOthgV7252N8V+FwY= +golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190403152447-81d4e9dc473e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20190419153524-e8e3143a4f4a/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20190422165155-953cdadca894/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20190531175056-4c3a928424d2/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= +golang.org/x/text v0.3.5 h1:i6eZZ+zk0SOf0xgBpEpPD18qWcJda6q1sxt3S0kzyUQ= +golang.org/x/text v0.3.5/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20190329151228-23e29df326fe/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= +golang.org/x/tools v0.0.0-20190416151739-9c9e1878f421/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= +golang.org/x/tools v0.0.0-20190420181800-aa740d480789/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= +golang.org/x/tools v0.0.0-20190531172133-b3315ee88b7d/go.mod h1:/rFqwRUd4F7ZHNgwSSTFct+R/Kf4OFW1sUzUTQQTgfc= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI= +gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c h1:dUUwHk2QECo/6vqA44rthZ8ie2QXMNeKRTHCNY2nXvo= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= diff --git a/helper.go b/helper.go new file mode 100644 index 0000000..26ffce2 --- /dev/null +++ b/helper.go @@ -0,0 +1,72 @@ +package vietguys + +import ( + "time" + + "go.mongodb.org/mongo-driver/bson" +) + +const ( + otpValidMinute = 30 +) + +// Check phone or ip can send sms +func (s Service) checkCanSend(phone, ip string) bool { + var ( + canSend = true + startOfToday = startOfDate(time.Now()) + ) + + // Check phone number first + if len(phone) > 0 && s.PhoneMaxSendPerDay > 0 { + total, _ := s.DB.CountDocuments(bgCtx, bson.M{ + "phoneNumber": phone, + "createdAt": bson.M{ + "$gte": startOfToday, + }, + }) + canSend = total > int64(s.PhoneMaxSendPerDay) + } + + // Check ip, but only check if can send + if canSend && len(ip) > 0 && s.IPMaxSendPerDay > 0 { + total, _ := s.DB.CountDocuments(bgCtx, bson.M{ + "ip": ip, + "createdAt": bson.M{ + "$gte": startOfToday, + }, + }) + canSend = total > int64(s.IPMaxSendPerDay) + } + + return canSend +} + +// Check otp right or not +func (s Service) checkOTP(phone, otpCode string) bool { + var ( + timeAgo = timeBeforeNowInMin(otpValidMinute) + ) + + total, _ := s.DB.CountDocuments(bgCtx, bson.M{ + "phoneNumber": phone, + "code": otpCode, + "isCodeValid": true, + "createdAt": bson.M{ + "$gte": timeAgo, + }, + }) + isValid := total > 0 + + // If success, set code valid to false + if isValid { + s.DB.UpdateOne(bgCtx, bson.M{ + "code": otpCode, + }, bson.M{ + "$set": bson.M{ + "isCodeValid": false, + }, + }) + } + return isValid +} diff --git a/log.go b/log.go new file mode 100644 index 0000000..d13a7d9 --- /dev/null +++ b/log.go @@ -0,0 +1,33 @@ +package vietguys + +import ( + "fmt" + "time" + + "go.mongodb.org/mongo-driver/bson/primitive" +) + +// Log ... +type Log struct { + ID primitive.ObjectID `bson:"_id"` + Carrier string `bson:"carrier"` + Type string `bson:"type"` + PhoneNumber string `bson:"phoneNumber"` + Code string `bson:"code"` + IsCodeValid bool `bson:"isCodeValid"` + Content string `bson:"content"` + IP string `bson:"ip"` + Success bool `bson:"success"` + Result string `bson:"result"` + CreatedAt time.Time `bson:"createdAt"` + + tableName string +} + +// Save log to db +func (s Service) saveLog(doc Log) { + if _, err := s.DB.InsertOne(bgCtx, doc); err != nil { + fmt.Println("*** Error when create log", err) + fmt.Println("*** Log", doc) + } +} diff --git a/random_otp.go b/random_otp.go new file mode 100644 index 0000000..a4d00e3 --- /dev/null +++ b/random_otp.go @@ -0,0 +1,30 @@ +package vietguys + +import ( + "math/rand" + "time" +) + +// Constant ... +const ( + otpCharacters = "123456789" + otpLength = 6 +) + +// randomIntBetweenRange ... +func randomIntBetweenRange(min int, max int) int { + rand.Seed(time.Now().UnixNano()) + return rand.Intn(max-min) + min +} + +// randomOTPCode ... +func randomOTPCode() string { + var code string + var length = len(otpCharacters) + for i := 0; i < otpLength; i++ { + // Random character index + randIndex := randomIntBetweenRange(0, length) + code += string(otpCharacters[randIndex]) + } + return code +} diff --git a/send_otp.go b/send_otp.go new file mode 100644 index 0000000..1016951 --- /dev/null +++ b/send_otp.go @@ -0,0 +1,106 @@ +package vietguys + +import ( + "encoding/json" + "errors" + "fmt" + "io/ioutil" + "net/http" + "net/url" + "strings" + + "github.com/Selly-Modules/mongodb" +) + +// SendOTPResult ... +type SendOTPResult struct { + Carrier string + Error int + ErrorCode int + MsgID string + Message string + Log string +} + +// SendOTP to phone number +// Phone format: 84935123456 +// Content must have "%s", for otp code substitution +func (s Service) SendOTP(phone, ip, contentFormat string) error { + // Generate otp code + otpCode := randomOTPCode() + + // Format content + content := fmt.Sprintf(contentFormat, otpCode) + + // Just remove char "+" if existed + strings.Replace(phone, "+", "", 1) + + // Check that phone or ip is not over quota + canSend := s.checkCanSend(phone, ip) + if canSend { + return errors.New("ip or phone has reached over limited quota per day") + } + + // Create payload + params := url.Values{} + params.Add("u", s.User) + params.Add("pwd", s.Password) + params.Add("from", s.From) + params.Add("json", "1") + params.Add("phone", phone) + params.Add("sms", content) + payload := strings.NewReader(params.Encode()) + + // Create request + client := s.Client + req, err := http.NewRequest(http.MethodPost, s.Endpoint, payload) + if err != nil { + return err + } + + // Add necessary headers + req.Header.Add("Content-Type", "application/x-www-form-urlencoded") + + // Call + res, err := client.Do(req) + if err != nil { + return err + } + + // Make sure close body + defer res.Body.Close() + + // Ready body + body, err := ioutil.ReadAll(res.Body) + rawResult := string(body) + fmt.Println(rawResult) + if err != nil { + fmt.Println("error : ", err.Error()) + return err + } + + var result SendOTPResult + if err = json.Unmarshal(body, &result); err != nil { + return err + } + + go func() { + // Save log to db + log := Log{ + ID: mongodb.NewObjectID(), + Carrier: result.Carrier, + Type: SMSTypeOTP, + Code: otpCode, + IsCodeValid: true, + PhoneNumber: phone, + IP: ip, + Content: content, + CreatedAt: now(), + Success: result.Error == 0, + Result: rawResult, + } + s.saveLog(log) + }() + + return nil +} diff --git a/time.go b/time.go new file mode 100644 index 0000000..41035ba --- /dev/null +++ b/time.go @@ -0,0 +1,32 @@ +package vietguys + +import "time" + +// +// NOTE: due to unique timezone in server's code, all using time will be convert to HCM timezone (UTC +7) +// All functions generate time, must be call util functions here +// WARNING: don't accept call time.Now() directly +// + +const timezoneHCM = "Asia/Ho_Chi_Minh" + +// getHCMLocation ... +func getHCMLocation() *time.Location { + l, _ := time.LoadLocation(timezoneHCM) + return l +} + +// Now ... +func now() time.Time { + return time.Now().In(getHCMLocation()) +} + +// TimeBeforeNowInMin ... +func timeBeforeNowInMin(min int) time.Time { + return time.Now().Add(time.Minute * time.Duration(min) * -1).In(getHCMLocation()) +} + +// StartOfDate ... +func startOfDate(t time.Time) time.Time { + return time.Date(t.Year(), t.Month(), t.Day(), 0, 0, 0, 0, getHCMLocation()) +} diff --git a/verify_otp.go b/verify_otp.go new file mode 100644 index 0000000..83a0fae --- /dev/null +++ b/verify_otp.go @@ -0,0 +1,11 @@ +package vietguys + +import "strings" + +// VerifyOTP verify otp code is right or not +func (s Service) VerifyOTP(phone, otpCode string) bool { + // Just remove char "+" if existed + strings.Replace(phone, "+", "", 1) + + return s.checkOTP(phone, otpCode) +} diff --git a/vietguys.go b/vietguys.go new file mode 100644 index 0000000..c7c580f --- /dev/null +++ b/vietguys.go @@ -0,0 +1,78 @@ +package vietguys + +import ( + "context" + "errors" + "fmt" + "net/http" + + "github.com/Selly-Modules/mongodb" + "go.mongodb.org/mongo-driver/mongo" +) + +// MongoDBConfig ... +type MongoDBConfig struct { + Host, User, Password, DBName, Mechanism, Source string +} + +// Config ... +type Config struct { + // Endpoint which will use to call api + Endpoint string + // For auth + User string + // For auth + Password string + // Brand name + From string + // MongoDB config, for save documents + MongoDB MongoDBConfig + // Limit send time per ip + IPMaxSendPerDay int + // Limit send time per phone number + PhoneMaxSendPerDay int +} + +// Service ... +type Service struct { + Config + Client *http.Client + DB *mongo.Collection +} + +var s *Service +var bgCtx = context.Background() + +// NewInstance for using send sms method +func NewInstance(config Config) error { + if config.Endpoint == "" || config.User == "" || config.Password == "" || config.From == "" || config.MongoDB.Host == "" { + return errors.New("please provide all information that needed: endpoint, user, password, from, mongodb") + } + + // Connect MongoDB + err := mongodb.Connect( + config.MongoDB.Host, + config.MongoDB.User, + config.MongoDB.Password, + config.MongoDB.DBName, + config.MongoDB.Mechanism, + config.MongoDB.Source, + ) + if err != nil { + fmt.Println("Cannot init module VIETGUYS", err) + return err + } + + s = &Service{ + Config: config, + Client: &http.Client{}, + DB: mongodb.GetInstance().Collection(logCollectionName), + } + + return nil +} + +// GetInstance ... +func GetInstance() *Service { + return s +}