stdlib/strings: Corrected test_isa to work on Text instead of AbstractArrayRead.

[nit.git] / lib / standard / file.nit

diff --git a/lib/standard/file.nit b/lib/standard/file.nit
index 239f785..64094f7 100644 (file)
--- a/lib/standard/file.nit
+++ b/lib/standard/file.nit
@@ -20,6 +20,14 @@ intrude import string
 import string_search
 import time
 
+in "C Header" `{
+       #include <dirent.h>
+       #include <string.h>
+       #include <sys/types.h>
+       #include <sys/stat.h>
+       #include <unistd.h>
+`}
+
 redef class Object
 # Simple I/O
 
@@ -207,6 +215,21 @@ redef class String
        # Remove a file, return true if success
        fun file_delete: Bool do return to_cstring.file_delete
 
+       # Copy content of file at `self` to `dest`
+       fun file_copy_to(dest: String)
+       do
+               var input = new IFStream.open(self)
+               var output = new OFStream.open(dest)
+
+               while not input.eof do
+                       var buffer = input.read(1024)
+                       output.write buffer
+               end
+
+               input.close
+               output.close
+       end
+
        # remove the trailing extension "ext"
        fun strip_extension(ext: String): String
        do
@@ -240,7 +263,7 @@ redef class String
        fun dirname: String
        do
                var l = _length - 1 # Index of the last char
-               if l > 0 and self[l] == '/' then l -= 1 # remove trailing `/`
+               if l > 0 and self.chars[l] == '/' then l -= 1 # remove trailing `/`
                var pos = last_index_of_from('/', l)
                if pos > 0 then
                        return substring(0, pos)
@@ -309,7 +332,7 @@ redef class String
        do
                if path.is_empty then return self
                if self.is_empty then return path
-               if path[0] == '/' then return path
+               if path.chars[0] == '/' then return path
                return "{self}/{path}"
        end
 
@@ -353,7 +376,36 @@ redef class String
        end
 
        # returns files contained within the directory represented by self
-       fun files : Set[ String ] is extern import HashSet, HashSet::add, NativeString::to_s, String::to_cstring, HashSet[String] as( Set[String] ), String as( Object )
+       fun files : Set[ String ] is extern import HashSet[String], HashSet[String].add, NativeString.to_s, String.to_cstring, HashSet[String].as(Set[String]) `{
+               char *dir_path;
+               DIR *dir;
+
+               dir_path = String_to_cstring( recv );
+               if ((dir = opendir(dir_path)) == NULL)
+               {
+                       perror( dir_path );
+                       exit( 1 );
+               }
+               else
+               {
+                       HashSet_of_String results;
+                       String file_name;
+                       struct dirent *de;
+
+                       results = new_HashSet_of_String();
+
+                       while ( ( de = readdir( dir ) ) != NULL )
+                               if ( strcmp( de->d_name, ".." ) != 0 &&
+                                       strcmp( de->d_name, "." ) != 0 )
+                               {
+                                       file_name = NativeString_to_s( strdup( de->d_name ) );
+                                       HashSet_of_String_add( results, file_name );
+                               }
+
+                       closedir( dir );
+                       return HashSet_of_String_as_Set_of_String( results );
+               }
+       `}
 end
 
 redef class NativeString