//go:build !ios

package download

import (
	"bytes"
	"context"
	"crypto"
	"crypto/sha1"
	"crypto/tls"
	"encoding/base64"
	"encoding/binary"
	"encoding/json"
	"fmt"
	"io"
	"math/bits"
	"net/http"
	"net/http/cookiejar"
	"net/url"
	"os"
	"path/filepath"
	"reflect"
	"regexp"
	"sort"
	"strconv"
	"strings"

	"time"

	"github.com/99designs/keyring"
	"github.com/AlecAivazis/survey/v2"
	"github.com/AlecAivazis/survey/v2/terminal"
	"github.com/PuerkitoBio/goquery"
	"github.com/apex/log"
	"github.com/blacktop/ipsw/internal/srp"
	"github.com/blacktop/ipsw/internal/utils"
	"github.com/pkg/errors"
)

/*
	NOTES:
		- https://github.com/picklepete/pyicloud
		- https://github.com/michaljirman/fmidevice
		- https://github.com/majd/ipatool
		- https://github.com/fastlane/fastlane
*/

const (
	developerURL        = "https://developer.apple.com"
	downloadURLNew      = "https://download.developer.apple.com"
	downloadURL         = "https://developer.apple.com/download/"
	downloadOSesURL     = "https://developer.apple.com/download/os/"
	downloadMoreURL     = "https://developer.apple.com/download/all/"
	downloadAppsURL     = "https://developer.apple.com/download/applications/"
	downloadProfilesURL = "https://developer.apple.com/bug-reporting/profiles-and-logs/"

	downloadActionURL      = "https://developer.apple.com/devcenter/download.action"
	listDownloadsActionURL = "https://developer.apple.com/services-account/QH65B2/downloadws/listDownloads.action"
	adcDownloadURL         = "https://developerservices2.apple.com/services/download?path="

	authURL       = "https://idmsa.apple.com/appleauth/auth"
	loginURL      = authURL + "/signin"
	initURL       = loginURL + "/init"
	completeURL   = loginURL + "/complete?isRememberMeEnabled=false"
	trustURL      = "https://idmsa.apple.com/appleauth/auth/2sv/trust"
	itcServiceKey = "https://appstoreconnect.apple.com/olympus/v1/app/config?hostname=itunesconnect.apple.com"

	olympusSessionURL = "https://appstoreconnect.apple.com/olympus/v1/session"

	userAgent = "Configurator/2.18 (Macintosh; OS X 15.4.1; 24E263) AppleWebKit/0621.1.15.11.10"

	hashcashVersion        = 1
	hashcashHeader         = "X-APPLE-HC"
	hashcashCallengeHeader = "X-Apple-HC-Challenge"
	hashcashBitsHeader     = "X-Apple-HC-Bits"
)

const (
	ERROR_CODE_TOO_MANY_CODES_SENT      = -22981 // Too many verification codes have been sent.
	ERROR_CODE_BAD_CREDS                = -20101 // Your Apple ID or password was incorrect.
	ERROR_CODE_FAILED_TO_UPDATE_SESSION = -20528 // Error Description not available
	ERROR_CODE_BAD_VERIFICATION         = -21669 // Incorrect verification code.
	// TODO: flesh out the rest of the error codes
)

const (
	VaultName           = "ipsw-vault"
	AppName             = "io.blacktop.ipsw"
	KeychainServiceName = "ipsw-auth.service"
)

type DevConfig struct {
	// Login session
	SessionID string
	SCNT      string
	WidgetKey string
	// hashcash
	HashCash          string
	HashCashBits      string
	HashCashChallenge string
	// download config
	Proxy    string
	Insecure bool
	// download type config
	WatchList []string
	// behavior config
	SkipAll       bool
	ResumeAll     bool
	RestartAll    bool
	RemoveCommas  bool
	PreferSMS     bool
	PageSize      int
	Verbose       bool
	VaultPassword string
	ConfigDir     string
}

// DevPortal is the dev portal object
type DevPortal struct {
	Client *http.Client

	Vault keyring.Keyring

	config *DevConfig

	authService    authService
	authOptions    authOptions
	codeRequest    authOptions
	olympusSession olympusResponse
	// header values
	xAppleIDAccountCountry string
}

type credentials struct {
	Username      string `json:"username,omitempty"`
	Password      string `json:"password,omitempty"`
	DsPersonID    string `json:"directory_services_id,omitempty"`
	PasswordToken string `json:"password_token,omitempty"`
}

type session struct {
	SessionID string         `json:"session_id,omitempty"`
	SCNT      string         `json:"scnt,omitempty"`
	WidgetKey string         `json:"widget_key,omitempty"`
	HashCash  string         `json:"hashcash,omitempty"`
	Cookies   []*http.Cookie `json:"cookies,omitempty"`
}

type AppleAccountAuth struct {
	Credentials      credentials `json:"credentials"`
	DevPortalSession session     `json:"devport_session"`
	AppStoreSession  session     `json:"appstore_session"`
}

// DevDownload are all the downloads from https://developer.apple.com/download/
type DevDownload struct {
	Title string `json:"title,omitempty"`
	Build string `json:"build,omitempty"`
	URL   string `json:"url,omitempty"`
	Type  string `json:"type,omitempty"`
}

// Downloads listDownloads.action response
type Downloads struct {
	CreationTimestamp   time.Time `json:"creationTimestamp"`
	ResultCode          int       `json:"resultCode,omitempty"`
	UserLocale          string    `json:"userLocale,omitempty"`
	ProtocolVersion     string    `json:"protocolVersion,omitempty"`
	RequestURL          string    `json:"requestUrl,omitempty"`
	ResponseID          string    `json:"responseId,omitempty"`
	HTTPResponseHeaders struct {
		SetCookie string `json:"Set-Cookie,omitempty"`
	} `json:"httpResponseHeaders"`
	Downloads []MoreDownload
}

type authService struct {
	URL string `json:"authServiceUrl,omitempty"`
	Key string `json:"authServiceKey,omitempty"`
}

type auth struct {
	AccountName string   `json:"accountName,omitempty"`
	Password    string   `json:"password,omitempty"`
	RememberMe  bool     `json:"rememberMe"`
	TrustTokens []string `json:"trust_tokens,omitempty"`
	A           string   `json:"a,omitempty"`
	Protocols   []string `json:"protocols,omitempty"`
	M1          string   `json:"m1,omitempty"`
	C           string   `json:"c,omitempty"`
	M2          string   `json:"m2,omitempty"`
}

type trustedPhoneNumber struct {
	ID                 int    `json:"id,omitempty"`
	ObfuscatedNumber   string `json:"obfuscatedNumber,omitempty"`
	PushMode           string `json:"pushMode,omitempty"`
	NumberWithDialCode string `json:"numberWithDialCode,omitempty"`
}

type securityCode struct {
	Code                  string `json:"code,omitempty"`
	Length                int    `json:"length,omitempty"`
	TooManyCodesSent      bool   `json:"tooManyCodesSent,omitempty"`
	TooManyCodesValidated bool   `json:"tooManyCodesValidated,omitempty"`
	SecurityCodeLocked    bool   `json:"securityCodeLocked,omitempty"`
	SecurityCodeCooldown  bool   `json:"securityCodeCooldown,omitempty"`
}

