d

Author: markhuang1212

.account

@@ -0,0 +1 @@
+ACCOUNT_NAME ACCOUNT_PASSWD

backend/internal/api/setup_router.go

@@ -43,7 +43,7 @@ func SetupRouter(cc *core.CoreController) *gin.Engine {
 
 		cc.GradeQueue <- gr
 
-		c.Header("Location", "/api/result/"+gr.Id)
+		c.Header("Location", ApiPrefix+gr.Id)
 		c.Status(202)
 
 	})

backend/internal/auth/auth.go

@@ -0,0 +1,51 @@
+package auth
+
+import (
+	"bufio"
+	"errors"
+	"io"
+	"strings"
+	"sync"
+)
+
+type AuthController struct {
+	accounts map[string]string
+	lock     sync.RWMutex
+}
+
+func (ac *AuthController) Add(userName string, password string) {
+	ac.lock.Lock()
+	defer ac.lock.Unlock()
+
+	ac.accounts[userName] = password
+}
+
+func (ac *AuthController) Auth(userName string, password string) bool {
+	ac.lock.RLock()
+	defer ac.lock.RUnlock()
+
+	psd, ok := ac.accounts[userName]
+	return ok && psd == password
+}
+
+func NewAuthController(accountInfo io.Reader) (*AuthController, error) {
+	ret := AuthController{
+		accounts: make(map[string]string),
+	}
+	scanner := bufio.NewScanner(accountInfo)
+
+	for {
+		if !scanner.Scan() {
+			break
+		}
+		text := scanner.Text()
+		token := strings.Split(text, " ")
+		if len(token) != 2 {
+			return nil, errors.New("wrong format")
+		}
+		ret.Add(token[0], token[1])
+	}
+
+	return &ret, nil
+
+}

backend/internal/auth/auth_test.go

@@ -0,0 +1,24 @@
+package auth_test
+
+import (
+	"strings"
+	"testing"
+
+	"github.com/markhuang1212/code-grader/backend/internal/auth"
+	"github.com/stretchr/testify/assert"
+)
+
+var authInfo string = `user1 123456
+user2 abcdef
+`
+
+func TestAuth(t *testing.T) {
+	info := strings.NewReader(authInfo)
+	ac, err := auth.NewAuthController(info)
+	assert.Nil(t, err)
+
+	assert.True(t, ac.Auth("user1", "123456"))
+	assert.True(t, ac.Auth("user2", "abcdef"))
+
+	assert.False(t, ac.Auth("user2", "123456"))
+}