Something went wrong. Try again.
This repository has no description
Something went wrong. Try again.
3.1 kB · 122 lines
Go
at ci
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123package signup
// We have this extra code here for now since the xrpcclient package// only supports OAuth'd requests; these are unauthenticated or use PDS admin auth.
import ( "bytes" "encoding/json" "fmt" "io" "net/http" "net/url")
// makePdsRequest is a helper method to make requests to the PDS servicefunc (s *Signup) makePdsRequest(method, endpoint string, body interface{}, useAuth bool) (*http.Response, error) { jsonData, err := json.Marshal(body) if err != nil { return nil, err }
url := fmt.Sprintf("%s/xrpc/%s", s.config.Pds.Host, endpoint) req, err := http.NewRequest(method, url, bytes.NewBuffer(jsonData)) if err != nil { return nil, err }
req.Header.Set("Content-Type", "application/json")
if useAuth { req.SetBasicAuth("admin", s.config.Pds.AdminSecret) }
return http.DefaultClient.Do(req)}
// handlePdsError processes error responses from the PDS servicefunc (s *Signup) handlePdsError(resp *http.Response, action string) error { var errorResp struct { Error string `json:"error"` Message string `json:"message"` }
respBody, _ := io.ReadAll(resp.Body) if err := json.Unmarshal(respBody, &errorResp); err == nil && errorResp.Message != "" { return fmt.Errorf("Failed to %s: %s - %s.", action, errorResp.Error, errorResp.Message) }
// Fallback if we couldn't parse the error return fmt.Errorf("failed to %s, status code: %d", action, resp.StatusCode)}
func (s *Signup) inviteCodeRequest() (string, error) { body := map[string]any{"useCount": 1}
resp, err := s.makePdsRequest("POST", "com.atproto.server.createInviteCode", body, true) if err != nil { return "", err } defer resp.Body.Close()
if resp.StatusCode != http.StatusOK { return "", s.handlePdsError(resp, "create invite code") }
var result map[string]string json.NewDecoder(resp.Body).Decode(&result) return result["code"], nil}
func (s *Signup) createAccountRequest(username, password, email, code string) (string, error) { parsedURL, err := url.Parse(s.config.Pds.Host) if err != nil { return "", fmt.Errorf("invalid PDS host URL: %w", err) }
pdsDomain := parsedURL.Hostname()
body := map[string]string{ "email": email, "handle": fmt.Sprintf("%s.%s", username, pdsDomain), "password": password, "inviteCode": code, }
resp, err := s.makePdsRequest("POST", "com.atproto.server.createAccount", body, false) if err != nil { return "", err } defer resp.Body.Close()
if resp.StatusCode != http.StatusOK { return "", s.handlePdsError(resp, "create account") }
var result struct { DID string `json:"did"` } if err := json.NewDecoder(resp.Body).Decode(&result); err != nil { return "", fmt.Errorf("failed to decode create account response: %w", err) }
return result.DID, nil}
func (s *Signup) deleteAccountRequest(did string) error { body := map[string]string{ "did": did, }
resp, err := s.makePdsRequest("POST", "com.atproto.admin.deleteAccount", body, true) if err != nil { return err } defer resp.Body.Close()
if resp.StatusCode != http.StatusOK { return s.handlePdsError(resp, "delete account") }
return nil}