type authOptions struct {
	TrustedDeviceCount              int                  `json:"trustedDeviceCount,omitempty"`
	OtherTrustedDeviceClass         string               `json:"otherTrustedDeviceClass,omitempty"`
	TrustedPhoneNumbers             []trustedPhoneNumber `json:"trustedPhoneNumbers,omitempty"`
	PhoneNumber                     trustedPhoneNumber   `json:"phoneNumber"`
	TrustedPhoneNumber              trustedPhoneNumber   `json:"trustedPhoneNumber"`
	SecurityCode                    securityCode         `json:"securityCode"`
	Mode                            string               `json:"mode,omitempty"`
	Type                            string               `json:"type,omitempty"`
	AuthenticationType              string               `json:"authenticationType,omitempty"`
	RecoveryURL                     string               `json:"recoveryUrl,omitempty"`
	CantUsePhoneNumberURL           string               `json:"cantUsePhoneNumberUrl,omitempty"`
	RecoveryWebURL                  string               `json:"recoveryWebUrl,omitempty"`
	RepairPhoneNumberURL            string               `json:"repairPhoneNumberUrl,omitempty"`
	RepairPhoneNumberWebURL         string               `json:"repairPhoneNumberWebUrl,omitempty"`
	AboutTwoFactorAuthenticationURL string               `json:"aboutTwoFactorAuthenticationUrl,omitempty"`
	AutoVerified                    bool                 `json:"autoVerified,omitempty"`
	ShowAutoVerificationUI          bool                 `json:"showAutoVerificationUI,omitempty"`
	ManagedAccount                  bool                 `json:"managedAccount,omitempty"`
	Hsa2Account                     bool                 `json:"hsa2Account,omitempty"`
	RestrictedAccount               bool                 `json:"restrictedAccount,omitempty"`
	SupportsRecovery                bool                 `json:"supportsRecovery,omitempty"`
	SupportsCustodianRecovery       bool                 `json:"supportsCustodianRecovery,omitempty"`
	ServiceErrors                   []serviceError       `json:"serviceErrors,omitempty"`
	NoTrustedDevices                bool                 `json:"noTrustedDevices,omitempty"`
}

type scode struct {
	Code string `json:"code,omitempty"`
}
type phoneNumber struct {
	ID int `json:"id"`
}
type phone struct {
	SecurityCode scode       `json:"securityCode"`
	Number       phoneNumber `json:"phoneNumber"`
	Mode         string      `json:"mode,omitempty"`
}
type trustedDevice struct {
	SecurityCode scode `json:"securityCode"`
}

type signInResponse struct {
	AuthType string `json:"authType,omitempty"`
}

type serviceError struct {
	Code              string `json:"code,omitempty"`
	Title             string `json:"title,omitempty"`
	Message           string `json:"message,omitempty"`
	SuppressDismissal bool   `json:"suppressDismissal,omitempty"`
}

type serviceErrorsResponse struct {
	Errors   []serviceError `json:"service_errors,omitempty"`
	HasError bool           `json:"hasError,omitempty"`
}

type olympusProvider struct {
	ID           int      `json:"providerId,omitempty"`
	PublicID     string   `json:"publicProviderId,omitempty"`
	Name         string   `json:"name,omitempty"`
	ContentTypes []string `json:"contentTypes,omitempty"`
	SubType      string   `json:"subType,omitempty"`
	PLA          []struct {
		ID                       string   `json:"id,omitempty"`
		Version                  string   `json:"version,omitempty"`
		Types                    []string `json:"types,omitempty"`
		ContractCountryOfOrigins []string `json:"contractCountryOfOrigins,omitempty"`
	} `json:"pla,omitempty"`
}

type olympusResponse struct {
	User struct {
		FullNAme  string `json:"fullName,omitempty"`
		FirstName string `json:"firstName,omitempty"`
		LastName  string `json:"lastName,omitempty"`
		Email     string `json:"emailAddress,omitempty"`
		PrsID     string `json:"prsId,omitempty"`
	} `json:"user"`
	Provider           olympusProvider   `json:"provider"`
	Theme              string            `json:"theme,omitempty"`
	AvailableProviders []olympusProvider `json:"availableProviders,omitempty"`
	BackingType        string            `json:"backingType,omitempty"`
	BackingTypes       []string          `json:"backingTypes,omitempty"`
	Roles              []string          `json:"roles,omitempty"`
	UnverifiedRoles    []string          `json:"unverifiedRoles,omitempty"`
	FeatureFlags       []string          `json:"featureFlags,omitempty"`
	AgreeToTerms       bool              `json:"agreeToTerms,omitempty"`
	TermsSignatures    []string          `json:"termsSignatures,omitempty"`
	PublicUserID       string            `json:"publicUserId,omitempty"`
	OfacState          any               `json:"ofacState,omitempty"`
	CreationDate       int               `json:"creationDate,omitempty"`
}

type category struct {
	ID        int    `json:"id,omitempty"`
	Name      string `json:"name,omitempty"`
	SortOrder int    `json:"sortOrder,omitempty"`
}

type fformat struct {
	Extension   string `json:"extension,omitempty"`
	Description string `json:"description,omitempty"`
}

type dfile struct {
	Filename     string  `json:"filename,omitempty"`
	DisplayName  string  `json:"displayName,omitempty"`
	RemotePath   string  `json:"remotePath,omitempty"`
	FileSize     int     `json:"fileSize,omitempty"`
	SortOrder    int     `json:"sortOrder,omitempty"`
	DateCreated  string  `json:"dateCreated,omitempty"`
	DateModified string  `json:"dateModified,omitempty"`
	FileFormat   fformat `json:"fileFormat"`
}

func (d dfile) URL() string {
	// https://download.developer.apple.com/macOS/Kernel_Debug_Kit_13.5_build_22G74/Kernel_Debug_Kit_13.5_build_22G74.dmg
	if u, err := url.JoinPath(downloadURLNew, d.RemotePath); err == nil {
		return u
	}
	return fmt.Sprintf("%s/%s", downloadURLNew, d.RemotePath)
}

type MoreDownload struct {
	Name          string     `json:"name,omitempty"`
	Description   string     `json:"description,omitempty"`
	IsReleased    int        `json:"isReleased,omitempty"`
	DatePublished string     `json:"datePublished,omitempty"`
	DateCreated   string     `json:"dateCreated,omitempty"`
	DateModified  string     `json:"dateModified,omitempty"`
	Categories    []category `json:"categories,omitempty"`
	Files         []dfile    `json:"files,omitempty"`
}

// NewDevPortal returns a new DevPortal instance
func NewDevPortal(config *DevConfig) *DevPortal {
	jar, _ := cookiejar.New(nil)

	dp := DevPortal{
		Client: &http.Client{
			Jar: jar,
			Transport: &http.Transport{
				Proxy:           GetProxy(config.Proxy),
				TLSClientConfig: &tls.Config{InsecureSkipVerify: config.Insecure},
			},
		},
		config: config,
	}

	return &dp
}

