-
Notifications
You must be signed in to change notification settings - Fork 8
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
This change adds a configuration option to post comments to a chatter object instead of a casecomment object. The feature is disabled by default and has to be explicitly enabled via the configuration file: ```yaml salesforce: enable-chatter: true ``` The change also introduces the `salesforce-test` binary to test Salesforce queries. Closes: SET-627 Signed-off-by: Nicolas Bock <[email protected]>
- Loading branch information
1 parent
9cfc677
commit e9a493a
Showing
8 changed files
with
221 additions
and
15 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,175 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "log" | ||
| "strconv" | ||
|
|
||
| "github.com/canonical/athena-core/pkg/common" | ||
| "github.com/canonical/athena-core/pkg/config" | ||
| "gopkg.in/alecthomas/kingpin.v2" | ||
| ) | ||
|
|
||
| var configs = common.StringList( | ||
| kingpin.Flag("config", "Path to the athena configuration file").Default("/etc/athena/main.yaml").Short('c'), | ||
| ) | ||
|
|
||
| var allCases = kingpin.Flag("all-cases", "Get all cases").Default("false").Bool() | ||
| var allFeedComments = kingpin.Flag("all-feed-comments", "Get all FeedComments").Default("false").Bool() | ||
| var allFeedItems = kingpin.Flag("all-feed-items", "Get all FeedItems").Default("false").Bool() | ||
| var caseNumber = kingpin.Flag("case-id", "The case ID to query").Default("").String() | ||
| var commentVisibility = kingpin.Flag("visibility", "Set the comment visibility {public, private)").Default("private").String() | ||
| var getChatter = kingpin.Flag("chatter", "Get all chatter objects of case").Default("false").Bool() | ||
| var getComments = kingpin.Flag("comments", "Get all comments of case").Default("false").Bool() | ||
| var newChatter = kingpin.Flag("new-chatter", "Add a new chatter comment to the case").Default("").String() | ||
|
|
||
| func main() { | ||
| kingpin.HelpFlag.Short('h') | ||
| kingpin.Parse() | ||
|
|
||
| switch *commentVisibility { | ||
| case "public", "private": | ||
| // All good, do nothing. | ||
| default: | ||
| log.Fatal("Invalid visibility value. Allowed values are 'public' and 'private'.") | ||
| } | ||
|
|
||
| cfg, err := config.NewConfigFromFile(*configs) | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
|
|
||
| sfClient, err := common.NewSalesforceClient(cfg) | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
|
|
||
| if *allCases { | ||
| getAllCases(sfClient) | ||
| } | ||
|
|
||
| if *allFeedComments { | ||
| getAllFeedComments(sfClient) | ||
| } | ||
|
|
||
| if *allFeedItems { | ||
| getAllFeedItems(sfClient) | ||
| } | ||
|
|
||
| if len(*caseNumber) > 0 { | ||
| caseId := getCase(sfClient) | ||
| if *getComments { | ||
| getAllCaseComments(caseId, sfClient) | ||
| } | ||
|
|
||
| if *getChatter { | ||
| getAllChatterComments(caseId, sfClient) | ||
| } | ||
|
|
||
| if len(*newChatter) > 0 { | ||
| newChatterComment(sfClient, caseId, *newChatter) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| func newChatterComment(sfClient common.SalesforceClient, caseId string, comment string) { | ||
| log.Print("Added new chatter comment") | ||
| visibility := "" | ||
| switch *commentVisibility { | ||
| case "public": | ||
| visibility = "AllUsers" | ||
| case "private": | ||
| visibility = "InternalUsers" | ||
| default: | ||
| log.Fatal("Unknown visibility") | ||
| } | ||
| sfClient.SObject("FeedItem"). | ||
| Set("ParentId", caseId). | ||
| Set("Body", comment). | ||
| Set("Visibility", visibility). | ||
| Create() | ||
| } | ||
|
|
||
| func getAllChatterComments(caseId string, sfClient common.SalesforceClient) { | ||
| log.Print("Getting case chatter comments") | ||
| query := fmt.Sprintf("SELECT Id, Body FROM FeedItem WHERE ParentID = '%s'", caseId) | ||
| records, err := sfClient.Query(query) | ||
| if err != nil { | ||
| log.Fatalf("Failed to get chatter comments: %v", err) | ||
| } | ||
| if len(records.Records) == 0 { | ||
| log.Fatal("Could not find any chatter comments") | ||
| } | ||
| for _, comment := range records.Records { | ||
| log.Printf("%s: %s", comment["Id"], comment["Body"]) | ||
| } | ||
| } | ||
|
|
||
| func getAllCaseComments(caseId string, sfClient common.SalesforceClient) { | ||
| log.Print("Getting case comments") | ||
| query := fmt.Sprintf("SELECT Id, CommentBody FROM CaseComment WHERE ParentId = '%s'", caseId) | ||
| records, err := sfClient.Query(query) | ||
| if err != nil { | ||
| log.Fatalf("Failed to get case comments: %v", err) | ||
| } | ||
| if len(records.Records) == 0 { | ||
| log.Fatal("Could not find any case comments") | ||
| } | ||
| for _, comment := range records.Records { | ||
| log.Printf("%s", comment["CommentBody"]) | ||
| } | ||
| } | ||
|
|
||
| func getCase(sfClient common.SalesforceClient) string { | ||
| caseNumberAsInt, err := strconv.ParseInt(*caseNumber, 10, 64) | ||
| if err != nil { | ||
| log.Fatalf("Failed to parse the case number %s", *caseNumber) | ||
| } | ||
| caseNumberFormatted := fmt.Sprintf("%08d", caseNumberAsInt) | ||
|
|
||
| log.Printf("Searching for case %s", caseNumberFormatted) | ||
| query := fmt.Sprintf("SELECT Id, CaseNumber FROM Case WHERE CaseNumber = '%s'", caseNumberFormatted) | ||
| records, err := sfClient.Query(query) | ||
| if err != nil { | ||
| log.Fatalf("Failed to query Salesforce: %v", err) | ||
| } | ||
| if len(records.Records) > 0 { | ||
| log.Printf("%s: %s", records.Records[0]["Id"], records.Records[0]["CaseNumber"]) | ||
| } else { | ||
| log.Fatalf("Case with ID %s does not exist.\n", *caseNumber) | ||
| } | ||
| return fmt.Sprintf("%s", records.Records[0]["Id"]) | ||
| } | ||
|
|
||
| func getAllFeedItems(sfClient common.SalesforceClient) { | ||
| log.Println("All FeedItems:") | ||
| records, err := sfClient.Query("SELECT Id from FeedItem") | ||
| if err != nil { | ||
| log.Fatalln("Failed to query for all FeedItems") | ||
| } | ||
| for _, result := range records.Records { | ||
| log.Printf("%s", result["Id"]) | ||
| } | ||
| } | ||
|
|
||
| func getAllFeedComments(sfClient common.SalesforceClient) { | ||
| log.Println("All FeedComments:") | ||
| records, err := sfClient.Query("SELECT Id from FeedComment") | ||
| if err != nil { | ||
| log.Fatalln("Failed to query for all FeedComments") | ||
| } | ||
| for _, result := range records.Records { | ||
| log.Printf("%s", result["Id"]) | ||
| } | ||
| } | ||
|
|
||
| func getAllCases(sfClient common.SalesforceClient) { | ||
| log.Println("All cases:") | ||
| records, err := sfClient.Query("SELECT Id, CaseNumber from Case") | ||
| if err != nil { | ||
| log.Fatalln("Failed to query for all cases") | ||
| } | ||
| for _, result := range records.Records { | ||
| log.Printf("%s: %s", result["Id"], result["CaseNumber"]) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters