Redesign how tools are run

[builder.git] / source / copy.cpp

diff --git a/source/copy.cpp b/source/copy.cpp
index eea40fcdae6eda6f63dc30f09988736fa88a7fbc..8f6b81d46b8764e82b5a9e8ba7755888e34e8576 100644 (file)
--- a/source/copy.cpp
+++ b/source/copy.cpp
@@ -1,82 +1,74 @@
-/* $Id$
-
-This file is part of builder
-Copyright © 2006-2007 Mikko Rasa, Mikkosoft Productions
-Distributed under the LGPL
-*/
-
-#include <errno.h>
-#include <fstream>
-#include <iostream>
-#include <msp/path/utils.h>
+#ifndef _WIN32
+#include <unistd.h>
+#include <sys/stat.h>
+#endif
+#include <msp/fs/dir.h>
+#include <msp/fs/stat.h>
+#include <msp/fs/utils.h>
+#include <msp/io/file.h>
+#include <msp/io/print.h>
 #include "builder.h"
 #include "copy.h"
-#include "package.h"
+#include "installedfile.h"
+#include "internaltask.h"
 
 using namespace std;
 using namespace Msp;
 
-Copy::Copy(Builder &b, const Package &pkg, const Path::Path &s, const Path::Path &d):
-       InternalAction(b),
-       src(s),
-       dest(d)
+Copy::Copy(Builder &b):
+       Tool(b, "CP")
 {
-       announce(pkg.get_name(), "COPY", dest[-1]);
-       if(builder.get_verbose()>=2)
-               cout<<s<<" -> "<<d<<'\n';
-
-       if(!builder.get_dry_run())
-               worker=new Worker(*this);
+       set_run_internal(_run);
 }
 
-
-Copy::Worker::Worker(Copy &c):
-       copy(c)
+Target *Copy::create_target(const vector<Target *> &sources, const string &arg)
 {
-       launch();
+       FileTarget &file_tgt = dynamic_cast<FileTarget &>(*sources.front());
+       InstalledFile *inst = new InstalledFile(builder, *file_tgt.get_package(), file_tgt, arg);
+       inst->set_tool(*this);
+       return inst;
 }
 
-void Copy::Worker::main()
+bool Copy::_run(const InstalledFile &install)
 {
-       Path::mkpath(copy.dest.subpath(0, copy.dest.size()-1), 0755);
+       const FileTarget &source = install.get_source();
+       const FS::Path &src_path = source.get_path();
+       const FS::Path &dst_path = install.get_path();
 
-       // Remove old file.  Not doing this would cause Bad Stuff when installing libraries.
-       if(unlink(copy.dest.str().c_str())<0 && errno!=ENOENT)
+       try
        {
-               int err=errno;
-               cerr<<"Can't unlink "<<copy.dest<<": "<<strerror(err)<<'\n';
-               done=error=true;
-               return;
-       }
+               IO::File in(src_path.str());
+               IO::File out(dst_path.str(), IO::M_WRITE);
 
-       ifstream in(copy.src.str().c_str());
-       if(!in)
-       {
-               cerr<<"Can't open "<<copy.src<<" for reading\n";
-               done=error=true;
-               return;
+               // Actual transfer loop
+               char buf[16384];
+               while(!in.eof())
+               {
+                       unsigned len = in.read(buf, sizeof(buf));
+                       out.write(buf, len);
+               }
        }
-
-       ofstream out(copy.dest.str().c_str());
-       if(!out)
+       catch(const exception &e)
        {
-               cerr<<"Can't open "<<copy.dest<<" for writing\n";
-               done=error=true;
-               return;
-       }
-
-       // Actual transfer loop
-       char buf[16384];
-       while(!in.eof())
-       {
-               in.read(buf, sizeof(buf));
-               out.write(buf, in.gcount());
+               IO::print(IO::cerr, "%s\n", e.what());
+               return false;
        }
 
+#ifndef _WIN32
        // Preserve file permissions
        struct stat st;
-       Path::stat(copy.src, st);
-       chmod(copy.dest.str().c_str(), st.st_mode&0777);
+       if(stat(src_path.str().c_str(), &st)==0)
+               chmod(dst_path.str().c_str(), st.st_mode&0777);
+
+       const FS::Path &link = install.get_symlink();
+       if(!link.empty())
+       {
+               FS::Path relpath = FS::relative(dst_path, FS::dirname(link));
+               if(FS::exists(link))
+                       FS::unlink(link);
+               symlink(relpath.str().c_str(), link.str().c_str());
+       }
+#endif
 
-       done=true;
+       return true;
 }