// Init DevPortal sets up the DevPortal vault
func (dp *DevPortal) Init() (err error) {
	// create credential vault (if it doesn't exist)
	dp.Vault, err = keyring.Open(keyring.Config{
		ServiceName:                    KeychainServiceName,
		KeychainSynchronizable:         false,
		KeychainAccessibleWhenUnlocked: true,
		KeychainTrustApplication:       true,
		FileDir:                        dp.config.ConfigDir,
		FilePasswordFunc: func(string) (string, error) {
			if len(dp.config.VaultPassword) == 0 {
				msg := "Enter a password to decrypt your credentials vault: " + filepath.Join(dp.config.ConfigDir, VaultName)
				if _, err := os.Stat(filepath.Join(dp.config.ConfigDir, VaultName)); errors.Is(err, os.ErrNotExist) {
					msg = "Enter a password to encrypt your credentials to vault: " + filepath.Join(dp.config.ConfigDir, VaultName)
				}
				prompt := &survey.Password{
					Message: msg,
				}
				if err := survey.AskOne(prompt, &dp.config.VaultPassword); err != nil {
					if err == terminal.InterruptErr {
						log.Warn("Exiting...")
						os.Exit(0)
					}
					return "", err
				}
			}
			return dp.config.VaultPassword, nil
		},
	})
	if err != nil {
		return fmt.Errorf("failed to open vault: %s", err)
	}

	return nil
}

func (dp *DevPortal) GetSessionID() string {
	return dp.config.SessionID
}
func (dp *DevPortal) GetSCNT() string {
	return dp.config.SCNT
}
func (dp *DevPortal) GetWidgetKey() string {
	return dp.config.WidgetKey
}
func (dp *DevPortal) GetHashcash() string {
	return dp.config.HashCash
}

// Login to Apple
func (dp *DevPortal) Login(username, password string) error {
	if len(username) == 0 || len(password) == 0 {
		creds, err := dp.Vault.Get(VaultName)
		if err != nil { // failed to get credentials from vault (prompt user for credentials)
			log.Errorf("failed to get credentials from vault: %v", err)
			// get username
			if len(username) == 0 {
				prompt := &survey.Input{
					Message: "Please type your username:",
				}
				if err := survey.AskOne(prompt, &username); err != nil {
					if err == terminal.InterruptErr {
						log.Warn("Exiting...")
						os.Exit(0)
					}
					return err
				}
			}
			// get password
			if len(password) == 0 {
				prompt := &survey.Password{
					Message: "Please type your password:",
				}
				if err := survey.AskOne(prompt, &password); err != nil {
					if err == terminal.InterruptErr {
						log.Warn("Exiting...")
						os.Exit(0)
					}
					return err
				}
			}
			// save credentials to vault
			dat, err := json.Marshal(&AppleAccountAuth{
				Credentials: credentials{
					Username: username,
					Password: password,
				},
			})
			if err != nil {
				return fmt.Errorf("failed to marshal keychain credentials: %v", err)
			}
			if err := dp.Vault.Set(keyring.Item{
				Key:         VaultName,
				Data:        dat,
				Label:       AppName,
				Description: "application password",
			}); err != nil {
				return fmt.Errorf("failed to save credentials to vault: %v", err)
			}
		} else { // credentials found in vault
			var auth AppleAccountAuth
			if err := json.Unmarshal(creds.Data, &auth); err != nil {
				return fmt.Errorf("failed to unmarshal keychain credentials (this can happen if the creds in the vault are somehow incomplete/corrupt; manually removing them and trying again might fix it): %v", err)
			}
			username = auth.Credentials.Username
			password = auth.Credentials.Password
			auth = AppleAccountAuth{}
		}
	}

	if err := dp.loadSession(); err != nil { // load previous session (if error, login)
		if err := dp.getITCServiceKey(); err != nil {
			return err
		}

		return dp.signIn(username, password)
	}

	return nil
}

func (dp *DevPortal) generateHashCash() (string, error) {
	var hashcash string

	hcbits, err := strconv.Atoi(dp.config.HashCashBits)
	if err != nil {
		return "", fmt.Errorf("failed to convert hashcash bits %s to int: %v", dp.config.HashCashBits, err)
	}

	counter := 0

	for {
		hash := sha1.New()
		hashcash = fmt.Sprintf("%s:%s:%s:%s::%s",
			strconv.Itoa(hashcashVersion),             // ver
			dp.config.HashCashBits,                    // bits
			time.Now().UTC().Format("20060102150405"), // date
			dp.config.HashCashChallenge,               // res
			strconv.Itoa(counter),                     // counter
		)
		hash.Write([]byte(hashcash))
		lz := bits.LeadingZeros32(binary.BigEndian.Uint32(hash.Sum(nil)[:4]))
		if lz >= hcbits {
			break
		}
		counter++
	}

	return hashcash, nil
}

func (dp *DevPortal) updateRequestHeaders(req *http.Request) {
	req.Header.Set("X-Requested-With", "XMLHttpRequest")
	req.Header.Set("Content-Type", "application/json")
	req.Header.Set("Accept", "application/json")

	req.Header.Set("X-Apple-Id-Session-Id", dp.config.SessionID)
	req.Header.Set("X-Apple-Widget-Key", dp.config.WidgetKey)
	req.Header.Set("Scnt", dp.config.SCNT)
	req.Header.Set(hashcashHeader, dp.config.HashCash)

	req.Header.Add("User-Agent", userAgent)
}

func (dp *DevPortal) getITCServiceKey() error {

	response, err := dp.Client.Get(itcServiceKey)
	if err != nil {
		return err
	}
	defer response.Body.Close()

	body, err := io.ReadAll(response.Body)
	if err != nil {
		return err
	}

	if err := json.Unmarshal(body, &dp.authService); err != nil {
		return fmt.Errorf("failed to deserialize response body JSON: %v", err)
	}

	log.Debugf("GET iTC Service Key: (%d):\n%s\n", response.StatusCode, string(body))

	if response.StatusCode != 200 {
		return fmt.Errorf("failed to get iTC Service Key: response received %s", response.Status)
	}

	dp.config.WidgetKey = dp.authService.Key

	return nil
}

func (dp *DevPortal) getHashcachHeaders() error {
	response, err := dp.Client.Get(loginURL)
	if err != nil {
		return err
	}

	if response.StatusCode != 200 {
		return fmt.Errorf("failed to get iTC Service Key: response received %s", response.Status)
	}

	// 🆕 hashcash headers
	dp.config.HashCashBits = response.Header.Get(hashcashBitsHeader)
	dp.config.HashCashChallenge = response.Header.Get(hashcashCallengeHeader)
	if dp.config.HashCashBits != "" || dp.config.HashCashChallenge != "" {
		dp.config.HashCash, err = dp.generateHashCash()
		if err != nil {
			return fmt.Errorf("failed to generate hashcash: %v", err)
		}
	}

	return nil
}

type srpInitResponse struct {
	Iteration int    `json:"iteration,omitempty"`
	Salt      string `json:"salt,omitempty"`
	Protocol  string `json:"protocol,omitempty"`
	B         string `json:"b,omitempty"`
	C         string `json:"c,omitempty"`
}

