Add utility module

[ric-plt/xapp-frame.git] / pkg / xapp / Utils.go

diff --git a/pkg/xapp/Utils.go b/pkg/xapp/Utils.go
new file mode 100755 (executable)
index 0000000..90bc093
--- /dev/null
+++ b/pkg/xapp/Utils.go
@@ -0,0 +1,153 @@
+/*
+==================================================================================
+  Copyright (c) 2019 AT&T Intellectual Property.
+  Copyright (c) 2019 Nokia
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+==================================================================================
+*/
+
+package xapp
+
+import (
+       "archive/zip"
+       "io"
+       "io/ioutil"
+       "os"
+       "path/filepath"
+       "strings"
+)
+
+type Utils struct {
+       baseDir string
+       status  string
+}
+
+func NewUtils() *Utils {
+       b := Config.GetString("controls.symptomdata.baseDir")
+       if b == "" {
+               b = "/tmp/symptomdata/"
+       }
+
+       return &Utils{
+               baseDir: b,
+       }
+}
+
+func (u *Utils) FileExists(f string) bool {
+       _, err := os.Stat(f)
+       return err == nil || os.IsExist(err)
+}
+
+func (u *Utils) CreateDir(path string) error {
+       if u.FileExists(path) {
+               os.RemoveAll(path)
+       }
+       err := os.MkdirAll(path, os.ModePerm)
+       if err != nil {
+               return err
+       }
+       os.Chmod(path, os.ModePerm)
+       return nil
+}
+
+func (u *Utils) DeleteFile(fileName string) {
+       os.Remove(fileName)
+}
+
+func (u *Utils) AddFileToZip(zipWriter *zip.Writer, filePath string, filename string) error {
+       fileToZip, err := os.Open(filename)
+       if err != nil {
+               return err
+       }
+       defer fileToZip.Close()
+
+       info, err := fileToZip.Stat()
+       if err != nil {
+               return err
+       }
+
+       header, err := zip.FileInfoHeader(info)
+       if err != nil {
+               return err
+       }
+
+       if strings.HasPrefix(filename, filePath) {
+               filename = strings.TrimPrefix(filename, filePath)
+       }
+       header.Name = filename
+       header.Method = zip.Deflate
+
+       writer, err := zipWriter.CreateHeader(header)
+       if err != nil {
+               return err
+       }
+       if info.Size() > 0 {
+               _, err = io.Copy(writer, fileToZip)
+       }
+       return err
+}
+
+func (u *Utils) ZipFiles(newZipFile *os.File, filePath string, files []string) error {
+       defer newZipFile.Close()
+       zipWriter := zip.NewWriter(newZipFile)
+       defer zipWriter.Close()
+       for _, file := range files {
+               if err := u.AddFileToZip(zipWriter, filePath, file); err != nil {
+                       Logger.Error("AddFileToZip() failed: %+v", err.Error())
+                       return err
+               }
+       }
+
+       return nil
+}
+
+func (u *Utils) FetchFiles(filePath string, fileList []string) []string {
+       files, err := ioutil.ReadDir(filePath)
+       if err != nil {
+               Logger.Error("ioutil.ReadDir failed: %+v", err)
+               return nil
+       }
+       for _, file := range files {
+               if !file.IsDir() {
+                       fileList = append(fileList, filepath.Join(filePath, file.Name()))
+               } else {
+                       subPath := filepath.Join(filePath, file.Name())
+                       subFiles, _ := ioutil.ReadDir(subPath)
+                       for _, subFile := range subFiles {
+                               if !subFile.IsDir() {
+                                       fileList = append(fileList, filepath.Join(subPath, subFile.Name()))
+                               } else {
+                                       fileList = u.FetchFiles(filepath.Join(subPath, subFile.Name()), fileList)
+                               }
+                       }
+               }
+       }
+       return fileList
+}
+
+func (u *Utils) WriteToFile(fileName string, data string) error {
+       f, err := os.Create(fileName)
+       defer f.Close()
+
+       if err != nil {
+               Logger.Error("Unable to create file %s': %+v", fileName, err)
+       } else {
+               _, err := io.WriteString(f, data)
+               if err != nil {
+                       Logger.Error("Unable to write to file '%s': %+v", fileName, err)
+                       u.DeleteFile(fileName)
+               }
+       }
+       return err
+}