feat(rubick): rename project and add production release bundle

msn/msn_screener.go

@@ -0,0 +1,228 @@
+package msn
+
+import (
+	"encoding/json"
+	"fmt"
+
+	"github.com/enetx/g"
+)
+
+// ScreenerFilter represents available screener filter presets
+type ScreenerFilter string
+
+const (
+	FilterTopPerformers   ScreenerFilter = "top-performers"
+	FilterWorstPerformers ScreenerFilter = "worst-performers"
+	FilterHighDividend    ScreenerFilter = "high-dividend"
+	FilterLowPE           ScreenerFilter = "low-pe"
+	Filter52WeekHigh      ScreenerFilter = "52w-high"
+	Filter52WeekLow       ScreenerFilter = "52w-low"
+	FilterHighVolume      ScreenerFilter = "high-volume"
+	FilterLargeMarketCap  ScreenerFilter = "large-cap"
+)
+
+// Filter key mappings for MSN Screener API
+var screenerFilterKeys = map[ScreenerFilter]string{
+	FilterTopPerformers:   "st_list_topperfs",
+	FilterWorstPerformers: "st_list_poorperfs",
+	FilterHighDividend:    "st_list_highdividend",
+	FilterLowPE:           "st_list_lowpe",
+	Filter52WeekHigh:      "st_list_52wkhi",
+	Filter52WeekLow:       "st_list_52wklow",
+	FilterHighVolume:      "st_list_highvol",
+	FilterLargeMarketCap:  "st_list_largecap",
+}
+
+// Region key mappings for MSN Screener API
+var screenerRegionKeys = map[string]string{
+	"id": "st_reg_id", // Indonesia
+	"us": "st_reg_us", // United States
+	"gb": "st_reg_gb", // United Kingdom
+	"de": "st_reg_de", // Germany
+	"jp": "st_reg_jp", // Japan
+	"hk": "st_reg_hk", // Hong Kong
+	"sg": "st_reg_sg", // Singapore
+	"au": "st_reg_au", // Australia
+	"in": "st_reg_in", // India
+	"cn": "st_reg_cn", // China
+}
+
+// ScreenerConfig holds screener configuration
+type ScreenerConfig struct {
+	Region    string         // Country code (e.g., "id" for Indonesia)
+	Filter    ScreenerFilter // Filter preset
+	Limit     int            // Max results
+	PageIndex int            // Page number (0-indexed)
+}
+
+// ScreenerAPIResponse is the raw response from Finance/Screener
+type ScreenerAPIResponse struct {
+	Count    int           `json:"count"`
+	MatchIDs []string      `json:"matchIds"`
+	Quote    []QuoteData   `json:"quote"`
+	Equity   []EquityData  `json:"equity"`
+	Fund     []interface{} `json:"fund"`
+}
+
+// RunScreener executes the stock screener with given configuration
+func (c *MSNClient) RunScreener(config ScreenerConfig) (*ScreenerResponse, error) {
+	if config.Region == "" {
+		config.Region = "id" // Default to Indonesia
+	}
+	if config.Limit <= 0 {
+		config.Limit = 50
+	}
+
+	// Build filter array
+	filters := buildScreenerFilters(config.Region, config.Filter)
+
+	req := ScreenerRequest{
+		Filter:          filters,
+		Order:           ScreenerOrder{Key: "st_1yr_asc_order", Dir: "desc"},
+		ReturnValueType: []string{"quote", "equity"},
+		ScreenerType:    "stock",
+		Limit:           config.Limit,
+	}
+
+	reqBody, err := json.Marshal(req)
+	if err != nil {
+		return nil, fmt.Errorf("failed to marshal screener request: %w", err)
+	}
+
+	c.waitForRateLimit()
+
+	apiURL := fmt.Sprintf("%sFinance/Screener?apikey=%s&wrapodata=false",
+		MSNAssetsBaseURL,
+		MSNAPIKey,
+	)
+
+	httpReq := c.client.Post(g.String(apiURL)).
+		SetHeaders("Content-Type", "text/plain;charset=UTF-8")
+	for k, v := range c.commonHeaders() {
+		httpReq = httpReq.SetHeaders(k, v)
+	}
+	httpReq = httpReq.Body(g.String(string(reqBody)))
+
+	resp := httpReq.Do()
+	if resp.IsErr() {
+		return nil, fmt.Errorf("screener request failed: %w", resp.Err())
+	}
+
+	r := resp.Ok()
+	if r.StatusCode != 200 {
+		body := r.Body.String().Ok().Std()
+		return nil, fmt.Errorf("screener API returned status %d: %s", r.StatusCode, body)
+	}
+
+	body := r.Body.String().Ok().Std()
+
+	var apiResp ScreenerAPIResponse
+	if err := json.Unmarshal([]byte(body), &apiResp); err != nil {
+		return nil, fmt.Errorf("failed to parse screener response: %w", err)
+	}
+
+	// Merge quote and equity data into ScreenerStock
+	stocks := mergeScreenerResults(apiResp)
+
+	return &ScreenerResponse{
+		Value:    stocks,
+		Total:    apiResp.Count,
+		Count:    apiResp.Count,
+		MatchIDs: apiResp.MatchIDs,
+	}, nil
+}
+
+// buildScreenerFilters creates filter array based on region and preset
+func buildScreenerFilters(region string, filter ScreenerFilter) []ScreenerFilterItem {
+	filters := make([]ScreenerFilterItem, 0, 2)
+
+	// Add filter preset
+	if filterKey, ok := screenerFilterKeys[filter]; ok {
+		filters = append(filters, ScreenerFilterItem{
+			Key:      filterKey,
+			KeyGroup: "st_list_",
+			IsRange:  false,
+		})
+	}
+
+	// Add region filter
+	if regionKey, ok := screenerRegionKeys[region]; ok {
+		filters = append(filters, ScreenerFilterItem{
+			Key:      regionKey,
+			KeyGroup: "st_reg_",
+			IsRange:  false,
+		})
+	}
+
+	return filters
+}
+
+// mergeScreenerResults combines quote and equity data into ScreenerStock slice
+func mergeScreenerResults(apiResp ScreenerAPIResponse) []ScreenerStock {
+	// Build equity map by instrumentId
+	equityMap := make(map[string]*EquityData)
+	for i := range apiResp.Equity {
+		eq := &apiResp.Equity[i]
+		// Use instrumentId from the "_p" field if available
+		if id := eq.ID; id != "" {
+			equityMap[id] = eq
+		}
+	}
+
+	stocks := make([]ScreenerStock, 0, len(apiResp.Quote))
+	for _, q := range apiResp.Quote {
+		stock := ScreenerStock{
+			ID:             q.InstrumentID,
+			InstrumentID:   q.InstrumentID,
+			Symbol:         q.Symbol,
+			ShortName:      q.ShortName,
+			DisplayName:    q.DisplayName,
+			ExchangeID:     q.ExchangeID,
+			ExchangeCode:   q.ExchangeCode,
+			Country:        q.Country,
+			Price:          q.Price,
+			PriceChange:    q.PriceChange,
+			PriceChangePct: q.PriceChangePct,
+			MarketCap:      q.MarketCap,
+			Volume:         q.AccumulatedVolume,
+			Price52wHigh:   q.Price52wHigh,
+			Price52wLow:    q.Price52wLow,
+			Return1Year:    q.Return1Year,
+			ReturnYTD:      q.ReturnYTD,
+		}
+
+		// Merge equity data if available
+		if eq, ok := equityMap[q.InstrumentID]; ok {
+			stock.Sector = eq.Sector
+			stock.Industry = eq.Industry
+		}
+
+		stocks = append(stocks, stock)
+	}
+
+	return stocks
+}
+
+// ParseScreenerFilter converts string to ScreenerFilter
+func ParseScreenerFilter(s string) (ScreenerFilter, error) {
+	switch s {
+	case "top-performers", "top":
+		return FilterTopPerformers, nil
+	case "worst-performers", "worst":
+		return FilterWorstPerformers, nil
+	case "high-dividend", "dividend":
+		return FilterHighDividend, nil
+	case "low-pe", "pe":
+		return FilterLowPE, nil
+	case "52w-high", "52high":
+		return Filter52WeekHigh, nil
+	case "52w-low", "52low":
+		return Filter52WeekLow, nil
+	case "high-volume", "volume":
+		return FilterHighVolume, nil
+	case "large-cap", "largecap":
+		return FilterLargeMarketCap, nil
+	default:
+		return "", fmt.Errorf("unknown filter: %s (valid: top-performers, worst-performers, high-dividend, low-pe, 52w-high, 52w-low, high-volume, large-cap)", s)
+	}
+}