type srpCompleteResponse struct {
	AccountName           string         `json:"accountName,omitempty"`
	RememberMe            bool           `json:"rememberMe,omitempty"`
	Skip2Fa               bool           `json:"skip2FA,omitempty"`
	Pause2Fa              bool           `json:"pause2FA,omitempty"`
	DomainAutoFill        bool           `json:"domainAutoFill,omitempty"`
	M1                    string         `json:"m1,omitempty"`
	C                     string         `json:"c,omitempty"`
	M2                    string         `json:"m2,omitempty"`
	ParseRememberMeTokens bool           `json:"parseRememberMeTokens,omitempty"`
	ServiceErrors         []serviceError `json:"serviceErrors,omitempty"`
}

func (dp *DevPortal) generateSRP(username, password string) (*http.Response, error) {
	s, err := srp.NewWithHash(crypto.SHA256, 2048)
	if err != nil {
		return nil, err
	}

	buf := new(bytes.Buffer)

	if err := json.NewEncoder(buf).Encode(&auth{
		AccountName: username,
		A:           base64.StdEncoding.EncodeToString(s.A.Bytes()),
		Protocols:   []string{"s2k", "s2k_fo"},
	}); err != nil {
		return nil, fmt.Errorf("failed to encode auth request: %v", err)
	}

	req, err := http.NewRequest("POST", initURL, buf)
	if err != nil {
		return nil, fmt.Errorf("failed to create http POST request: %v", err)
	}

	req.Header.Set("Content-Type", "application/json")
	req.Header.Set("X-Requested-With", "XMLHttpRequest")
	req.Header.Set("X-Apple-Widget-Key", dp.config.WidgetKey)
	req.Header.Set(hashcashHeader, dp.config.HashCash)
	req.Header.Add("User-Agent", userAgent)
	req.Header.Set("Accept", "application/json, text/javascript")

	initResponse, err := dp.Client.Do(req)
	if err != nil {
		return nil, err
	}
	defer initResponse.Body.Close()

	body, err := io.ReadAll(initResponse.Body)
	if err != nil {
		return nil, err
	}

	log.Debugf("SRP INIT: (%d):\n%s\n", initResponse.StatusCode, string(body))

	var srpInit srpInitResponse
	if err := json.Unmarshal(body, &srpInit); err != nil {
		return nil, fmt.Errorf("failed to deserialize response body JSON: %v", err)
	}

	saltBytes, err := base64.StdEncoding.DecodeString(srpInit.Salt)
	if err != nil {
		return nil, err
	}

	bBytes, err := base64.StdEncoding.DecodeString(srpInit.B)
	if err != nil {
		return nil, err
	}

	cli, err := s.NewClient([]byte(username), []byte(password), saltBytes, srpInit.Iteration)
	if err != nil {
		return nil, err
	}

	m1, m2, err := cli.Generate(saltBytes, bBytes)
	if err != nil {
		return nil, err
	}

	buf.Reset()

	json.NewEncoder(buf).Encode(&auth{
		AccountName: username,
		C:           srpInit.C,
		M1:          base64.StdEncoding.EncodeToString(m1),
		M2:          base64.StdEncoding.EncodeToString(m2),
		RememberMe:  false,
	})

	req, err = http.NewRequest("POST", completeURL, buf)
	if err != nil {
		return nil, fmt.Errorf("failed to create http POST request: %v", err)
	}

	req.Header.Set("Content-Type", "application/json")
	req.Header.Set("X-Requested-With", "XMLHttpRequest")
	req.Header.Set("X-Apple-Widget-Key", dp.config.WidgetKey)
	req.Header.Set(hashcashHeader, dp.config.HashCash)
	req.Header.Add("User-Agent", userAgent)
	req.Header.Set("Accept", "application/json, text/javascript")

	completeResponse, err := dp.Client.Do(req)
	if err != nil {
		return nil, err
	}
	defer completeResponse.Body.Close()

	body, err = io.ReadAll(completeResponse.Body)
	if err != nil {
		return nil, err
	}

	log.Debugf("SRP COMPLETE: (%d):\n%s\n", completeResponse.StatusCode, string(body))

	var srpComp srpCompleteResponse
	if err := json.Unmarshal(body, &srpComp); err != nil {
		return nil, fmt.Errorf("failed to deserialize response body JSON: %v", err)
	}

	if len(srpComp.ServiceErrors) > 0 {
		return nil, fmt.Errorf("failed to complete SRP: %s", srpComp.ServiceErrors[0].Message)
	}

	return completeResponse, nil
}

func (dp *DevPortal) signIn(username, password string) error {

	if err := dp.getHashcachHeaders(); err != nil {
		return fmt.Errorf("failed to get hashcash headers: %v", err)
	}

	buf := new(bytes.Buffer)

	json.NewEncoder(buf).Encode(&auth{
		AccountName: username,
		Password:    password,
		RememberMe:  true,
	})

	req, err := http.NewRequest("POST", loginURL, buf)
	if err != nil {
		return fmt.Errorf("failed to create http POST request: %v", err)
	}

	req.Header.Set("Content-Type", "application/json")
	req.Header.Set("X-Requested-With", "XMLHttpRequest")
	req.Header.Set("X-Apple-Widget-Key", dp.config.WidgetKey)
	req.Header.Set(hashcashHeader, dp.config.HashCash)
	req.Header.Add("User-Agent", userAgent)
	req.Header.Set("Accept", "application/json, text/javascript")

	response, err := dp.Client.Do(req)
	if err != nil {
		return err
	}
	defer response.Body.Close()

	body, err := io.ReadAll(response.Body)
	if err != nil {
		return err
	}

	log.Debugf("POST Login: (%d):\n%s\n", response.StatusCode, string(body))

	if response.StatusCode == 503 { // try NEW SRP login
		response, err = dp.generateSRP(username, password)
		if err != nil {
			return err
		}
	}

	if response.StatusCode == 409 {
		dp.xAppleIDAccountCountry = response.Header.Get("X-Apple-Id-Account-Country")
		dp.config.SessionID = response.Header.Get("X-Apple-Id-Session-Id")
		dp.config.SCNT = response.Header.Get("Scnt")

		if err := dp.getAuthOptions(); err != nil {
			return err
		}

		phoneID := 1
		codeType := "phone"

		// SMS was sent automatically
		if dp.authOptions.NoTrustedDevices && len(dp.authOptions.TrustedPhoneNumbers) == 1 {
			codeType = "phone"
			// User needs to choose a phone to send to
		} else if dp.authOptions.NoTrustedDevices && len(dp.authOptions.TrustedPhoneNumbers) > 1 {
			codeType = "phone"
			phoneNumber := 0
			var choices []string
			for _, num := range dp.authOptions.TrustedPhoneNumbers {
				choices = append(choices, num.NumberWithDialCode)
			}
			prompt := &survey.Select{
				Message: "Choose a phone number to send the SMS code to:",
				Options: choices,
			}
			if err := survey.AskOne(prompt, &phoneNumber); err != nil {
				if err == terminal.InterruptErr {
					log.Warn("Exiting...")
					os.Exit(0)
				}
				return err
			}
			phoneID = dp.authOptions.TrustedPhoneNumbers[phoneNumber].ID
			if err := dp.requestCode(phoneID); err != nil {
				return err
			}

		} else { // Code is shown on trusted devices
			codeType = "trusteddevice"
			if dp.config.PreferSMS {
				codeType = "phone"
				if err := dp.requestCode(1); err != nil {
					if dp.codeRequest.SecurityCode.TooManyCodesSent {
						codeType = "trusteddevice"
						log.Warn("you must use the trusted device code (SMS codes have been disabled on your account)")
					} else {
						return err
					}
				}
			}
		}

		code := ""
		// USED FOR DEBUGGING
		// cwd, err := os.Getwd()
		// if err != nil {
		// 	log.Error(err.Error())
		// }
		// cpath := filepath.Join(cwd, "..", "..", "test-caches", "CODE")
		// fmt.Printf("Enter code in file (%s): ", cpath)
		// for {
		// 	codeSTR, err := os.ReadFile(cpath)
		// 	if err != nil {
		// 		return err
		// 	}
		// 	if len(codeSTR) > 0 {
		// 		code = string(codeSTR)
		// 		// remove code for next time
		// 		defer func() {
		// 			if err := os.WriteFile(cpath, []byte(""), 0660); err != nil {
		// 				log.Error(err.Error())
		// 			}
		// 		}()
		// 		break
		// 	}
		// }
		if len(code) == 0 {
			prompt := &survey.Password{
				Message: "Please type your verification code:",
			}
			if err := survey.AskOne(prompt, &code); err != nil {
				if err == terminal.InterruptErr {
					log.Warn("Exiting...")
					os.Exit(0)
				}
				return err
			}
		}

		if err := dp.verifyCode(codeType, code, phoneID); err != nil {
			return err
		}

		if err := dp.trustSession(); err != nil {
			return err
		}

	} else if response.StatusCode != 200 {
		return fmt.Errorf("failed to sign in; expected status code 409 (for two factor auth): response received %s", response.Status)
	}

	if err := dp.storeSession(); err != nil {
		return err
	}

	return nil
}

