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>
31 # File Abstract Stream
32 abstract class FStream
34 # The path of the file.
35 var path
: nullable String = null
38 var _file
: nullable NativeFile = null
40 fun file_stat
: FileStat
41 do return _file
.file_stat
end
50 # Open the same file again.
51 # The original path is reused, therefore the reopened file can be a different file.
55 _file
= new NativeFile.io_open_read
(path
.to_cstring
)
63 var i
= _file
.io_close
69 var nb
= _file
.io_read
(_buffer
.items
, _buffer
.capacity
)
79 redef var end_reached
: Bool = false
81 # Open the file at `path` for reading.
82 init open
(path
: String)
86 _file
= new NativeFile.io_open_read
(path
.to_cstring
)
87 assert not _file
.address_is_null
else
88 print
"Error: Opening file at '{path}' failed with '{sys.errno.strerror}'"
93 private init without_file
do end
104 if s
isa FlatText then
105 write_native
(s
.to_cstring
, s
.length
)
107 for i
in s
.substrings
do write_native
(i
.to_cstring
, i
.length
)
111 redef fun is_writable
do return _writable
115 var i
= _file
.io_close
119 # Is the file open in write mode
122 # Write `len` bytes from `native`.
123 private fun write_native
(native
: NativeString, len
: Int)
126 var err
= _file
.io_write
(native
, len
)
129 printn
("Problem in writing : ", err
, " ", len
, "\n")
133 # Open the file at `path` for writing.
134 init open
(path
: String)
136 _file
= new NativeFile.io_open_write
(path
.to_cstring
)
137 assert not _file
.address_is_null
else
138 print
"Error: Opening file at '{path}' failed with '{sys.errno.strerror}'"
145 private init without_file
do end
148 ###############################################################################
152 super PollableIStream
155 _file
= new NativeFile.native_stdin
160 redef fun poll_in
: Bool is extern "file_stdin_poll_in"
166 _file
= new NativeFile.native_stdout
175 _file
= new NativeFile.native_stderr
181 ###############################################################################
183 redef class Streamable
184 # Like `write_to` but take care of creating the file
185 fun write_to_file
(filepath
: String)
187 var stream
= new OFStream.open
(filepath
)
194 # return true if a file with this names exists
195 fun file_exists
: Bool do return to_cstring
.file_exists
197 # The status of a file. see POSIX stat(2).
198 fun file_stat
: FileStat do return to_cstring
.file_stat
200 # The status of a file or of a symlink. see POSIX lstat(2).
201 fun file_lstat
: FileStat do return to_cstring
.file_lstat
203 # Remove a file, return true if success
204 fun file_delete
: Bool do return to_cstring
.file_delete
206 # Copy content of file at `self` to `dest`
207 fun file_copy_to
(dest
: String)
209 var input
= new IFStream.open
(self)
210 var output
= new OFStream.open
(dest
)
212 while not input
.eof
do
213 var buffer
= input
.read
(1024)
221 # Remove the trailing extension `ext`.
223 # `ext` usually starts with a dot but could be anything.
225 # assert "file.txt".strip_extension(".txt") == "file"
226 # assert "file.txt".strip_extension("le.txt") == "fi"
227 # assert "file.txt".strip_extension("xt") == "file.t"
229 # if `ext` is not present, `self` is returned unmodified.
231 # assert "file.txt".strip_extension(".tar.gz") == "file.txt"
232 fun strip_extension
(ext
: String): String
234 if has_suffix
(ext
) then
235 return substring
(0, length
- ext
.length
)
240 # Extract the basename of a path and remove the extension
242 # assert "/path/to/a_file.ext".basename(".ext") == "a_file"
243 # assert "path/to/a_file.ext".basename(".ext") == "a_file"
244 # assert "path/to".basename(".ext") == "to"
245 # assert "path/to/".basename(".ext") == "to"
246 # assert "path".basename("") == "path"
247 # assert "/path".basename("") == "path"
248 # assert "/".basename("") == "/"
249 # assert "".basename("") == ""
250 fun basename
(ext
: String): String
252 var l
= length
- 1 # Index of the last char
253 while l
> 0 and self.chars
[l
] == '/' do l
-= 1 # remove all trailing `/`
254 if l
== 0 then return "/"
255 var pos
= chars
.last_index_of_from
('/', l
)
258 n
= substring
(pos
+1, l-pos
)
260 return n
.strip_extension
(ext
)
263 # Extract the dirname of a path
265 # assert "/path/to/a_file.ext".dirname == "/path/to"
266 # assert "path/to/a_file.ext".dirname == "path/to"
267 # assert "path/to".dirname == "path"
268 # assert "path/to/".dirname == "path"
269 # assert "path".dirname == "."
270 # assert "/path".dirname == "/"
271 # assert "/".dirname == "/"
272 # assert "".dirname == "."
275 var l
= length
- 1 # Index of the last char
276 while l
> 0 and self.chars
[l
] == '/' do l
-= 1 # remove all trailing `/`
277 var pos
= chars
.last_index_of_from
('/', l
)
279 return substring
(0, pos
)
280 else if pos
== 0 then
287 # Return the canonicalized absolute pathname (see POSIX function `realpath`)
288 fun realpath
: String do
289 var cs
= to_cstring
.file_realpath
290 var res
= cs
.to_s_with_copy
291 # cs.free_malloc # FIXME memory leak
295 # Simplify a file path by remove useless ".", removing "//", and resolving ".."
296 # ".." are not resolved if they start the path
297 # starting "/" is not removed
298 # trainling "/" is removed
300 # Note that the method only wonrk on the string:
301 # * no I/O access is performed
302 # * the validity of the path is not checked
304 # assert "some/./complex/../../path/from/../to/a////file//".simplify_path == "path/to/a/file"
305 # assert "../dir/file".simplify_path == "../dir/file"
306 # assert "dir/../../".simplify_path == ".."
307 # assert "dir/..".simplify_path == "."
308 # assert "//absolute//path/".simplify_path == "/absolute/path"
309 # assert "//absolute//../".simplify_path == "/"
310 fun simplify_path
: String
312 var a
= self.split_with
("/")
313 var a2
= new Array[String]
315 if x
== "." then continue
316 if x
== "" and not a2
.is_empty
then continue
317 if x
== ".." and not a2
.is_empty
and a2
.last
!= ".." then
323 if a2
.is_empty
then return "."
324 if a2
.length
== 1 and a2
.first
== "" then return "/"
328 # Correctly join two path using the directory separator.
330 # Using a standard "{self}/{path}" does not work when `self` is the empty string.
331 # This method ensure that the join is valid.
333 # assert "hello".join_path("world") == "hello/world"
334 # assert "hel/lo".join_path("wor/ld") == "hel/lo/wor/ld"
335 # assert "".join_path("world") == "world"
336 # assert "/hello".join_path("/world") == "/world"
338 # Note: you may want to use `simplify_path` on the result
340 # Note: I you want to join a great number of path, you can write
342 # [p1, p2, p3, p4].join("/")
343 fun join_path
(path
: String): String
345 if path
.is_empty
then return self
346 if self.is_empty
then return path
347 if path
.chars
[0] == '/' then return path
348 return "{self}/{path}"
351 # Create a directory (and all intermediate directories if needed)
354 var dirs
= self.split_with
("/")
355 var path
= new FlatBuffer
356 if dirs
.is_empty
then return
357 if dirs
[0].is_empty
then
358 # it was a starting /
362 if d
.is_empty
then continue
365 path
.to_s
.to_cstring
.file_mkdir
369 # Delete a directory and all of its content, return `true` on success
371 # Does not go through symbolic links and may get stuck in a cycle if there
372 # is a cycle in the filesystem.
376 for file
in self.files
do
377 var file_path
= self.join_path
(file
)
378 var stat
= file_path
.file_lstat
380 ok
= file_path
.rmdir
and ok
382 ok
= file_path
.file_delete
and ok
387 # Delete the directory itself
388 if ok
then to_cstring
.rmdir
393 # Change the current working directory
396 # assert getcwd == "/etc"
398 # assert getcwd == "/"
401 fun chdir
do to_cstring
.file_chdir
403 # Return right-most extension (without the dot)
405 # Only the last extension is returned.
406 # There is no special case for combined extensions.
408 # assert "file.txt".file_extension == "txt"
409 # assert "file.tar.gz".file_extension == "gz"
411 # For file without extension, `null` is returned.
412 # Hoever, for trailing dot, `""` is returned.
414 # assert "file".file_extension == null
415 # assert "file.".file_extension == ""
417 # The starting dot of hidden files is never considered.
419 # assert ".file.txt".file_extension == "txt"
420 # assert ".file".file_extension == null
421 fun file_extension
: nullable String
423 var last_slash
= chars
.last_index_of
('.')
424 if last_slash
> 0 then
425 return substring
( last_slash
+1, length
)
431 # returns files contained within the directory represented by self
432 fun files
: Set[ String ] is extern import HashSet[String], HashSet[String].add
, NativeString.to_s
, String.to_cstring
, HashSet[String].as(Set[String]) `{
436 dir_path = String_to_cstring( recv );
437 if ((dir = opendir(dir_path)) == NULL)
444 HashSet_of_String results;
448 results = new_HashSet_of_String();
450 while ( ( de = readdir( dir ) ) != NULL )
451 if ( strcmp( de->d_name, ".." ) != 0 &&
452 strcmp( de->d_name, "." ) != 0 )
454 file_name = NativeString_to_s( strdup( de->d_name ) );
455 HashSet_of_String_add( results, file_name );
459 return HashSet_of_String_as_Set_of_String( results );
464 redef class NativeString
465 private fun file_exists
: Bool is extern "string_NativeString_NativeString_file_exists_0"
466 private fun file_stat
: FileStat is extern "string_NativeString_NativeString_file_stat_0"
467 private fun file_lstat
: FileStat `{
468 struct stat* stat_element;
470 stat_element = malloc(sizeof(struct stat));
471 res = lstat(recv, stat_element);
472 if (res == -1) return NULL;
475 private fun file_mkdir
: Bool is extern "string_NativeString_NativeString_file_mkdir_0"
476 private fun rmdir
: Bool `{ return rmdir(recv); `}
477 private fun file_delete: Bool is extern "string_NativeString_NativeString_file_delete_0"
478 private fun file_chdir is extern "string_NativeString_NativeString_file_chdir_0"
479 private fun file_realpath: NativeString is extern "file_NativeString_realpath"
482 # This class is system dependent ... must reify the vfs
483 extern class FileStat `{ struct stat * `}
484 # Returns the permission bits of file
485 fun mode
: Int is extern "file_FileStat_FileStat_mode_0"
486 # Returns the last access time
487 fun atime
: Int is extern "file_FileStat_FileStat_atime_0"
488 # Returns the last status change time
489 fun ctime
: Int is extern "file_FileStat_FileStat_ctime_0"
490 # Returns the last modification time
491 fun mtime
: Int is extern "file_FileStat_FileStat_mtime_0"
493 fun size
: Int is extern "file_FileStat_FileStat_size_0"
495 # Returns true if it is a regular file (not a device file, pipe, sockect, ...)
496 fun is_reg
: Bool `{ return S_ISREG(recv->st_mode); `}
497 # Returns true if it is a directory
498 fun is_dir: Bool `{ return S_ISDIR(recv->st_mode); `}
499 # Returns true if it is a character device
500 fun is_chr
: Bool `{ return S_ISCHR(recv->st_mode); `}
501 # Returns true if it is a block device
502 fun is_blk: Bool `{ return S_ISBLK(recv->st_mode); `}
503 # Returns true if the type is fifo
504 fun is_fifo
: Bool `{ return S_ISFIFO(recv->st_mode); `}
505 # Returns true if the type is a link
506 fun is_lnk: Bool `{ return S_ISLNK(recv->st_mode); `}
507 # Returns true if the type is a socket
508 fun is_sock
: Bool `{ return S_ISSOCK(recv->st_mode); `}
511 # Instance of this class are standard FILE * pointers
512 private extern class NativeFile `{ FILE* `}
513 fun io_read
(buf
: NativeString, len
: Int): Int is extern "file_NativeFile_NativeFile_io_read_2"
514 fun io_write
(buf
: NativeString, len
: Int): Int is extern "file_NativeFile_NativeFile_io_write_2"
515 fun io_close
: Int is extern "file_NativeFile_NativeFile_io_close_0"
516 fun file_stat
: FileStat is extern "file_NativeFile_NativeFile_file_stat_0"
518 new io_open_read
(path
: NativeString) is extern "file_NativeFileCapable_NativeFileCapable_io_open_read_1"
519 new io_open_write
(path
: NativeString) is extern "file_NativeFileCapable_NativeFileCapable_io_open_write_1"
520 new native_stdin
is extern "file_NativeFileCapable_NativeFileCapable_native_stdin_0"
521 new native_stdout
is extern "file_NativeFileCapable_NativeFileCapable_native_stdout_0"
522 new native_stderr
is extern "file_NativeFileCapable_NativeFileCapable_native_stderr_0"
528 var stdin
: PollableIStream protected writable = new Stdin
531 var stdout
: OStream protected writable = new Stdout
533 # Standard output for errors
534 var stderr
: OStream protected writable = new Stderr
538 # Print `objects` on the standard output (`stdout`).
539 protected fun printn
(objects
: Object...)
541 sys
.stdout
.write
(objects
.to_s
)
544 # Print an `object` on the standard output (`stdout`) and add a newline.
545 protected fun print
(object
: Object)
547 sys
.stdout
.write
(object
.to_s
)
548 sys
.stdout
.write
("\n")
551 # Read a character from the standard input (`stdin`).
552 protected fun getc
: Char
554 return sys
.stdin
.read_char
.ascii
557 # Read a line from the standard input (`stdin`).
558 protected fun gets
: String
560 return sys
.stdin
.read_line
563 # Return the working (current) directory
564 protected fun getcwd
: String do return file_getcwd
.to_s
565 private fun file_getcwd
: NativeString is extern "string_NativeString_NativeString_file_getcwd_0"