Merge branch 'feat/icon'

1 files changed, 62 insertions, 0 deletions

diff --git a/backend/fortee/fortee.go b/backend/fortee/fortee.go
new file mode 100644
index 0000000..cfe4eff
--- /dev/null
+++ b/backend/fortee/fortee.go
@@ -0,0 +1,62 @@
+package fortee
+
+import (
+	"context"
+	"errors"
+	"net/http"
+)
+
+const (
+	Endpoint = "https://fortee.jp"
+)
+
+var (
+	ErrLoginFailed  = errors.New("fortee login failed")
+	ErrUserNotFound = errors.New("fortee user not found")
+)
+
+func Login(ctx context.Context, username string, password string) (string, error) {
+	client, err := NewClientWithResponses(Endpoint, WithRequestEditorFn(addAcceptHeader))
+	if err != nil {
+		return "", err
+	}
+	res, err := client.PostLoginWithFormdataBodyWithResponse(ctx, PostLoginFormdataRequestBody{
+		Username: username,
+		Password: password,
+	})
+	if err != nil {
+		return "", err
+	}
+	if res.StatusCode() != http.StatusOK {
+		return "", ErrLoginFailed
+	}
+	resOk := res.JSON200
+	if !resOk.LoggedIn {
+		return "", ErrLoginFailed
+	}
+	if resOk.User == nil {
+		return "", ErrLoginFailed
+	}
+	return resOk.User.Username, nil
+}
+
+func GetUserAvatarURL(ctx context.Context, username string) (string, error) {
+	client, err := NewClientWithResponses(Endpoint, WithRequestEditorFn(addAcceptHeader))
+	if err != nil {
+		return "", err
+	}
+	res, err := client.GetUserWithResponse(ctx, username)
+	if err != nil {
+		return "", err
+	}
+	if res.StatusCode() != http.StatusOK {
+		return "", ErrUserNotFound
+	}
+	return res.JSON200.AvatarURL, nil
+}
+
+// fortee API denies requests without Accept header.
+func addAcceptHeader(_ context.Context, req *http.Request) error {
+	req.Header.Set("Accept", "application/json")
+	return nil
+}