func (dp *DevPortal) getAuthOptions() error {

	req, err := http.NewRequest("GET", "https://idmsa.apple.com/appleauth/auth", nil)
	if err != nil {
		return fmt.Errorf("failed to create http GET request: %v", err)
	}
	dp.updateRequestHeaders(req)

	response, err := dp.Client.Do(req)
	if err != nil {
		return err
	}
	defer response.Body.Close()

	body, err := io.ReadAll(response.Body)
	if err != nil {
		return err
	}

	log.Debugf("GET getAuthOptions (%d):\n%s\n", response.StatusCode, string(body))

	if err := json.Unmarshal(body, &dp.authOptions); err != nil {
		return fmt.Errorf("failed to deserialize response body JSON: %v", err)
	}

	if 200 > response.StatusCode || 300 <= response.StatusCode {
		return fmt.Errorf("failed to get auth options: response received %s", response.Status)
	}

	return nil
}

func (dp *DevPortal) requestCode(phoneID int) error {
	buf := new(bytes.Buffer)

	json.NewEncoder(buf).Encode(&phone{
		Number: phoneNumber{
			ID: phoneID,
		},
		Mode: "sms",
	})

	req, err := http.NewRequest("PUT", "https://idmsa.apple.com/appleauth/auth/verify/phone", buf)
	if err != nil {
		return fmt.Errorf("failed to create http PUT request: %v", err)
	}
	dp.updateRequestHeaders(req)

	response, err := dp.Client.Do(req)
	if err != nil {
		return err
	}
	defer response.Body.Close()

	body, err := io.ReadAll(response.Body)
	if err != nil {
		return err
	}

	log.Debugf("PUT requestCode (%d):\n%s\n", response.StatusCode, string(body))

	if err := json.Unmarshal(body, &dp.codeRequest); err != nil {
		return fmt.Errorf("failed to deserialize response body JSON: %v", err)
	}

	if 200 > response.StatusCode || 300 <= response.StatusCode {
		var errStr string
		if dp.codeRequest.ServiceErrors != nil {
			for _, svcErr := range dp.codeRequest.ServiceErrors {
				errStr += fmt.Sprintf(": %s", svcErr.Message)
			}
			return fmt.Errorf("failed to verify code: response received %s%s", response.Status, errStr)
		}

		if response.StatusCode == 423 { // code rate limiting
			log.Error(errStr)
			return nil
		}

		return fmt.Errorf("failed to verify code: response received %s%s", response.Status, errStr)
	}

	return nil
}

func (dp *DevPortal) verifyCode(codeType, code string, phoneID int) error {
	buf := new(bytes.Buffer)

	if codeType == "phone" {
		json.NewEncoder(buf).Encode(&phone{
			Number: phoneNumber{
				ID: phoneID,
			},
			Mode: "sms",
			SecurityCode: scode{
				Code: code,
			},
		})
	} else { // trusteddevice
		json.NewEncoder(buf).Encode(&trustedDevice{
			SecurityCode: scode{
				Code: code,
			},
		})
	}

	req, err := http.NewRequest("POST", fmt.Sprintf("https://idmsa.apple.com/appleauth/auth/verify/%s/securitycode", codeType), buf)
	if err != nil {
		return fmt.Errorf("failed to create http POST request: %v", err)
	}
	dp.updateRequestHeaders(req)

	response, err := dp.Client.Do(req)
	if err != nil {
		return err
	}
	defer response.Body.Close()

	body, err := io.ReadAll(response.Body)
	if err != nil {
		return err
	}

	log.Debugf("POST verifyCode (%d):\n%s\n", response.StatusCode, string(body))

	if 200 > response.StatusCode || 300 <= response.StatusCode {
		if len(body) > 0 {
			var svcErr serviceErrorsResponse
			if err := json.Unmarshal(body, &svcErr); err != nil {
				return fmt.Errorf("failed to deserialize response body JSON: %v", err)
			}
			if svcErr.HasError {
				var errStr string
				for _, svcErr := range svcErr.Errors {
					errStr += fmt.Sprintf(": %s", svcErr.Message)
				}
				return fmt.Errorf("failed to verify code: response received %s%s", response.Status, errStr)
			}
		}

		return fmt.Errorf("failed to verify code: response received %s", response.Status)
	}

	return nil
}

// trustSession tells Apple to trust computer for 2FA
func (dp *DevPortal) trustSession() error {

	req, err := http.NewRequest("GET", trustURL, nil)
	if err != nil {
		return fmt.Errorf("failed to create http GET request: %v", err)
	}
	dp.updateRequestHeaders(req)

	response, err := dp.Client.Do(req)
	if err != nil {
		return err
	}
	defer response.Body.Close()

	body, err := io.ReadAll(response.Body)
	if err != nil {
		return err
	}

	log.Debugf("GET trustSession: (%d):\n%s\n", response.StatusCode, string(body))

	if 200 > response.StatusCode || 300 <= response.StatusCode {
		if len(body) > 0 {
			var svcErr serviceErrorsResponse
			if err := json.Unmarshal(body, &svcErr); err != nil {
				return fmt.Errorf("failed to deserialize response body JSON: %v", err)
			}
			if svcErr.HasError {
				var errStr string
				for _, svcErr := range svcErr.Errors {
					errStr += fmt.Sprintf(": %s", svcErr.Message)
				}
				return fmt.Errorf("failed to update to trusted session: response received %s%s", response.Status, errStr)
			}
		}
		return fmt.Errorf("failed to update to trusted session: response received %s", response.Status)
	}

	return nil
}

