1 # This file is part of NIT ( http://www.nitlanguage.org ).
3 # Copyright 2004-2008 Jean Privat <jean@pryen.org>
4 # Copyright 2008 Floréal Morandat <morandat@lirmm.fr>
5 # Copyright 2008 Jean-Sébastien Gélinas <calestar@gmail.com>
7 # This file is free software, which comes along with NIT. This software is
8 # distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
9 # without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
10 # PARTICULAR PURPOSE. You can modify it is you want, provided this header
11 # is kept unaltered, and a notification of the changes is added.
12 # You are allowed to redistribute it and sell it, alone or is a part of
15 # File manipulations (create, read, write, etc.)
26 #include <sys/types.h>
34 # Print `objects` on the standard output (`stdout`).
35 protected fun printn
(objects
: Object...)
37 stdout
.write
(objects
.to_s
)
40 # Print an `object` on the standard output (`stdout`) and add a newline.
41 protected fun print
(object
: Object)
43 stdout
.write
(object
.to_s
)
47 # Read a character from the standard input (`stdin`).
48 protected fun getc
: Char
50 return stdin
.read_char
.ascii
53 # Read a line from the standard input (`stdin`).
54 protected fun gets
: String
56 return stdin
.read_line
59 # Return the working (current) directory
60 protected fun getcwd
: String do return file_getcwd
.to_s
61 private fun file_getcwd
: NativeString is extern "string_NativeString_NativeString_file_getcwd_0"
64 # File Abstract Stream
65 abstract class FStream
67 # The path of the file.
68 readable var _path
: nullable String = null
71 var _file
: nullable NativeFile = null
73 fun file_stat
: FileStat
74 do return _file
.file_stat
end
83 # Open the same file again.
84 # The original path is reused, therefore the reopened file can be a different file.
88 _file
= new NativeFile.io_open_read
(_path
.to_cstring
)
96 var i
= _file
.io_close
100 redef fun fill_buffer
102 var nb
= _file
.io_read
(_buffer
._items
, _buffer
._capacity
)
112 redef readable var _end_reached
: Bool = false
114 # Open the file at `path` for reading.
115 init open
(path
: String)
119 _file
= new NativeFile.io_open_read
(_path
.to_cstring
)
120 assert cant_open_file
: _file
!= null
124 private init without_file
do end
135 write_native
(s
.to_cstring
, s
.length
)
138 redef fun is_writable
do return _writable
142 var i
= _file
.io_close
146 # Is the file open in write mode
149 # Write `len` bytes from `native`.
150 private fun write_native
(native
: NativeString, len
: Int)
153 var err
= _file
.io_write
(native
, len
)
156 printn
("Problem in writing : ", err
, " ", len
, "\n")
160 # Open the file at `path` for writing.
161 init open
(path
: String)
163 _file
= new NativeFile.io_open_write
(path
.to_cstring
)
164 assert cant_open_file
: _file
!= null
170 private init without_file
do end
173 ###############################################################################
178 _file
= new NativeFile.native_stdin
183 # Is these something to read? (non blocking)
184 # FIXME: should be generalized
185 fun poll_in
: Bool is extern "file_stdin_poll_in"
191 _file
= new NativeFile.native_stdout
192 _path
= "/dev/stdout"
200 _file
= new NativeFile.native_stderr
201 _path
= "/dev/stderr"
206 ###############################################################################
208 redef class Streamable
209 # Like `write_to` but take care of creating the file
210 fun write_to_file
(filepath
: String)
212 var stream
= new OFStream.open
(filepath
)
219 # return true if a file with this names exists
220 fun file_exists
: Bool do return to_cstring
.file_exists
222 fun file_stat
: FileStat do return to_cstring
.file_stat
223 fun file_lstat
: FileStat do return to_cstring
.file_lstat
225 # Remove a file, return true if success
226 fun file_delete
: Bool do return to_cstring
.file_delete
228 # Copy content of file at `self` to `dest`
229 fun file_copy_to
(dest
: String)
231 var input
= new IFStream.open
(self)
232 var output
= new OFStream.open
(dest
)
234 while not input
.eof
do
235 var buffer
= input
.read
(1024)
243 # remove the trailing extension "ext"
244 fun strip_extension
(ext
: String): String
246 if has_suffix
(ext
) then
247 return substring
(0, length
- ext
.length
)
252 # Extract the basename of a path and remove the extension
254 # assert "/path/to/a_file.ext".basename(".ext") == "a_file"
255 # assert "path/to/a_file.ext".basename(".ext") == "a_file"
256 # assert "path/to".basename(".ext") == "to"
257 # assert "path/to/".basename(".ext") == "to"
258 # assert "path".basename("") == "path"
259 # assert "/path".basename("") == "path"
260 # assert "/".basename("") == "/"
261 # assert "".basename("") == ""
262 fun basename
(ext
: String): String
264 var l
= _length
- 1 # Index of the last char
265 while l
> 0 and self.chars
[l
] == '/' do l
-= 1 # remove all trailing `/`
266 if l
== 0 then return "/"
267 var pos
= last_index_of_from
('/', l
)
270 n
= substring
(pos
+1, l-pos
)
272 return n
.strip_extension
(ext
)
275 # Extract the dirname of a path
277 # assert "/path/to/a_file.ext".dirname == "/path/to"
278 # assert "path/to/a_file.ext".dirname == "path/to"
279 # assert "path/to".dirname == "path"
280 # assert "path/to/".dirname == "path"
281 # assert "path".dirname == "."
282 # assert "/path".dirname == "/"
283 # assert "/".dirname == "/"
284 # assert "".dirname == "."
287 var l
= _length
- 1 # Index of the last char
288 while l
> 0 and self.chars
[l
] == '/' do l
-= 1 # remove all trailing `/`
289 var pos
= last_index_of_from
('/', l
)
291 return substring
(0, pos
)
292 else if pos
== 0 then
299 # Return the canonicalized absolute pathname (see POSIX function `realpath`)
300 fun realpath
: String do
301 var cs
= to_cstring
.file_realpath
302 var res
= cs
.to_s_with_copy
303 # cs.free_malloc # FIXME memory leak
307 # Simplify a file path by remove useless ".", removing "//", and resolving ".."
308 # ".." are not resolved if they start the path
309 # starting "/" is not removed
310 # trainling "/" is removed
312 # Note that the method only wonrk on the string:
313 # * no I/O access is performed
314 # * the validity of the path is not checked
316 # assert "some/./complex/../../path/from/../to/a////file//".simplify_path == "path/to/a/file"
317 # assert "../dir/file".simplify_path == "../dir/file"
318 # assert "dir/../../".simplify_path == ".."
319 # assert "dir/..".simplify_path == "."
320 # assert "//absolute//path/".simplify_path == "/absolute/path"
321 fun simplify_path
: String
323 var a
= self.split_with
("/")
324 var a2
= new Array[String]
326 if x
== "." then continue
327 if x
== "" and not a2
.is_empty
then continue
328 if x
== ".." and not a2
.is_empty
and a2
.last
!= ".." then
334 if a2
.is_empty
then return "."
338 # Correctly join two path using the directory separator.
340 # Using a standard "{self}/{path}" does not work when `self` is the empty string.
341 # This method ensure that the join is valid.
343 # assert "hello".join_path("world") == "hello/world"
344 # assert "hel/lo".join_path("wor/ld") == "hel/lo/wor/ld"
345 # assert "".join_path("world") == "world"
346 # assert "/hello".join_path("/world") == "/world"
348 # Note: you may want to use `simplify_path` on the result
350 # Note: I you want to join a great number of path, you can write
352 # [p1, p2, p3, p4].join("/")
353 fun join_path
(path
: String): String
355 if path
.is_empty
then return self
356 if self.is_empty
then return path
357 if path
.chars
[0] == '/' then return path
358 return "{self}/{path}"
361 # Create a directory (and all intermediate directories if needed)
364 var dirs
= self.split_with
("/")
365 var path
= new Buffer
366 if dirs
.is_empty
then return
367 if dirs
[0].is_empty
then
368 # it was a starting /
372 if d
.is_empty
then continue
375 path
.to_s
.to_cstring
.file_mkdir
379 # Change the current working directory
382 # assert getcwd == "/etc"
384 # assert getcwd == "/"
387 fun chdir
do to_cstring
.file_chdir
389 # Return right-most extension (without the dot)
390 fun file_extension
: nullable String
392 var last_slash
= last_index_of
('.')
393 if last_slash
>= 0 then
394 return substring
( last_slash
+1, length
)
400 # returns files contained within the directory represented by self
401 fun files
: Set[ String ] is extern import HashSet[String], HashSet[String].add
, NativeString.to_s
, String.to_cstring
, HashSet[String].as(Set[String]) `{
405 dir_path = String_to_cstring( recv );
406 if ((dir = opendir(dir_path)) == NULL)
413 HashSet_of_String results;
417 results = new_HashSet_of_String();
419 while ( ( de = readdir( dir ) ) != NULL )
420 if ( strcmp( de->d_name, ".." ) != 0 &&
421 strcmp( de->d_name, "." ) != 0 )
423 file_name = NativeString_to_s( strdup( de->d_name ) );
424 HashSet_of_String_add( results, file_name );
428 return HashSet_of_String_as_Set_of_String( results );
433 redef class NativeString
434 private fun file_exists
: Bool is extern "string_NativeString_NativeString_file_exists_0"
435 private fun file_stat
: FileStat is extern "string_NativeString_NativeString_file_stat_0"
436 private fun file_lstat
: FileStat `{
437 struct stat* stat_element;
439 stat_element = malloc(sizeof(struct stat));
440 res = lstat(recv, stat_element);
441 if (res == -1) return NULL;
444 private fun file_mkdir
: Bool is extern "string_NativeString_NativeString_file_mkdir_0"
445 private fun file_delete
: Bool is extern "string_NativeString_NativeString_file_delete_0"
446 private fun file_chdir
is extern "string_NativeString_NativeString_file_chdir_0"
447 private fun file_realpath
: NativeString is extern "file_NativeString_realpath"
450 # This class is system dependent ... must reify the vfs
451 extern class FileStat `{ struct stat * `}
452 # Returns the permission bits of file
453 fun mode: Int is extern "file_FileStat_FileStat_mode_0"
454 # Returns the last access time
455 fun atime: Int is extern "file_FileStat_FileStat_atime_0"
456 # Returns the last status change time
457 fun ctime: Int is extern "file_FileStat_FileStat_ctime_0"
458 # Returns the last modification time
459 fun mtime: Int is extern "file_FileStat_FileStat_mtime_0"
461 fun size: Int is extern "file_FileStat_FileStat_size_0"
463 # Returns true if it is a regular file (not a device file, pipe, sockect, ...)
464 fun is_reg: Bool `{ return S_ISREG(recv->st_mode); `}
465 # Returns true if it is a directory
466 fun is_dir
: Bool `{ return S_ISDIR(recv->st_mode); `}
467 # Returns true if it is a character device
468 fun is_chr: Bool `{ return S_ISCHR(recv->st_mode); `}
469 # Returns true if it is a block device
470 fun is_blk
: Bool `{ return S_ISBLK(recv->st_mode); `}
471 # Returns true if the type is fifo
472 fun is_fifo: Bool `{ return S_ISFIFO(recv->st_mode); `}
473 # Returns true if the type is a link
474 fun is_lnk
: Bool `{ return S_ISLNK(recv->st_mode); `}
475 # Returns true if the type is a socket
476 fun is_sock: Bool `{ return S_ISSOCK(recv->st_mode); `}
479 # Instance of this class are standard FILE * pointers
480 private extern NativeFile `{ FILE* `}
481 fun io_read(buf: NativeString, len: Int): Int is extern "file_NativeFile_NativeFile_io_read_2"
482 fun io_write(buf: NativeString, len: Int): Int is extern "file_NativeFile_NativeFile_io_write_2"
483 fun io_close: Int is extern "file_NativeFile_NativeFile_io_close_0"
484 fun file_stat: FileStat is extern "file_NativeFile_NativeFile_file_stat_0"
486 new io_open_read(path: NativeString) is extern "file_NativeFileCapable_NativeFileCapable_io_open_read_1"
487 new io_open_write(path: NativeString) is extern "file_NativeFileCapable_NativeFileCapable_io_open_write_1"
488 new native_stdin is extern "file_NativeFileCapable_NativeFileCapable_native_stdin_0"
489 new native_stdout is extern "file_NativeFileCapable_NativeFileCapable_native_stdout_0"
490 new native_stderr is extern "file_NativeFileCapable_NativeFileCapable_native_stderr_0"
494 fun stdin: Stdin do return once new Stdin
497 fun stdout: OFStream do return once new Stdout
499 # Standard output for error.
500 fun stderr: OFStream do return once new Stderr