Merge remote-tracking branch 'backend/main'

1 files changed, 156 insertions, 0 deletions

diff --git a/backend/internal/ibd/userinfo.go b/backend/internal/ibd/userinfo.go
new file mode 100644
index 0000000..ed61497
--- /dev/null
+++ b/backend/internal/ibd/userinfo.go
@@ -0,0 +1,156 @@
+package ibd
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"io"
+	"log/slog"
+	"net/http"
+)
+
+const (
+	userInfoUrl = "https://myibd.investors.com/services/userprofile.aspx?format=json"
+)
+
+func (c *Client) UserInfo(ctx context.Context, cookie *http.Cookie) (*UserProfile, error) {
+	req, err := http.NewRequestWithContext(ctx, http.MethodGet, userInfoUrl, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	req.AddCookie(cookie)
+
+	resp, err := c.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer func(Body io.ReadCloser) {
+		_ = Body.Close()
+	}(resp.Body)
+
+	content, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return nil, fmt.Errorf("failed to read response body: %w", err)
+	}
+
+	if resp.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf(
+			"unexpected status code %d: %s",
+			resp.StatusCode,
+			string(content),
+		)
+	}
+
+	up := new(UserProfile)
+	if err = up.UnmarshalJSON(content); err != nil {
+		return nil, err
+	}
+
+	return up, nil
+}
+
+type UserStatus string
+
+const (
+	UserStatusUnknown    UserStatus = ""
+	UserStatusVisitor    UserStatus = "Visitor"
+	UserStatusSubscriber UserStatus = "Subscriber"
+)
+
+type UserProfile struct {
+	DisplayName string
+	Email       string
+	FirstName   string
+	LastName    string
+	Status      UserStatus
+}
+
+func (u *UserProfile) UnmarshalJSON(bytes []byte) error {
+	var resp userProfileResponse
+	if err := json.Unmarshal(bytes, &resp); err != nil {
+		return err
+	}
+
+	u.DisplayName = resp.UserProfile.UserDisplayName
+	u.Email = resp.UserProfile.UserEmailAddress
+	u.FirstName = resp.UserProfile.UserFirstName
+	u.LastName = resp.UserProfile.UserLastName
+
+	switch resp.UserProfile.UserTrialStatus {
+	case "Visitor":
+		u.Status = UserStatusVisitor
+	case "Subscriber":
+		u.Status = UserStatusSubscriber
+	default:
+		slog.Warn("Unknown user status", "status", resp.UserProfile.UserTrialStatus)
+		u.Status = UserStatusUnknown
+	}
+
+	return nil
+}
+
+type userProfileResponse struct {
+	UserProfile userProfile `json:"userProfile"`
+}
+
+type userProfile struct {
+	UserSubType            string `json:"userSubType"`
+	UserId                 string `json:"userId"`
+	UserDisplayName        string `json:"userDisplayName"`
+	Countrycode            string `json:"countrycode"`
+	IsEUCountry            string `json:"isEUCountry"`
+	Log                    string `json:"log"`
+	AgeGroup               string `json:"ageGroup"`
+	Gender                 string `json:"gender"`
+	InvestingExperience    string `json:"investingExperience"`
+	NumberOfTrades         string `json:"numberOfTrades"`
+	Occupation             string `json:"occupation"`
+	TypeOfInvestments      string `json:"typeOfInvestments"`
+	UserEmailAddress       string `json:"userEmailAddress"`
+	UserEmailAddressSHA1   string `json:"userEmailAddressSHA1"`
+	UserEmailAddressSHA256 string `json:"userEmailAddressSHA256"`
+	UserEmailAddressMD5    string `json:"userEmailAddressMD5"`
+	UserFirstName          string `json:"userFirstName"`
+	UserLastName           string `json:"userLastName"`
+	UserZip                string `json:"userZip"`
+	UserTrialStatus        string `json:"userTrialStatus"`
+	UserProductsOnTrial    string `json:"userProductsOnTrial"`
+	UserProductsOwned      string `json:"userProductsOwned"`
+	UserAdTrade            string `json:"userAdTrade"`
+	UserAdTime             string `json:"userAdTime"`
+	UserAdHold             string `json:"userAdHold"`
+	UserAdJob              string `json:"userAdJob"`
+	UserAdAge              string `json:"userAdAge"`
+	UserAdOutSell          string `json:"userAdOutSell"`
+	UserVisitCount         string `json:"userVisitCount"`
+	RoleLeaderboard        bool   `json:"role_leaderboard"`
+	RoleOws                bool   `json:"role_ows"`
+	RoleIbdlive            bool   `json:"role_ibdlive"`
+	RoleFounderclub        bool   `json:"role_founderclub"`
+	RoleEibd               bool   `json:"role_eibd"`
+	RoleIcom               bool   `json:"role_icom"`
+	RoleEtables            bool   `json:"role_etables"`
+	RoleTru10              bool   `json:"role_tru10"`
+	RoleMarketsurge        bool   `json:"role_marketsurge"`
+	RoleSwingtrader        bool   `json:"role_swingtrader"`
+	RoleAdfree             bool   `json:"role_adfree"`
+	RoleMarketdiem         bool   `json:"role_marketdiem"`
+	RoleWsjPlus            bool   `json:"role_wsj_plus"`
+	RoleWsj                bool   `json:"role_wsj"`
+	RoleBarrons            bool   `json:"role_barrons"`
+	RoleMarketwatch        bool   `json:"role_marketwatch"`
+	UserAdRoles            string `json:"userAdRoles"`
+	TrialDailyPrintNeg     bool   `json:"trial_daily_print_neg"`
+	TrialDailyPrintNon     bool   `json:"trial_daily_print_non"`
+	TrialWeeklyPrintNeg    bool   `json:"trial_weekly_print_neg"`
+	TrialWeeklyPrintNon    bool   `json:"trial_weekly_print_non"`
+	TrialDailyComboNeg     bool   `json:"trial_daily_combo_neg"`
+	TrialDailyComboNon     bool   `json:"trial_daily_combo_non"`
+	TrialWeeklyComboNeg    bool   `json:"trial_weekly_combo_neg"`
+	TrialWeeklyComboNon    bool   `json:"trial_weekly_combo_non"`
+	TrialEibdNeg           bool   `json:"trial_eibd_neg"`
+	TrialEibdNon           bool   `json:"trial_eibd_non"`
+	UserVideoPreference    string `json:"userVideoPreference"`
+	UserProfessionalStatus bool   `json:"userProfessionalStatus"`
+}