Merge pull request #21 from essentialkaos/develop

Version 1.5.0

changelog.md

@@ -1,5 +1,10 @@
 ## Changelog
 
+#### v1.5.0
+
+* `[tmp]` Improved error handling
+* `[tmp]` Changed name pattern of temporary files and directories
+
 #### v1.4.5
 
 * `[pid]` Fixed bug with pid file creation

tmp/tmp.go

@@ -9,6 +9,7 @@ package tmp
 // ////////////////////////////////////////////////////////////////////////////////// //
 
 import (
+	"fmt"
 	"os"
 	"path"
 
@@ -33,12 +34,26 @@ var Dir = "/tmp"
 // ////////////////////////////////////////////////////////////////////////////////// //
 
 // NewTemp create new Temp structure
-func NewTemp(args ...string) *Temp {
-	if len(args) == 0 {
-		return &Temp{Dir: Dir}
+func NewTemp(args ...string) (*Temp, error) {
+	tempDir := path.Clean(Dir)
+
+	if len(args) != 0 {
+		tempDir = path.Clean(args[0])
+	}
+
+	if !fsutil.IsExist(tempDir) {
+		return nil, fmt.Errorf("Directory %s is not exist", tempDir)
+	}
+
+	if !fsutil.IsDir(tempDir) {
+		return nil, fmt.Errorf("%s is not a directory", tempDir)
+	}
+
+	if !fsutil.IsWritable(tempDir) {
+		return nil, fmt.Errorf("Directory %s is not writable", tempDir)
 	}
 
-	return &Temp{Dir: path.Clean(args[0])}
+	return &Temp{Dir: tempDir}, nil
 }
 
 // ////////////////////////////////////////////////////////////////////////////////// //
@@ -110,12 +125,13 @@ func (t *Temp) Clean() {
 
 // ////////////////////////////////////////////////////////////////////////////////// //
 
+// getTempName return name of temporary file
 func getTempName(dir, name string) string {
 	var result string
 
 	for {
 		if name != "" {
-			result = path.Join(dir, "_"+name+"_"+rand.String(12))
+			result = path.Join(dir, "_"+rand.String(12)+"_"+name)
 		} else {
 			result = path.Join(dir, "_tmp_"+rand.String(12))
 		}

tmp/tmp_test.go

@@ -37,19 +37,34 @@ func (ts *TmpSuite) SetUpSuite(c *C) {
 }
 
 func (ts *TmpSuite) TestMk(c *C) {
-	t := NewTemp()
+	t, err := NewTemp()
 
+	c.Assert(err, IsNil)
 	c.Assert(t, NotNil)
 	c.Assert(t.Dir, Equals, "/tmp")
 
 	t.Clean()
 }
 
 func (ts *TmpSuite) TestErrors(c *C) {
-	t := NewTemp("/")
+	t, err := NewTemp("/")
 
-	c.Assert(t, NotNil)
-	c.Assert(t.Dir, Equals, "/")
+	c.Assert(err, NotNil)
+	c.Assert(t, IsNil)
+
+	t, err = NewTemp("/tmpz")
+
+	c.Assert(err, NotNil)
+	c.Assert(t, IsNil)
+
+	os.Create(ts.TempDir + "/test_")
+
+	t, err = NewTemp(ts.TempDir + "/test_")
+
+	c.Assert(err, NotNil)
+	c.Assert(t, IsNil)
+
+	t = &Temp{Dir: "/"}
 
 	tmpDir, err := t.MkDir("test")
 
@@ -64,8 +79,9 @@ func (ts *TmpSuite) TestErrors(c *C) {
 }
 
 func (ts *TmpSuite) TestMkDir(c *C) {
-	t := NewTemp(ts.TempDir)
+	t, err := NewTemp(ts.TempDir)
 
+	c.Assert(err, IsNil)
 	c.Assert(t, NotNil)
 	c.Assert(t.Dir, Equals, ts.TempDir)
 
@@ -85,8 +101,9 @@ func (ts *TmpSuite) TestMkDir(c *C) {
 }
 
 func (ts *TmpSuite) TestMkFile(c *C) {
-	t := NewTemp(ts.TempDir)
+	t, err := NewTemp(ts.TempDir)
 
+	c.Assert(err, IsNil)
 	c.Assert(t, NotNil)
 	c.Assert(t.Dir, Equals, ts.TempDir)
 
@@ -107,15 +124,16 @@ func (ts *TmpSuite) TestMkFile(c *C) {
 }
 
 func (ts *TmpSuite) TestMkName(c *C) {
-	t := NewTemp(ts.TempDir)
+	t, err := NewTemp(ts.TempDir)
 
+	c.Assert(err, IsNil)
 	c.Assert(t, NotNil)
 	c.Assert(t.Dir, Equals, ts.TempDir)
 
 	c.Assert(t.MkName(), Not(Equals), "")
 	c.Assert(t.MkName("1234"), Not(Equals), "")
 
-	ln := len(ts.TempDir + "/_1234")
+	ln := len(ts.TempDir + "/")
 
-	c.Assert(t.MkName("1234")[:ln], Equals, ts.TempDir+"/_1234")
+	c.Assert(t.MkName("1234.json")[ln+14:], Equals, "1234.json")
 }