make disableInvites actually disable existing invites as well (#301)

bluesky-social · Sep 8, 2023 · f934d62 · f934d62
2 parents 2fcb59a + 1a69665
commit f934d62
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/api/extra.go b/api/extra.go
@@ -164,7 +164,6 @@ func (dr *ProdHandleResolver) resolveWellKnown(ctx context.Context, handle strin
 }
 
 func (dr *ProdHandleResolver) resolveDNS(ctx context.Context, handle string) (string, error) {
-
 	res, err := net.LookupTXT("_atproto." + handle)
 	if err != nil {
 		return "", fmt.Errorf("handle lookup failed: %w", err)

diff --git a/cmd/gosky/admin.go b/cmd/gosky/admin.go
@@ -459,9 +459,19 @@ var disableInvitesCmd = &cli.Command{
 			handle = resp
 		}
 
-		return atproto.AdminDisableAccountInvites(ctx, xrpcc, &atproto.AdminDisableAccountInvites_Input{
+		if err := atproto.AdminDisableAccountInvites(ctx, xrpcc, &atproto.AdminDisableAccountInvites_Input{
 			Account: handle,
-		})
+		}); err != nil {
+			return err
+		}
+
+		if err := atproto.AdminDisableInviteCodes(ctx, xrpcc, &atproto.AdminDisableInviteCodes_Input{
+			Accounts: []string{handle},
+		}); err != nil {
+			return err
+		}
+
+		return nil
 	},
 }