add feature to restrict full email address not by domain

typester · Aug 14, 2014 · 3b01529 · 3b01529
1 parent a265c9c
commit 3b01529
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/config_sample.yml b/config_sample.yml
@@ -20,7 +20,8 @@ auth:
 
 # # restrict domain. (optional)
 # domain:
-#   - yourdomain.com
+#   - yourdomain.com    # restrict by domain
+#   - example@gmail.com # or specific address
 
 # document root for static files
 htdocs: ./

diff --git a/httpd.go b/httpd.go
@@ -208,9 +208,15 @@ func restrictDomain(domain []string) martini.Handler {
 			var user *User
 			if len(domain) > 0 {
 				for _, d := range domain {
-					if strings.HasSuffix(email, "@"+d) {
-						user = &User{email}
-						break
+					if strings.Contains(d, "@") {
+						if d == email {
+							user = &User{email}
+						}
+					} else {
+						if strings.HasSuffix(email, "@"+d) {
+							user = &User{email}
+							break
+						}
 					}
 				}
 			} else {