From 0cbce4ae05d28c38afa10b9d2a37a9fb0c80a214 Mon Sep 17 00:00:00 2001
From: Yihau Chen <a122092487@gmail.com>
Date: Tue, 15 Jun 2021 17:03:01 +0800
Subject: [PATCH] implement tokenprog SetAuthority

---
 tokenprog/instruction.go | 41 ++++++++++++++++++++++++++++++++++++++--
 1 file changed, 39 insertions(+), 2 deletions(-)

diff --git a/tokenprog/instruction.go b/tokenprog/instruction.go
index 3acd617d..f7b117db 100644
--- a/tokenprog/instruction.go
+++ b/tokenprog/instruction.go
@@ -197,8 +197,45 @@ func Revoke(srcPubkey, authPubkey common.PublicKey, signerPubkeys []common.Publi
 	}
 }
 
-func SetAuthority() types.Instruction {
-	panic("not implement yet")
+type AuthorityType uint8
+
+const (
+	AuthorityTypeMintTokens AuthorityType = iota
+	AuthorityTypeFreezeAccount
+	AuthorityTypeAccountOwner
+	AuthorityTypeCloseAccount
+)
+
+func SetAuthority(accountPubkey, newAuthPubkey common.PublicKey, authType AuthorityType, authPubkey common.PublicKey, signerPubkeys []common.PublicKey) types.Instruction {
+	data, err := common.SerializeData(struct {
+		Instruction   Instruction
+		AuthorityType AuthorityType
+		Option        bool
+		NewAuthPubkey common.PublicKey
+	}{
+		Instruction:   InstructionSetAuthority,
+		AuthorityType: authType,
+		Option:        newAuthPubkey != common.PublicKey{},
+		NewAuthPubkey: newAuthPubkey,
+	})
+	if err != nil {
+		panic(err)
+	}
+
+	accounts := make([]types.AccountMeta, 0, 2+len(signerPubkeys))
+	accounts = append(accounts,
+		types.AccountMeta{PubKey: accountPubkey, IsSigner: false, IsWritable: true},
+		types.AccountMeta{PubKey: authPubkey, IsSigner: len(signerPubkeys) == 0, IsWritable: false},
+	)
+	for _, signerPubkey := range signerPubkeys {
+		accounts = append(accounts, types.AccountMeta{PubKey: signerPubkey, IsSigner: true, IsWritable: false})
+	}
+
+	return types.Instruction{
+		ProgramID: common.TokenProgramID,
+		Accounts:  accounts,
+		Data:      data,
+	}
 }
 
 func MintTo(mintPubkey, destPubkey, authPubkey common.PublicKey, signerPubkeys []common.PublicKey, amount uint64) types.Instruction {