func (dp *DevPortal) refreshSession() error {
	// check if olympus session is expired (prevents login rate limiting)
	if err := dp.getOlympusSession(); err != nil {
		// if olympus session is expired, we need to login again
		return dp.Login("", "")
	}
	return nil
}

func (dp *DevPortal) getOlympusSession() error {

	req, err := http.NewRequest("GET", olympusSessionURL, nil)
	if err != nil {
		return fmt.Errorf("failed to create http GET request: %v", err)
	}
	dp.updateRequestHeaders(req)

	response, err := dp.Client.Do(req)
	if err != nil {
		return err
	}
	defer response.Body.Close()

	body, err := io.ReadAll(response.Body)
	if err != nil {
		return err
	}

	log.Debugf("GET getOlympusSession (%d):\n%s\n", response.StatusCode, string(body))

	if 200 > response.StatusCode || 300 <= response.StatusCode {
		return fmt.Errorf("failed to get auth options: response received %s", response.Status)
	}

	if err := json.Unmarshal(body, &dp.olympusSession); err != nil {
		var wat any
		json.Unmarshal(body, &wat)
		log.Errorf("%#v", wat)
		return fmt.Errorf("failed to deserialize response body JSON: %v", err)
	}

	return nil
}

func (dp *DevPortal) storeSession() error {
	// get dev auth from vault
	sess, err := dp.Vault.Get(VaultName)
	if err != nil {
		return fmt.Errorf("failed to get dev auth from vault: %v", err)
	}

	var auth AppleAccountAuth
	if err := json.Unmarshal(sess.Data, &auth); err != nil {
		return fmt.Errorf("failed to unmarshal dev auth: %v", err)
	}

	auth.DevPortalSession = session{
		SessionID: dp.GetSessionID(),
		SCNT:      dp.GetSCNT(),
		WidgetKey: dp.GetWidgetKey(),
		HashCash:  dp.GetHashcash(),
		Cookies:   dp.Client.Jar.Cookies(&url.URL{Scheme: "https", Host: "idmsa.apple.com"}),
	}

	// save dev auth to vault
	data, err := json.Marshal(&auth)
	if err != nil {
		return fmt.Errorf("failed to marshal session: %v", err)
	}

	// clear dev auth mem
	auth = AppleAccountAuth{}

	if err := dp.Vault.Set(keyring.Item{
		Key:         VaultName,
		Data:        data,
		Label:       AppName,
		Description: "application password",
	}); err != nil {
		return fmt.Errorf("failed to save session to vault: %v", err)
	}

	return nil
}

func (dp *DevPortal) loadSession() error {
	// get dev auth from vault
	sess, err := dp.Vault.Get(VaultName)
	if err != nil {
		return fmt.Errorf("failed to get dev auth from vault: %v", err)
	}

	var auth AppleAccountAuth
	if err := json.Unmarshal(sess.Data, &auth); err != nil {
		return fmt.Errorf("failed to unmarshal dev auth: %v", err)
	}

	dp.config.SessionID = auth.DevPortalSession.SessionID
	dp.config.SCNT = auth.DevPortalSession.SCNT
	dp.config.WidgetKey = auth.DevPortalSession.WidgetKey
	dp.config.HashCash = auth.DevPortalSession.HashCash
	dp.Client.Jar.SetCookies(&url.URL{Scheme: "https", Host: "idmsa.apple.com"}, auth.DevPortalSession.Cookies)

	// clear dev auth mem
	auth = AppleAccountAuth{}

	if err := dp.getOlympusSession(); err != nil {
		return err
	}

	return nil
}

// Watch watches for NEW downloads
func (dp *DevPortal) Watch(ctx context.Context, downloadType, folder string, duration time.Duration) error {

	var prevDownloads []MoreDownload
	var prevIPSWs map[string][]DevDownload
	var prevProfiles map[string]string

	for {
		// scrape dev portal
		switch downloadType {
		case "more":
			dloads, err := dp.getDownloads()
			if err != nil {
				return fmt.Errorf("failed to get the '%s' downloads: %v", downloadType, err)
			}

			// check for NEW downloads
			if reflect.DeepEqual(prevDownloads, dloads.Downloads) { // "8b42055e-8d9d-4bbb-800b-6a44c45c7b48"
				time.Sleep(duration)

				if err := dp.refreshSession(); err != nil {
					return err
				}

				continue

			} else {
				prevDownloads = dloads.Downloads
			}

			for _, dl := range dloads.Downloads {
				for _, watchPattern := range dp.config.WatchList {
					re, err := regexp.Compile(watchPattern)
					if err != nil {
						return fmt.Errorf("failed to compile regex watch pattern '%s': %v", watchPattern, err)
					}
					if re.MatchString(dl.Name) {
						for _, f := range dl.Files {
							dp.Download(f.URL(), folder)
						}
					}
				}
			}
		case "os":
			ipsws, err := dp.getDevDownloads()
			if err != nil {
				return fmt.Errorf("failed to get developer downloads: %v", err)
			}

			// check for NEW downloads
			if reflect.DeepEqual(prevIPSWs, ipsws) {
				time.Sleep(5 * time.Minute)

				if err := dp.refreshSession(); err != nil {
					return err
				}

				continue

			} else {
				prevIPSWs = ipsws
			}

			for version := range ipsws {
				for _, watchPattern := range dp.config.WatchList {
					re, err := regexp.Compile(watchPattern)
					if err != nil {
						return fmt.Errorf("failed to compile regex watch pattern '%s': %v", watchPattern, err)
					}
					if re.MatchString(version) {
						for _, ipsw := range ipsws[version] {
							if err := dp.Download(ipsw.URL, folder); err != nil {
								log.Errorf("failed to download %s: %v", ipsw.URL, err)
							}
						}
					}
				}
			}
		case "profile":
			profiles, err := dp.getDevLoggingProfiles()
			if err != nil {
				return fmt.Errorf("failed to get developer downloads: %v", err)
			}

			// check for NEW downloads
			if reflect.DeepEqual(prevProfiles, profiles) {
				time.Sleep(5 * time.Minute)

				if err := dp.refreshSession(); err != nil {
					return err
				}

				continue

			} else {
				prevProfiles = profiles
			}

			for name, url := range profiles {
				for _, watchPattern := range dp.config.WatchList {
					re, err := regexp.Compile(watchPattern)
					if err != nil {
						return fmt.Errorf("failed to compile regex watch pattern '%s': %v", watchPattern, err)
					}
					if re.MatchString(name) {
						output := filepath.Join(folder, strings.ReplaceAll(name, " ", "_"))
						if err := os.MkdirAll(output, 0750); err != nil {
							return fmt.Errorf("failed to create folder '%s': %v", output, err)
						}
						dp.Download(url, output)
					}
				}
			}
		}
	}
}

