@@ -4,6 +4,7 @@ package anthropic
44import (
55 "errors"
66 "os"
7+ "strings"
78
89 "github.com/fatih/color"
910 "github.com/jedib0t/go-pretty/v6/table"
@@ -17,7 +18,8 @@ var _ analyzers.Analyzer = (*Analyzer)(nil)
1718
1819const (
1920 // Key Types
20- APIKey = "API-Key"
21+ APIKey = "API-Key"
22+ AdminKey = "Admin-Key"
2123)
2224
2325type Analyzer struct {
@@ -28,7 +30,6 @@ type Analyzer struct {
2830type SecretInfo struct {
2931 Valid bool
3032 Type string // key type - TODO: Handle Anthropic Admin Keys
31- Reference string
3233 AnthropicResources []AnthropicResource
3334 Permissions string // always full_access
3435 Misc map [string ]string
@@ -39,6 +40,7 @@ type AnthropicResource struct {
3940 ID string
4041 Name string
4142 Type string
43+ Parent * AnthropicResource
4244 Metadata map [string ]string
4345}
4446
@@ -73,7 +75,7 @@ func AnalyzeAndPrintPermissions(cfg *config.Config, key string) {
7375 }
7476
7577 if info .Valid {
76- color .Green ("[!] Valid Anthropic API key \n \n " )
78+ color .Green ("[!] Valid Anthropic %s \n \n " , info . Type )
7779 // no user information
7880 // print full access permission
7981 printPermission (info .Permissions )
@@ -88,16 +90,23 @@ func AnalyzePermissions(cfg *config.Config, key string) (*SecretInfo, error) {
8890 // create a HTTP client
8991 client := analyzers .NewAnalyzeClient (cfg )
9092
91- var secretInfo = & SecretInfo {
92- Type : APIKey , // TODO: implement Admin-Key type as well
93- }
93+ keyType := getKeyType (key )
9494
95- if err := listModels ( client , key , secretInfo ); err != nil {
96- return nil , err
95+ var secretInfo = & SecretInfo {
96+ Type : keyType ,
9797 }
9898
99- if err := listMessageBatches (client , key , secretInfo ); err != nil {
100- return nil , err
99+ switch keyType {
100+ case APIKey :
101+ if err := captureAPIKeyResources (client , key , secretInfo ); err != nil {
102+ return nil , err
103+ }
104+ case AdminKey :
105+ if err := captureAdminKeyResources (client , key , secretInfo ); err != nil {
106+ return nil , err
107+ }
108+ default :
109+ return nil , errors .New ("unsupported key type" )
101110 }
102111
103112 // anthropic key has full access only
@@ -133,6 +142,14 @@ func secretInfoToAnalyzerResult(info *SecretInfo) *analyzers.AnalyzerResult {
133142 },
134143 }
135144
145+ if Anthropicresource .Parent != nil {
146+ binding .Resource .Parent = & analyzers.Resource {
147+ Name : Anthropicresource .Parent .Name ,
148+ FullyQualifiedName : Anthropicresource .Parent .ID ,
149+ Type : Anthropicresource .Parent .Type ,
150+ }
151+ }
152+
136153 for key , value := range Anthropicresource .Metadata {
137154 binding .Resource .Metadata [key ] = value
138155 }
@@ -162,3 +179,14 @@ func printAnthropicResources(resources []AnthropicResource) {
162179 }
163180 t .Render ()
164181}
182+
183+ // getKeyType return the type of key
184+ func getKeyType (key string ) string {
185+ if strings .Contains (key , "sk-ant-admin01" ) {
186+ return AdminKey
187+ } else if strings .Contains (key , "sk-ant-api03" ) {
188+ return APIKey
189+ }
190+
191+ return ""
192+ }
0 commit comments