X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fcopy.cpp;h=b756e0c037412012a828075735a7b6fb8eb3b679;hb=8932cbfa8cdb0b9edf9894bd4c56980dab0a2f4b;hp=f8c581bd5664ca123f4f657da77883c7dff1003b;hpb=a2adbd9c0a8d7a7567848c4c6bdbf0de6ba32bb1;p=builder.git

diff --git a/source/copy.cpp b/source/copy.cpp
index f8c581b..b756e0c 100644
--- a/source/copy.cpp
+++ b/source/copy.cpp
@@ -1,89 +1,97 @@
-/* $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 <sys/stat.h>
 #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"
 
 using namespace std;
 using namespace Msp;
 
-Copy::Copy(Builder &b, const Package &pkg, const FS::Path &s, const FS::Path &d):
-	InternalAction(b),
-	src(s),
-	dest(d)
+Copy::Copy(Builder &b):
+	Tool(b, "CP")
+{ }
+
+Target *Copy::create_target(const list<Target *> &sources, const string &arg) const
 {
-	announce(pkg.get_name(), "COPY", dest[-1]);
-	if(builder.get_verbose()>=2)
-		cout<<s<<" -> "<<d<<'\n';
+	FileTarget &file_tgt = dynamic_cast<FileTarget &>(*sources.front());
+	const SourcePackage &pkg = dynamic_cast<const SourcePackage &>(*file_tgt.get_package());
+	InstalledFile *inst = new InstalledFile(builder, pkg, file_tgt, arg);
+	inst->set_tool(*this);
+	return inst;
+}
 
-	if(!builder.get_dry_run())
-		worker=new Worker(*this);
+Task *Copy::run(const Target &target) const
+{
+	const InstalledFile &install = dynamic_cast<const InstalledFile &>(target);
+	Worker *worker = new Worker(install);
+	return new InternalTask(worker);
 }
 
 
-Copy::Worker::Worker(Copy &c):
-	copy(c)
+Copy::Worker::Worker(const InstalledFile &t):
+	target(t)
 {
 	launch();
 }
 
 void Copy::Worker::main()
 {
-	FS::mkpath(FS::dirname(copy.dest), 0755);
+	const FileTarget &source = target.get_source();
+	const FS::Path &src_path = source.get_path();
+	const FS::Path &dst_path = target.get_path();
+	FS::mkpath(FS::dirname(dst_path), 0755);
 
-	try
-	{
-		// Remove old file.  Not doing this would cause Bad Stuff when installing libraries.
-		unlink(copy.dest);
-	}
-	catch(const SystemError &e)
+	// Remove old file.  Not doing this would cause Bad Stuff when installing libraries.
+	if(FS::exists(dst_path))
 	{
-		if(e.get_error_code()!=ENOENT)
+		try
+		{
+			unlink(dst_path);
+		}
+		catch(const exception &e)
 		{
-			cerr<<e.what()<<'\n';
-			done=error=true;
+			IO::print(IO::cerr, "%s\n", e.what());
+			status = Task::ERROR;
 			return;
 		}
 	}
 
-	ifstream in(copy.src.str().c_str());
-	if(!in)
+	try
 	{
-		cerr<<"Can't open "<<copy.src<<" for reading\n";
-		done=error=true;
-		return;
-	}
+		IO::File in(src_path.str());
+		IO::File out(dst_path.str(), IO::M_WRITE);
 
-	ofstream out(copy.dest.str().c_str());
-	if(!out)
+		// Actual transfer loop
+		char buf[16384];
+		while(!in.eof())
+		{
+			unsigned len = in.read(buf, sizeof(buf));
+			out.write(buf, len);
+		}
+	}
+	catch(const exception &e)
 	{
-		cerr<<"Can't open "<<copy.dest<<" for writing\n";
-		done=error=true;
+		IO::print(IO::cerr, "%s\n", e.what());
+		status = Task::ERROR;
 		return;
 	}
 
-	// Actual transfer loop
-	char buf[16384];
-	while(!in.eof())
+	// Preserve file permissions
+	struct stat st;
+	if(stat(src_path.str().c_str(), &st)==0)
+		chmod(dst_path.str().c_str(), st.st_mode&0777);
+
+	const FS::Path &link = target.get_symlink();
+	if(!link.empty())
 	{
-		in.read(buf, sizeof(buf));
-		out.write(buf, in.gcount());
+		FS::Path relpath = FS::relative(dst_path, FS::dirname(link));
+		symlink(relpath.str().c_str(), link.str().c_str());
 	}
 
-	// Preserve file permissions
-	struct stat st=FS::stat(copy.src);
-	chmod(copy.dest.str().c_str(), st.st_mode&0777);
-
-	done=true;
+	status = Task::SUCCESS;
 }