store.go

package steam

import (
	"encoding/json"
	"errors"
	"fmt"
	"net/http"
	"net/url"
)

var ErrInvalidPhoneNumber = errors.New("invalid phone number specified")

type PhoneAPIResponse struct {
	Success   bool   `json:"success"`
	State     string `json:"state"`
	ErrorText string `json:"errorText"`
}

func (session *Session) PrepareForSteamStore() {
	community, _ := url.Parse("https://steamcommunity.com")
	store, _ := url.Parse("https://store.steampowered.com")

	session.client.Jar.SetCookies(store, session.client.Jar.Cookies(community))
}

func (session *Session) ValidatePhoneNumber(number string) error {
	resp, err := session.client.Get("https://store.steampowered.com/phone/validate?phoneNumber=" + url.QueryEscape(number))
	if resp != nil {
		defer resp.Body.Close()
	}

	if err != nil {
		return err
	}

	var response PhoneAPIResponse
	if err = json.NewDecoder(resp.Body).Decode(&response); err != nil {
		return err
	}

	if !response.Success {
		return ErrInvalidPhoneNumber
	}

	return nil
}

func (session *Session) AddPhoneNumber(number string) error {
	resp, err := session.client.Get("https://store.steampowered.com/phone/add_ajaxop?" + url.Values{
		"op":        {"get_phone_number"},
		"input":     {number},
		"sessionID": {session.sessionID},
		"confirmed": {"0"},
	}.Encode())
	if resp != nil {
		defer resp.Body.Close()
	}

	if err != nil {
		return err
	}

	var response PhoneAPIResponse
	if err = json.NewDecoder(resp.Body).Decode(&response); err != nil {
		return err
	}

	if response.State != "get_sms_code" {
		return errors.New(response.ErrorText)
	}

	return nil
}

func (session *Session) InitiateRemovePhoneNumber() error {
	resp, err := session.client.PostForm("https://store.steampowered.com/phone/remove_confirm_sms", url.Values{
		"sessionID": {session.sessionID},
		"bWasEdit":  {""},
	})
	if resp != nil {
		defer resp.Body.Close()
	}

	if err != nil {
		return err
	}

	if resp.StatusCode != http.StatusOK {
		return fmt.Errorf("http error: %d", resp.StatusCode)
	}

	return nil
}

func (session *Session) ConfirmRemovePhoneNumber(mobileCode string) error {
	resp, err := session.client.PostForm("https://store.steampowered.com/phone/remove_confirm_smscode_entry", url.Values{
		"sessionID": {session.sessionID},
		"bWasEdit":  {""},
		"smscode":   {mobileCode},
	})
	if resp != nil {
		defer resp.Body.Close()
	}

	// FIXME: Make a regexp for error.
	if err != nil {
		return err
	}

	if resp.StatusCode != http.StatusOK {
		return fmt.Errorf("http error: %d", resp.StatusCode)
	}

	return nil
}

func (session *Session) ReSendVerificationCode() error {
	resp, err := session.client.Get("https://store.steampowered.com/phone/add_ajaxop?" + url.Values{
		"op":        {"resend_sms"},
		"input":     {""},
		"sessionID": {session.sessionID},
		"confirmed": {"0"},
	}.Encode())
	if resp != nil {
		defer resp.Body.Close()
	}

	if err != nil {
		return err
	}

	var response PhoneAPIResponse
	if err = json.NewDecoder(resp.Body).Decode(&response); err != nil {
		return err
	}

	if !response.Success {
		return errors.New(response.ErrorText)
	}

	if response.State != "get_sms_code" {
		return fmt.Errorf("unknown state: %s", response.State)
	}

	return nil
}

func (session *Session) VerifyPhoneNumber(code string) error {
	resp, err := session.client.Get("https://store.steampowered.com/phone/add_ajaxop?" + url.Values{
		"op":        {"get_sms_code"},
		"input":     {code},
		"sessionID": {session.sessionID},
		"confirmed": {"0"},
	}.Encode())
	if resp != nil {
		defer resp.Body.Close()
	}

	if err != nil {
		return err
	}

	var response PhoneAPIResponse
	if err = json.NewDecoder(resp.Body).Decode(&response); err != nil {
		return err
	}

	if response.State != "done" {
		return errors.New(response.ErrorText)
	}

	return nil
}