package session import ( "context" "fmt" "hash/crc32" "net" "net/http" "strconv" "strings" "time" pscookie "prestaproxy/internal/prestashop/cookie" "gorm.io/gorm" ) type Service struct { db *gorm.DB prefix string } type defaults struct { LanguageID int64 CurrencyID int64 ShopID int64 ShopGroupID int64 CountryISO string } func NewService(db *gorm.DB, prefix string) *Service { return &Service{db: db, prefix: prefix} } func (s *Service) NewAnonymous(ctx context.Context, req *http.Request, cookieName string) (*pscookie.SessionContext, error) { if s == nil || s.db == nil { return nil, fmt.Errorf("prestashop session service is not initialized") } def, err := s.loadDefaults(ctx) if err != nil { return nil, err } guestID, err := s.insertGuest(ctx) if err != nil { return nil, err } connectionID, err := s.insertConnection(ctx, def, guestID, req) if err != nil { return nil, err } now := time.Now().UTC() values := map[string]string{ "checksum": anonymousChecksum(guestID, connectionID, def.LanguageID, def.CurrencyID, def.ShopID), "date_add": now.Format("2006-01-02 15:04:05"), "id_cart": "", "id_connections": strconv.FormatInt(connectionID, 10), "id_currency": strconv.FormatInt(def.CurrencyID, 10), "id_guest": strconv.FormatInt(guestID, 10), "id_lang": strconv.FormatInt(def.LanguageID, 10), "id_language": strconv.FormatInt(def.LanguageID, 10), "iso_code_country": def.CountryISO, } orderedKeys := []string{ "date_add", "id_lang", "id_cart", "id_language", "iso_code_country", "id_currency", "id_guest", "id_connections", "checksum", } if def.ShopID > 0 { values["id_shop"] = strconv.FormatInt(def.ShopID, 10) orderedKeys = append(orderedKeys[:6], append([]string{"id_shop"}, orderedKeys[6:]...)...) } return &pscookie.SessionContext{ CookieName: cookieName, LanguageID: int64Ptr(def.LanguageID), CurrencyID: int64Ptr(def.CurrencyID), ShopID: int64Ptr(def.ShopID), GuestID: int64Ptr(guestID), IsLoggedIn: false, Values: values, OrderedKeys: orderedKeys, ParseStatus: pscookie.ParseStatusAnonymous, }, nil } func (s *Service) loadDefaults(ctx context.Context) (*defaults, error) { def := &defaults{ LanguageID: 1, CurrencyID: 1, ShopID: 1, ShopGroupID: 1, CountryISO: "US", } configTable := s.prefix + "configuration" shopTable := s.prefix + "shop" countryTable := s.prefix + "country" var configs []struct { Name string Value string } configQuery := fmt.Sprintf("SELECT name, value FROM %s WHERE name IN ('PS_LANG_DEFAULT', 'PS_CURRENCY_DEFAULT', 'PS_COUNTRY_DEFAULT')", configTable) if err := s.db.WithContext(ctx).Raw(configQuery).Scan(&configs).Error; err != nil { return nil, err } countryID := int64(0) for _, cfg := range configs { switch cfg.Name { case "PS_LANG_DEFAULT": if parsed, err := strconv.ParseInt(cfg.Value, 10, 64); err == nil && parsed > 0 { def.LanguageID = parsed } case "PS_CURRENCY_DEFAULT": if parsed, err := strconv.ParseInt(cfg.Value, 10, 64); err == nil && parsed > 0 { def.CurrencyID = parsed } case "PS_COUNTRY_DEFAULT": if parsed, err := strconv.ParseInt(cfg.Value, 10, 64); err == nil && parsed > 0 { countryID = parsed } } } var shop struct { ID int64 `gorm:"column:id_shop"` GroupID int64 `gorm:"column:id_shop_group"` } shopQuery := fmt.Sprintf("SELECT id_shop, id_shop_group FROM %s ORDER BY id_shop LIMIT 1", shopTable) if err := s.db.WithContext(ctx).Raw(shopQuery).Scan(&shop).Error; err != nil { return nil, err } if shop.ID > 0 { def.ShopID = shop.ID } if shop.GroupID > 0 { def.ShopGroupID = shop.GroupID } if countryID > 0 { var country struct { ISOCode string `gorm:"column:iso_code"` } countryQuery := fmt.Sprintf("SELECT iso_code FROM %s WHERE id_country = ? LIMIT 1", countryTable) if err := s.db.WithContext(ctx).Raw(countryQuery, countryID).Scan(&country).Error; err != nil { return nil, err } if country.ISOCode != "" { def.CountryISO = country.ISOCode } } return def, nil } func (s *Service) insertGuest(ctx context.Context) (int64, error) { sqlDB, err := s.db.DB() if err != nil { return 0, fmt.Errorf("resolve sql db for guest insert: %w", err) } tableName := s.prefix + "guest" columns, values, err := s.guestInsert(ctx) if err != nil { return 0, err } query := insertQuery(tableName, columns) result, err := sqlDB.ExecContext(ctx, query, values...) if err != nil { return 0, fmt.Errorf("insert guest: %w", err) } id, err := result.LastInsertId() if err != nil { return 0, fmt.Errorf("guest last insert id: %w", err) } return id, nil } func (s *Service) insertConnection(ctx context.Context, def *defaults, guestID int64, req *http.Request) (int64, error) { sqlDB, err := s.db.DB() if err != nil { return 0, fmt.Errorf("resolve sql db for connection insert: %w", err) } tableName := s.prefix + "connections" columns, values, err := s.connectionInsert(ctx, def, guestID, req) if err != nil { return 0, err } query := insertQuery(tableName, columns) result, err := sqlDB.ExecContext(ctx, query, values...) if err != nil { return 0, fmt.Errorf("insert connection: %w", err) } id, err := result.LastInsertId() if err != nil { return 0, fmt.Errorf("connection last insert id: %w", err) } return id, nil } func (s *Service) guestInsert(ctx context.Context) ([]string, []any, error) { available, err := s.tableColumns(ctx, s.prefix+"guest") if err != nil { return nil, nil, fmt.Errorf("load guest columns: %w", err) } columns := make([]string, 0) values := make([]any, 0) addColumn := func(name string, value any) { if available[name] { columns = append(columns, name) values = append(values, value) } } addColumn("id_customer", 0) addColumn("id_operating_system", 0) addColumn("id_web_browser", 0) addColumn("javascript", 0) addColumn("screen_resolution_x", 0) addColumn("screen_resolution_y", 0) addColumn("screen_color", 0) addColumn("sun_java", 0) addColumn("adobe_flash", 0) addColumn("adobe_director", 0) addColumn("apple_quicktime", 0) addColumn("real_player", 0) addColumn("windows_media", 0) addColumn("accept_language", "") addColumn("mobile_theme", 0) return columns, values, nil } func (s *Service) connectionInsert(ctx context.Context, def *defaults, guestID int64, req *http.Request) ([]string, []any, error) { available, err := s.tableColumns(ctx, s.prefix+"connections") if err != nil { return nil, nil, fmt.Errorf("load connections columns: %w", err) } now := time.Now().UTC().Format("2006-01-02 15:04:05") columns := make([]string, 0) values := make([]any, 0) addColumn := func(name string, value any) { if available[name] { columns = append(columns, name) values = append(values, value) } } addColumn("id_guest", guestID) addColumn("id_shop", def.ShopID) addColumn("id_shop_group", def.ShopGroupID) addColumn("id_page", 0) addColumn("ip_address", ipAsUint32(req)) addColumn("date_add", now) addColumn("date_upd", now) addColumn("http_referer", referer(req)) addColumn("request_uri", requestURI(req)) return columns, values, nil } func (s *Service) tableColumns(ctx context.Context, tableName string) (map[string]bool, error) { type columnRow struct { ColumnName string `gorm:"column:COLUMN_NAME"` } var rows []columnRow query := ` SELECT COLUMN_NAME FROM information_schema.columns WHERE table_schema = DATABASE() AND table_name = ? ` if err := s.db.WithContext(ctx).Raw(query, tableName).Scan(&rows).Error; err != nil { return nil, err } columns := make(map[string]bool, len(rows)) for _, row := range rows { columns[row.ColumnName] = true } return columns, nil } func insertQuery(tableName string, columns []string) string { if len(columns) == 0 { return fmt.Sprintf("INSERT INTO %s () VALUES ()", tableName) } return fmt.Sprintf( "INSERT INTO %s (%s) VALUES (%s)", tableName, strings.Join(columns, ", "), placeholders(len(columns)), ) } func placeholders(n int) string { parts := make([]string, n) for i := range parts { parts[i] = "?" } return strings.Join(parts, ", ") } func referer(req *http.Request) string { if req == nil { return "" } return req.Referer() } func requestURI(req *http.Request) string { if req == nil || req.URL == nil { return "" } return req.URL.RequestURI() } func ipAsUint32(req *http.Request) uint32 { if req == nil { return 0 } raw := req.Header.Get("X-Forwarded-For") if raw == "" { raw = req.RemoteAddr } if strings.Contains(raw, ",") { raw = strings.TrimSpace(strings.Split(raw, ",")[0]) } host := raw if parsedHost, _, err := net.SplitHostPort(raw); err == nil { host = parsedHost } ip := net.ParseIP(host) if ip == nil { return 0 } ip = ip.To4() if ip == nil { return 0 } return uint32(ip[0])<<24 | uint32(ip[1])<<16 | uint32(ip[2])<<8 | uint32(ip[3]) } func anonymousChecksum(values ...int64) string { buf := make([]byte, 0, len(values)*8) for _, v := range values { buf = strconv.AppendInt(buf, v, 10) buf = append(buf, '|') } return strconv.FormatUint(uint64(crc32.ChecksumIEEE(buf)), 10) } func int64Ptr(value int64) *int64 { if value == 0 { return nil } v := value return &v }