Merge pull request #72 from sailpoint-oss/fix/golangTime

Temporary Fix for Workgroups created and modified times missing secon…
sailpoint-oss · Oct 18, 2024 · ead148f · ead148f
2 parents e752e90 + 78dfa90
commit ead148f
Show file tree

Hide file tree

Showing 2 changed files with 121 additions and 10 deletions.
diff --git a/api_beta/model_workgroup_dto.go b/api_beta/model_workgroup_dto.go
diff --git a/sdk-resources/postscript.js b/sdk-resources/postscript.js
@@ -172,6 +172,84 @@ const fixFiles = function (myArray) {
       rawDataArra = fileOut.slice();
       fileOut = [];
     }
+
+    if (file.includes("model_workgroup_dto.go")) {
+
+    const sailPointTimeCode = `type SailPointTime struct {
+    time.Time
+}
+    
+func (m *SailPointTime) UnmarshalJSON(data []byte) error {
+        if string(data) == "null" || string(data) == \`""\` {
+            return nil;
+        }
+    
+        // Strip the quotes from the data
+        str := string(data);
+        str = str[1 : len(str)-1];
+    
+        // Try parsing with seconds first
+        formats := []string{
+            time.RFC3339,                   // Format with seconds (e.g., "2012-04-23T18:25:43Z")
+            "2006-01-02T15:04Z",            // Format without seconds (e.g., "2012-04-23T18:25Z")
+            "2006-01-02T15:04:05.999999999", // Optional milliseconds and nanoseconds
+        }
+    
+        var err error;
+        for _, format := range formats {
+            var t time.Time;
+            t, err = time.Parse(format, str);
+            if err == nil {
+                *m = SailPointTime{t};
+                return nil;
+            }
+        }
+    
+        return nil;
+}
+
+func (o *WorkgroupDto) UnmarshalJSON(data []byte) (err error) {`;
+
+
+      for (const line of rawDataArra) {
+        if (line.includes('Created *time.Time `json:"created,omitempty"`')) {
+          fileOut.push(line.replace('Created *time.Time `json:"created,omitempty"`', 'Created *SailPointTime `json:"created,omitempty"`'));
+          madeChange = true;
+        } else if (line.includes('Modified *time.Time `json:"modified,omitempty"`')) {
+          fileOut.push(line.replace('Modified *time.Time `json:"modified,omitempty"`', 'Modified *SailPointTime  `json:"modified,omitempty"`'));
+          madeChange = true;
+        } else if (line.includes('func (o *WorkgroupDto) GetCreated() time.Time {')) {
+          fileOut.push(line.replace('func (o *WorkgroupDto) GetCreated() time.Time {', 'func (o *WorkgroupDto) GetCreated() SailPointTime {'));
+          madeChange = true;
+        } else if (line.includes('func (o *WorkgroupDto) GetCreatedOk() (*time.Time, bool) {')) {
+          fileOut.push(line.replace('func (o *WorkgroupDto) GetCreatedOk() (*time.Time, bool) {', 'func (o *WorkgroupDto) GetCreatedOk() (*SailPointTime, bool) {'));
+          madeChange = true;
+        } else if (line.includes('func (o *WorkgroupDto) GetModified() time.Time {')) {
+          fileOut.push(line.replace('func (o *WorkgroupDto) GetModified() time.Time {', 'func (o *WorkgroupDto) GetModified() SailPointTime {'));
+          madeChange = true;
+        } else if (line.includes('var ret time.Time')) {
+          fileOut.push(line.replace('var ret time.Time', 'var ret SailPointTime'));
+          madeChange = true;
+        } else if (line.includes('func (o *WorkgroupDto) GetModifiedOk() (*time.Time, bool) {')) {
+          fileOut.push(line.replace('func (o *WorkgroupDto) GetModifiedOk() (*time.Time, bool) {', 'func (o *WorkgroupDto) GetModifiedOk() (*SailPointTime, bool) {'));
+          madeChange = true;
+        } else if (line.includes('func (o *WorkgroupDto) SetModified(v time.Time) {')) {
+          fileOut.push(line.replace('func (o *WorkgroupDto) SetModified(v time.Time) {', 'func (o *WorkgroupDto) SetModified(v SailPointTime) {'));
+          madeChange = true;
+        } else if (line.includes('func (o *WorkgroupDto) SetCreated(v time.Time) {')) {
+          fileOut.push(line.replace('func (o *WorkgroupDto) SetCreated(v time.Time) {', 'func (o *WorkgroupDto) SetCreated(v SailPointTime) {'));
+          madeChange = true;
+        } else if (line.includes('func (o *WorkgroupDto) UnmarshalJSON(data []byte) (err error) {')) {
+          fileOut.push(line.replace('func (o *WorkgroupDto) UnmarshalJSON(data []byte) (err error) {', sailPointTimeCode));
+          madeChange = true;
+        } else {
+          fileOut.push(line);
+        }
+      }
+
+      rawDataArra = fileOut.slice();
+      fileOut = [];
+    }
 
     if (madeChange) {
       fixCheck += 1;