// DownloadPrompt prompts the user for which files to download from https://developer.apple.com/download
func (dp *DevPortal) DownloadPrompt(downloadType, folder string) error {
	switch downloadType {
	case "more":
		dloads, err := dp.getDownloads()
		if err != nil {
			return fmt.Errorf("failed to get the '%s' downloads: %v", downloadType, err)
		}

		var choices []string
		for _, dl := range dloads.Downloads {
			choices = append(choices, fmt.Sprintf("%s (%s)", dl.Name, dl.DateCreated))
		}

		dfiles := []int{}
		prompt := &survey.MultiSelect{
			Message:  "Select what file(s) to download:",
			Options:  choices,
			PageSize: dp.config.PageSize,
		}
		if err := survey.AskOne(prompt, &dfiles); err == terminal.InterruptErr {
			log.Warn("Exiting...")
			os.Exit(0)
		}

		for _, idx := range dfiles {
			for _, f := range dloads.Downloads[idx].Files {
				log.Debugf("Downloading: %s", f.URL())
				if err := dp.Download(f.URL(), folder); err != nil {
					log.Errorf("failed to download %s: %v", f.URL(), err)
				}
			}
		}
	case "os":
		ipsws, err := dp.getDevDownloads()
		if err != nil {
			return fmt.Errorf("failed to get the '%s' downloads: %v", downloadType, err)
		}

		versions := make([]string, 0, len(ipsws))
		for v := range ipsws {
			versions = append(versions, v)
		}
		sort.Strings(versions)

		version := ""
		promptVer := &survey.Select{
			Message:  "Choose an OS version:",
			Options:  versions,
			PageSize: 15,
		}
		if err := survey.AskOne(promptVer, &version); err != nil {
			if err == terminal.InterruptErr {
				log.Warn("Exiting...")
				os.Exit(0)
			}
			return err
		}

		if len(ipsws[version]) > 1 {
			var choices []string
			for _, ipsw := range ipsws[version] {
				choices = append(choices, ipsw.Title)
			}

			dfiles := []int{}
			prompt := &survey.MultiSelect{
				Message:  "Select what file(s) to download:",
				Options:  choices,
				PageSize: dp.config.PageSize,
			}
			if err := survey.AskOne(prompt, &dfiles); err != nil {
				if err == terminal.InterruptErr {
					log.Warn("Exiting...")
					os.Exit(0)
				}
				return err
			}

			for _, df := range dfiles {
				log.Debugf("Downloading: %s", ipsws[version][df].URL)
				dp.Download(ipsws[version][df].URL, folder)
			}
		} else {
			log.Debugf("Downloading: %s", ipsws[version][0].URL)
			dp.Download(ipsws[version][0].URL, folder)
		}
	case "profile":
		profiles, err := dp.getDevLoggingProfiles()
		if err != nil {
			return fmt.Errorf("failed to get the '%s' downloads: %v", downloadType, err)
		}

		var choices []string
		for name := range profiles {
			choices = append(choices, name)
		}

		sort.Strings(choices)

		dfiles := []string{}
		prompt := &survey.MultiSelect{
			Message:  "Select what file(s) to download:",
			Options:  choices,
			PageSize: dp.config.PageSize,
		}
		if err := survey.AskOne(prompt, &dfiles); err != nil {
			if err == terminal.InterruptErr {
				log.Warn("Exiting...")
				os.Exit(0)
			}
			return err
		}

		for _, df := range dfiles {
			output := filepath.Join(folder, strings.ReplaceAll(df, " ", "_"))
			if err := os.MkdirAll(output, 0750); err != nil {
				return fmt.Errorf("failed to create folder '%s': %v", output, err)
			}
			dp.Download(profiles[df], output)
		}
	}

	return nil
}

// Download downloads a file that requires a valid dev portal session
func (dp *DevPortal) Download(url, folder string) error {

	// proxy, insecure are null because we override the client below
	downloader := NewDownload(
		dp.config.Proxy,
		dp.config.Insecure,
		dp.config.SkipAll,
		dp.config.ResumeAll,
		dp.config.RestartAll,
		false,
		dp.config.Verbose,
	)
	// use authenticated client
	downloader.client = dp.Client

	destName := getDestName(url, dp.config.RemoveCommas)
	destName = filepath.Join(filepath.Clean(folder), filepath.Base(destName))

	if _, err := os.Stat(destName); os.IsNotExist(err) {

		log.WithFields(log.Fields{
			"file": destName,
		}).Info("Downloading")

		// download file
		downloader.URL = url
		downloader.DestName = destName

		err = downloader.Do()
		if err != nil {
			return fmt.Errorf("failed to download file: %v", err)
		}

	} else {
		log.Warnf("file already exists: %s", destName)
	}

	return nil
}

// DownloadADC downloads an ADC file that requires a valid ADCDownloadAuth cookie, but not full dev portal session auth
func (dp *DevPortal) DownloadADC(adcURL string) error {
	var adcDownloadAuth string

	u, err := url.Parse(adcURL)
	if err != nil {
		return fmt.Errorf("failed to parse url '%s': %v", adcURL, err)
	}

	req, err := http.NewRequest("GET", adcDownloadURL+u.Path, nil)
	if err != nil {
		return fmt.Errorf("failed to create http GET request: %v", err)
	}
	req.Header.Set("Content-Type", "*/*")

	response, err := dp.Client.Do(req)
	if err != nil {
		return err
	}

	if response.Header.Get("Set-Cookie") != "" {
		_, adcDownloadAuth, _ = strings.Cut(response.Header.Get("Set-Cookie"), "ADCDownloadAuth=")
	}

	// proxy, insecure are null because we override the client below
	downloader := NewDownload(
		dp.config.Proxy,
		dp.config.Insecure,
		dp.config.SkipAll,
		dp.config.ResumeAll,
		dp.config.RestartAll,
		false,
		dp.config.Verbose,
	)
	downloader.Headers = make(map[string]string)
	// use authenticated client
	downloader.client = dp.Client
	// set auth cookie (for authless downloads)
	downloader.Headers["Cookie"] = "ADCDownloadAuth=" + adcDownloadAuth

	// destName := getDestName(adcDownloadURL+path, dp.config.RemoveCommas)
	destName := getDestName(adcURL, dp.config.RemoveCommas)
	if _, err := os.Stat(destName); os.IsNotExist(err) {

		log.WithFields(log.Fields{
			"file": destName,
		}).Info("Downloading")

		// download file
		downloader.URL = adcURL
		downloader.DestName = destName

		return downloader.Do()
	}

	log.Warnf("file already exists: %s", destName)
	return nil
}

func (dp *DevPortal) DownloadKDK(version, build, folder string) (err error) {
	var urls []string
	urls = append(urls, fmt.Sprintf("%s?path=/macOS/Kernel_Debug_Kit_%s_build_%s/Kernel_Debug_Kit_%s_build_%s.dmg", downloadActionURL,
		version,
		build,
		version,
		build,
	))
	urls = append(urls, fmt.Sprintf("%s?path=/Developer_Tools/Kernel_Debug_Kit_%s_build_%s/Kernel_Debug_Kit_%s_build_%s.dmg", downloadActionURL,
		version,
		build,
		version,
		build,
	))
	urls = append(urls, fmt.Sprintf("https://download.developer.apple.com/macOS/Kernel_Debug_Kit_%s_build_%s/Kernel_Debug_Kit_%s_build_%s.dmg",
		version,
		build,
		version,
		build,
	))

	for _, url := range urls {
		log.WithField("url", url).Info("Downloading KDK")
		if err = dp.Download(url, folder); err == nil {
			return nil
		}
		utils.Indent(log.Warn, 2)(fmt.Sprintf("%v: Retrying...", err))
	}
	return
}

