rlm@0: (ns coderloop.export-files
rlm@0:   (:use clojure.java.io)
rlm@0:   (:use [clojure.contrib [duck-streams :only [file-str]]])
rlm@0:   (:use [rlm
rlm@0: 	 [classpath-utils :only [classpath]]
rlm@0: 	 [shell-write :only [sw]]])
rlm@0:   (:require clojure.string)
rlm@0:   (:import java.io.File))
rlm@0: 
rlm@0: (import '[org.apache.commons.io FileUtils])
rlm@0: 
rlm@0: ;;(use '[telmo.rlm [dist :only [dist]]])
rlm@0: 
rlm@0: (def *nailgun* true)
rlm@0: 
rlm@0: ;;; put standard base-path on the classpath
rlm@0: (def standard-base-directory (file-str "~/.clojure-exports/"))
rlm@0: 
rlm@0: (defn better-export-files [namespace name]
rlm@0:     (binding [*compile-path* (.getPath standard-base-directory)]
rlm@0:     ;;clear out classes -- prepare temp directory
rlm@0:     (FileUtils/forceMkdir standard-base-directory)
rlm@0:     (FileUtils/deleteDirectory standard-base-directory)
rlm@0:     (FileUtils/forceMkdir standard-base-directory)      
rlm@0:       ;; make classes
rlm@0:     (compile (symbol (str namespace)))
rlm@0:     
rlm@0:     ;; make jar
rlm@0: ;;    (jar {:jarfile jar-name :basedir temp-class-dir})
rlm@0:     
rlm@0:     ;; move jar
rlm@0: ;;    (FileUtils/copyFileToDirectory jar-name target-lib-dir)
rlm@0:     ))
rlm@0: 
rlm@0: (defmulti source-location class)
rlm@0: 
rlm@0: (defmethod source-location clojure.lang.Var [sym]
rlm@0: 	   (let [source? (:file (meta sym))]
rlm@0: 	     (if source?
rlm@0: 	       (.getResource (clojure.lang.RT/baseLoader) source?)
rlm@0: 	       nil)))
rlm@0: 
rlm@0: (defmethod source-location java.lang.Class [sym]
rlm@0: 	   (let [source? (.getCodeSource (.getProtectionDomain sym))]
rlm@0: 	     (if source?
rlm@0: 	       (.getLocation source?)
rlm@0: 	       nil)))
rlm@0: 
rlm@0: (defn all-dependent-namespaces [namespace]
rlm@0:   (set
rlm@0:    (concat
rlm@0:     [namespace]
rlm@0:     
rlm@0:     (vals (ns-aliases namespace))
rlm@0: 
rlm@0:     (map #(.ns %)
rlm@0: 	 (filter #(= (class %) clojure.lang.Var)
rlm@0: 		 (concat 
rlm@0: 		  
rlm@0: 		  (vals (ns-map namespace))
rlm@0: 		  (vals (ns-refers namespace))
rlm@0: 		  
rlm@0: 		  ))))))
rlm@0: 
rlm@0: (defn deep-dependent-namespaces [namespace-set]
rlm@0:   (let [new-namespaces (set (mapcat all-dependent-namespaces namespace-set))]
rlm@0:     ;;(println (count new-namespaces))
rlm@0:     (if (= new-namespaces namespace-set)
rlm@0:       namespace-set
rlm@0:       (recur new-namespaces))))
rlm@0: 
rlm@0: 
rlm@0: (defn dependencies-url
rlm@0:   "returns all of the files necessary to succesfully run the namespace."
rlm@0:   [namespace]
rlm@0:   (set
rlm@0:    (remove nil?
rlm@0: 	   (map source-location
rlm@0: 		(map second
rlm@0: 		     (mapcat ns-map (deep-dependent-namespaces #{namespace})))))))
rlm@0: 
rlm@0: 
rlm@0: (defn trans-print [x] (println x) x) 
rlm@0: 
rlm@0: (defn dependencies-file [namespace]
rlm@0:   (map file-str
rlm@0:        (set
rlm@0: 	(map #(.replaceFirst % "file:" "")
rlm@0: 	(map  #(if (.contains % ".jar!/")
rlm@0: 		 (clojure.string/replace % #"\.jar.*" ".jar")
rlm@0: 		 %)
rlm@0: 	      (map #(.getFile %) (dependencies-url namespace)))))))
rlm@0: 
rlm@0: 
rlm@0: 
rlm@0: (defn classpath-files []
rlm@0:   (map file-str (clojure.string/split (classpath) #":")))
rlm@0: 
rlm@0: ;;Every file that comes back from dependencies-file is also on the classpath.
rlm@0: ;;In order to recreate the right project structure, we need to copy the files 
rlm@0: ;;with the appropiate classpath nesting.
rlm@0: 
rlm@0: (defn bifurcate
rlm@0:   "split a collection between f and not f"
rlm@0:   [f coll]
rlm@0:   (list
rlm@0:    (filter f coll)
rlm@0:    (filter (comp not f) coll)))
rlm@0: 
rlm@0: (defn jar? [#^java.io.File f]
rlm@0:   (re-matches  #".*\.jar$" (.getCanonicalPath f)))
rlm@0: 
rlm@0: (defn contains-file? [#^java.io.File parent #^java.io.File child]
rlm@0:   (let [new-child (.getParentFile child)]
rlm@0:     (cond (nil? new-child) false
rlm@0: 	  (= new-child parent) true
rlm@0: 	  true (recur parent new-child))))
rlm@0: 
rlm@0: (defn destination [base-path-list #^java.io.File current-file #^java.io.File base-destination]
rlm@0:   (let [base-path (last (filter #(contains-file? % current-file) base-path-list))]
rlm@0:     (file-str
rlm@0:      (.replaceFirst (.getCanonicalPath current-file)
rlm@0: 		    (.getCanonicalPath base-path)
rlm@0: 		    (.getCanonicalPath base-destination)))))
rlm@0: 
rlm@0: (defn export-dependencies
rlm@0:   ([namespace #^java.io.File target src* lib*]
rlm@0:      (let [[jars sources] (bifurcate jar? (dependencies-file namespace))
rlm@0: 	   jars (if *nailgun*
rlm@0: 		  (conj jars (file-str "~/roBin/nailgun-0.7.1/nailgun-0.7.1.jar"))
rlm@0: 		  jars)
rlm@0: 	   
rlm@0: 	   lib (file-str (str (.getCanonicalPath target) File/separatorChar lib*))
rlm@0: 	   src (file-str (str (.getCanonicalPath target) File/separatorChar src*))]
rlm@0:        (if *nailgun*
rlm@0: 	 (do
rlm@0: 	   (FileUtils/copyFileToDirectory (file-str "~/roBin/nailgun-0.7.1/ng") lib)
rlm@0: 	   (sw (str "chmod +x " (.getCanonicalPath target) File/separatorChar lib* "/ng"))))
rlm@0:        (dorun (map #(FileUtils/copyFileToDirectory % lib) jars))
rlm@0:        (dorun (map #(FileUtils/copyFile % (destination (classpath-files) % src)) sources))))
rlm@0:   
rlm@0:   ([namespace #^java.io.File target]
rlm@0:      (export-dependencies namespace target "src" "lib")))
rlm@0: 
rlm@0: (defn run-script-text
rlm@0:   ([namespace]
rlm@0:      (run-script-text namespace "src" "lib"))
rlm@0:   ([namespace src lib] 
rlm@0:      (if *nailgun*
rlm@0:        (str
rlm@0: 	
rlm@0: 	"#!/usr/bin/perl
rlm@0: 
rlm@0: if (`./"lib"/ng ng-alias 2>&1 1>/dev/null` eq \"connect: Connection refused\\n\"){
rlm@0: 	
rlm@0: 	$nailgun_init = 
rlm@0: 	\"java -Xmn500M -Xms2000M -Xmx2000M \" . 
rlm@0: 	\"-classpath ./"src"/:"lib"/*\" . \" -server \". 
rlm@0: 	\"com.martiansoftware.nailgun.NGServer \". 
rlm@0: 	\" >/dev/null 2>/dev/null &\";
rlm@0: 	`$nailgun_init`;
rlm@0: 	}\n\n"
rlm@0: 	
rlm@0: 	
rlm@0: 	"while (`./"lib"/ng ng-alias 2>&1 1>/dev/null` eq \"connect: Connection refused\\n\"){
rlm@0:     }\n\n"
rlm@0: 	
rlm@0: 	"$command = "
rlm@0: 	
rlm@0: 	
rlm@0: 	"\"./"lib"/ng clojure.main "
rlm@0: 	"./"src"/"
rlm@0: 	(.replace (.replace (str *ns*) \. File/separatorChar) \- \_)
rlm@0: 	".clj "
rlm@0: 	" @ARGV\";\n"
rlm@0: 	
rlm@0: 	"print `$command`;\n")
rlm@0:        (str
rlm@0: 	"#!/bin/bash\n"
rlm@0: 	"java -Xmn500M -Xms2000M -Xmx2000M -server -cp ./"lib"/*:./"src" clojure.main "
rlm@0: 	"./"src"/"
rlm@0: 	(.replace (.replace (str *ns*) \. File/separatorChar) \- \_)
rlm@0: 	".clj "
rlm@0: 	" $@\n"))))
rlm@0: 
rlm@0: 
rlm@0: 
rlm@0: (defn make-run-script
rlm@0:   ([namespace base-dir name]
rlm@0:      (make-run-script namespace base-dir name "src" "lib"))
rlm@0:   ([namespace base-dir name src lib]
rlm@0:   (let [w (clojure.java.io/writer
rlm@0: 	   (str (.getCanonicalPath base-dir) File/separatorChar name))]
rlm@0:     (.write w (run-script-text namespace src lib))
rlm@0:     (.close w))
rlm@0:   (let [f (file-str  (str (.getCanonicalPath base-dir) File/separatorChar name))]
rlm@0:     (.setExecutable f true))))
rlm@0: 
rlm@0: 
rlm@0: (defn bzip-folder [#^java.io.File destination #^java.io.File directory]
rlm@0:   (apply (partial sw "tar" "-cvjf" (.getCanonicalPath destination))
rlm@0: 
rlm@0: 	  (concat
rlm@0: 	   (rest
rlm@0: 	    (map #(.replaceFirst
rlm@0: 		   %
rlm@0: 		   (str (.getCanonicalPath directory) File/separatorChar ) "")
rlm@0: 		 (map str (file-seq directory))))
rlm@0: 	    [:dir (.getCanonicalPath directory)])))
rlm@0: 
rlm@0: 
rlm@0: (defn bzip-export-files
rlm@0:   ([directory to name postfix]
rlm@0:      (bzip-export-files directory to name postfix "src" "lib"))
rlm@0: 
rlm@0:   ([directory to name postfix src lib]
rlm@0:      (apply (partial sw "tar" "-cvjf"
rlm@0: 		     (str (.getCanonicalPath to) File/separatorChar (str name postfix)))
rlm@0: 	    (concat
rlm@0: 	     [name src lib]
rlm@0: 	     [:dir (.getCanonicalPath directory)]))))
rlm@0: 
rlm@0: (defn export-archive
rlm@0:   ([name]
rlm@0:      (export-archive *ns* name))
rlm@0: 
rlm@0:   ([namespace name nailgun?]
rlm@0:      (binding [*nailgun* nailgun?]
rlm@0:        (export-archive namespace name)))
rlm@0:   
rlm@0:   ([namespace name]
rlm@0:      (FileUtils/forceMkdir standard-base-directory)
rlm@0:      (FileUtils/forceDelete standard-base-directory)
rlm@0:      (FileUtils/forceMkdir standard-base-directory)
rlm@0:      (let [new-dir (file-str (str
rlm@0: 			      (.getCanonicalPath standard-base-directory)
rlm@0: 			      (File/separatorChar)
rlm@0: 			      name))]
rlm@0:        (export-dependencies namespace new-dir)
rlm@0:        (make-run-script namespace new-dir name)
rlm@0:        (println (str name "-clojure.tar.bz2"))
rlm@0:        (println name)
rlm@0:        (bzip-export-files new-dir (file-str "~/coderloop/") name "-clojure.tar.bz2")
rlm@0:        (FileUtils/copyFileToDirectory
rlm@0: 	(file-str (str "~/coderloop/" name "-clojure.tar.bz2"))
rlm@0: 	(file-str "~/coderloop-test")))))