Produce an octal escape code for \177
[libs/core.git] / examples / syncdir.cpp
1 #include <algorithm>
2 #include <msp/core/application.h>
3 #include <msp/core/getopt.h>
4 #include <msp/fs/dir.h>
5 #include <msp/fs/stat.h>
6 #include <msp/fs/utils.h>
7 #include <msp/io/file.h>
8 #include <msp/io/print.h>
9 #include <msp/strings/format.h>
10
11 using namespace std;
12 using namespace Msp;
13
14 class SyncDir: public RegisteredApplication<SyncDir>
15 {
16 private:
17         FS::Path source;
18         FS::Path destination;
19
20 public:
21         SyncDir(int, char **);
22
23         int main();
24 private:
25         void sync_directory(const FS::Path &, const FS::Path &);
26         void copy_file(const FS::Path &, const FS::Path &);
27 };
28
29 SyncDir::SyncDir(int argc, char **argv)
30 {
31         if(argc<3)
32                 throw usage_error("Missing arguments", format("Usage: %s <source> <destination>", argv[0]));
33
34         source = argv[1];
35         destination = argv[2];
36 }
37
38 int SyncDir::main()
39 {
40         sync_directory(source, destination);
41         return 0;
42 }
43
44 void SyncDir::sync_directory(const FS::Path &src, const FS::Path &dest)
45 {
46         IO::print("Syncing %s to %s\n", src, dest);
47
48         FS::Stat st = FS::stat(dest);
49         if(!st)
50                 FS::mkpath(dest, 0755);
51         else if(!st.is_directory())
52         {
53                 FS::unlink(dest);
54                 FS::mkdir(dest, 0755);
55         }
56
57         vector<string> src_files = FS::list_files(src);
58         for(const string &fn: src_files)
59         {
60                 FS::Stat ss = FS::stat(src/fn);
61                 if(ss.is_directory())
62                         sync_directory(src/fn, dest/fn);
63                 else
64                 {
65                         FS::Stat ds = FS::stat(dest/fn);
66                         if(!ds || ds.get_size()!=ss.get_size())
67                                 copy_file(src/fn, dest/fn);
68                 }
69         }
70
71         for(const string &fn: FS::list_files(dest))
72                 if(find(src_files.begin(), src_files.end(), fn)==src_files.end())
73                 {
74                         IO::print("Removing obsolete %s\n", dest/fn);
75                         if(FS::is_dir(dest/fn))
76                                 FS::rmpath(dest/fn);
77                         else
78                                 FS::unlink(dest/fn);
79                 }
80 }
81
82 void SyncDir::copy_file(const FS::Path &src, const FS::Path &dest)
83 {
84         IO::File in(src.str(), IO::M_READ);
85         IO::File out(dest.str(), IO::M_WRITE);
86         char buf[16384];
87         while(!in.eof())
88         {
89                 unsigned len = in.read(buf, sizeof(buf));
90                 out.write(buf, len);
91         }
92 }