func (dp *DevPortal) GetDownloadsAsJSON(downloadType string, pretty bool) ([]byte, error) {
	switch downloadType {
	case "more":
		dloads, err := dp.getDownloads()
		if err != nil {
			return nil, fmt.Errorf("failed to get the '%s' downloads: %v", downloadType, err)
		}
		if pretty {
			return json.MarshalIndent(dloads, "", "    ")
		}
		return json.Marshal(dloads)
	case "os":
		ipsws, err := dp.getDevDownloads()
		if err != nil {
			return nil, fmt.Errorf("failed to get developer downloads: %v", err)
		}
		if pretty {
			return json.MarshalIndent(ipsws, "", "    ")
		}
		return json.Marshal(ipsws)
	case "profile":
		profs, err := dp.getDevLoggingProfiles()
		if err != nil {
			return nil, fmt.Errorf("failed to get developer downloads: %v", err)
		}
		if pretty {
			return json.MarshalIndent(profs, "", "    ")
		}
		return json.Marshal(profs)
	default:
		return nil, fmt.Errorf("invalid download type '%s'", downloadType)
	}
}

// getDownloads returns all the downloads in "More Downloads" - https://developer.apple.com/download/all/
func (dp *DevPortal) getDownloads() (*Downloads, error) {
	var downloads Downloads

	req, err := http.NewRequest("POST", listDownloadsActionURL, nil)
	if err != nil {
		return nil, fmt.Errorf("failed to create http POST request: %v", err)
	}
	req.Header.Set("Accept", "application/json")

	response, err := dp.Client.Do(req)
	if err != nil {
		return nil, err
	}
	defer response.Body.Close()

	body, err := io.ReadAll(response.Body)
	if err != nil {
		return nil, err
	}

	// os.WriteFile("downloads.json", body, 0644)

	if err := json.Unmarshal(body, &downloads); err != nil {
		return nil, fmt.Errorf("failed to deserialize response body JSON: %v", err)
	}

	log.Debugf("Get Downloads: (%d):\n%s\n", response.StatusCode, string(body))

	// sort by file name
	// sort.Slice(downloads.Downloads, func(i, j int) bool {
	// 	return downloads.Downloads[i].Name < downloads.Downloads[j].Name
	// })

	// sort by date created
	sort.Slice(downloads.Downloads, func(i, j int) bool {
		layout := "01/02/06 15:04"
		di, _ := time.Parse(layout, downloads.Downloads[i].DateCreated)
		dj, _ := time.Parse(layout, downloads.Downloads[j].DateCreated)
		return dj.Before(di)
	})

	return &downloads, nil
}

// getDevDownloads scrapes the https://developer.apple.com/download/ page for links
func (dp *DevPortal) getDevDownloads() (map[string][]DevDownload, error) {
	ipsws := make(map[string][]DevDownload)

	req, err := http.NewRequest("GET", downloadOSesURL, nil)
	if err != nil {
		return nil, fmt.Errorf("failed to create http GET request: %v", err)
	}

	response, err := dp.Client.Do(req)
	if err != nil {
		return nil, fmt.Errorf("failed to do GET request: %v", err)
	}
	defer response.Body.Close()

	if response.StatusCode != 200 {
		return nil, fmt.Errorf("failed to GET %s: response received %s", downloadOSesURL, response.Status)
	}

	doc, err := goquery.NewDocumentFromReader(response.Body)
	if err != nil {
		return nil, err
	}

	doc.Find("#main > section.section.section-downloads > div").Each(func(i int, s *goquery.Selection) {
		s.Find(".row").Each(func(index int, row *goquery.Selection) {

			// Get ALL the iOS ipsw links
			row.Find("ul.ios-list").Each(func(_ int, ul *goquery.Selection) {
				ul.Find("ul > li").Each(func(_ int, li *goquery.Selection) {
					a := li.Find("a[href]")
					href, _ := a.Attr("href")
					if strings.HasPrefix(href, "/services-account/download") {
						href = developerURL + href
					}
					p := li.Find("p")
					version := ul.Parent().Parent().Parent().Find("h3")
					ipsws[version.Text()] = append(ipsws[version.Text()], DevDownload{
						Title: strings.ReplaceAll(strings.ReplaceAll(a.Text(), "\u00a0", " "), "\n\t\t\t\t\t\t", ""),
						Build: p.Text(),
						URL:   href,
						Type:  "ios",
					})
				})
			})

			// Get ALL the tvOS ipsw links
			row.Find("ul.tvos-list").Each(func(_ int, ul *goquery.Selection) {
				ul.Find("li").Each(func(_ int, li *goquery.Selection) {
					a := li.Find("a[href]")
					href, _ := a.Attr("href")
					if strings.HasPrefix(href, "/services-account/download") {
						href = developerURL + href
					}
					p := li.Find("p")
					version := ul.Parent().Parent().Parent().Find("h3")
					ipsws[version.Text()] = append(ipsws[version.Text()], DevDownload{
						Title: strings.ReplaceAll(a.Text(), "\u00a0", " "),
						Build: p.Text(),
						URL:   href,
						// Type:  "tvos", TODO: this is commented out because macOS is also labeled as tvos
					})
				})
			})

		})
	})

	return ipsws, nil
}

// getDevLoggingProfiles scrapes the https://developer.apple.com/bug-reporting/profiles-and-logs/ page for links
func (dp *DevPortal) getDevLoggingProfiles() (map[string]string, error) {
	profiles := make(map[string]string)

	req, err := http.NewRequest("GET", downloadProfilesURL, nil)
	if err != nil {
		return nil, fmt.Errorf("failed to create http GET request: %v", err)
	}

	response, err := dp.Client.Do(req)
	if err != nil {
		return nil, fmt.Errorf("failed to do GET request: %v", err)
	}
	defer response.Body.Close()

	if response.StatusCode != 200 {
		return nil, fmt.Errorf("failed to GET %s: response received %s", downloadURL, response.Status)
	}

	doc, err := goquery.NewDocumentFromReader(response.Body)
	if err != nil {
		return nil, err
	}

	doc.Find("#main > section").Each(func(i int, s *goquery.Selection) {
		s.Find("li.profile > section").Each(func(index int, row *goquery.Selection) {
			var title string
			var platform string
			// Get ALL the profile links
			row.Find("section.column").Each(func(_ int, section *goquery.Selection) {
				section.Find("span").Each(func(_ int, span *goquery.Selection) {
					if _, ok := span.Attr("data-profile-detail"); ok {
						title = span.Text()
					}
					if _, ok := span.Attr("class"); ok {
						platform = span.Text()
					}
				})
				section.Find("ul > li").Each(func(_ int, li *goquery.Selection) {
					a := li.Find("a[href]")
					href, _ := a.Attr("href")
					if strings.HasSuffix(href, ".mobileconfig") {
						profiles[fmt.Sprintf("%s (%s)", title, platform)] = href
					}
				})
			})
		})
	})

	return profiles, nil
}
