package org.jenkinsci.plugins.svn.upload.step;

import java.io.*;

import org.tmatesoft.svn.core.SVNCommitInfo; import org.tmatesoft.svn.core.SVNDepth; import org.tmatesoft.svn.core.SVNException; import org.tmatesoft.svn.core.SVNNodeKind; import org.tmatesoft.svn.core.SVNURL; import org.tmatesoft.svn.core.auth.ISVNAuthenticationManager; import org.tmatesoft.svn.core.internal.io.dav.DAVRepositoryFactory; import org.tmatesoft.svn.core.internal.io.fs.FSRepositoryFactory; import org.tmatesoft.svn.core.internal.io.svn.SVNRepositoryFactoryImpl; import org.tmatesoft.svn.core.internal.wc.DefaultSVNOptions; import org.tmatesoft.svn.core.io.SVNRepository; import org.tmatesoft.svn.core.io.SVNRepositoryFactory; import org.tmatesoft.svn.core.wc.SVNClientManager; import org.tmatesoft.svn.core.wc.SVNRevision; import org.tmatesoft.svn.core.wc.SVNUpdateClient; import org.tmatesoft.svn.core.wc.SVNWCUtil;

public class SvnUpload {     private SVNClientManager clientManager;     private ISVNAuthenticationManager authManager;     private SVNRepository repository;     /**      *      * @param svnUrl svn地址      * @param svnUsername svn用户名称      * @param svnPasswd svn用户密码      * @throws SVNException 异常信息      */     public SvnUpload(String svnUrl, String svnUsername, String svnPasswd)throws SVNException{         try {             this.createDefaultAuthenticationManager(svnUsername, svnPasswd);             this.authSvn(svnUrl);         } catch (SVNException e) {             throw new RuntimeException(e.getMessage());         }     }     /**      * 通过不同的协议初始化版本库      */       private void setupLibrary() {           DAVRepositoryFactory.setup();           SVNRepositoryFactoryImpl.setup();           FSRepositoryFactory.setup();       }       /**      *      * @param username svn用户名称      * @param password svn用户密码      * @throws SVNException 异常信息      */     private void createDefaultAuthenticationManager(String username, String password)throws SVNException{         try {               // 身份验证               authManager = SVNWCUtil.createDefaultAuthenticationManager(username, password);         } catch (Exception e) {               throw new RuntimeException("SVN身份认证失败：" + e.getMessage());         }       }     /**      * 验证登录svn      * @param svnUrl 用户svn的仓库地址      * @throws SVNException 异常信息      */     private void authSvn(String svnUrl) throws SVNException {           // 初始化版本库           setupLibrary();           try {               repository = SVNRepositoryFactory.create(SVNURL.parseURIEncoded(svnUrl));           } catch (SVNException e) {               throw new RuntimeException("SVN创建库连接失败：" + e.getMessage());         }                  // 创建身份验证管理器           repository.setAuthenticationManager(authManager);           DefaultSVNOptions options = SVNWCUtil.createDefaultOptions(true);           try {               //创建SVN实例管理器             clientManager = SVNClientManager.newInstance(options,authManager);           } catch (Exception e) {               throw new RuntimeException("SVN实例管理器创建失败：" + e.getMessage());         }     }          /**      * 添加文件和目录到版本控制下      * @param wcPath 工作区路径      * @throws SVNException 异常信息      */       private void addEntry(File wcPath) throws SVNException{           try {               clientManager.getWCClient().doAdd(new File[] { wcPath }, true,                       false, false, SVNDepth.INFINITY, false, false, true);           } catch (SVNException e) {               throw new RuntimeException("SVN添加文件到版本控制下失败：" + e.getMessage());         }       }                    /**      * 将工作副本提交到svn      * @param wcPath  被提交的工作区路径      * @param keepLocks 是否在SVN仓库中打开或不打开文件      * @param commitMessage 提交信息      * @return 返回信息      * @throws SVNException 异常信息      */       private SVNCommitInfo commit(File wcPath, boolean keepLocks, String commitMessage) throws SVNException {           try {               return clientManager.getCommitClient().doCommit(                       new File[] { wcPath }, keepLocks, commitMessage, null,                       null, false, false, SVNDepth.INFINITY);           } catch (SVNException e) {               throw new RuntimeException("SVN提交失败：" + e.getMessage());         }       }              /**      * 确定path是否是一个工作空间      * @param path 文件路径      * @return 返回信息      * @throws SVNException 异常信息      */       private boolean isWorkingCopy(File path) throws SVNException{           if(!path.exists()){               return false;           }           try {               if(null == SVNWCUtil.getWorkingCopyRoot(path, false)){                   return false;               }           } catch (SVNException e) {               throw new RuntimeException("确定path是否是一个工作空间 失败：" + e.getMessage());         }           return true;       }              /**      * 确定一个URL在SVN上是否存在      * @param url svn访问地址      * @return 返回信息      * @throws SVNException 异常信息      */       private boolean isURLExist(SVNURL url) throws SVNException{           try {               SVNRepository svnRepository = SVNRepositoryFactory.create(url);             svnRepository.setAuthenticationManager(authManager);               SVNNodeKind nodeKind = svnRepository.checkPath("", -1);               return nodeKind == SVNNodeKind.NONE ? false : true;            } catch (SVNException e) {               throw new RuntimeException("确定一个URL在SVN上是否存在失败：" + e.getMessage());         }         }            /**        * 递归检查不在版本控制的文件，并add到svn        * @param wc  检查的文件      * @throws SVNException 异常信息      */       private void checkVersiondDirectory(File wc) throws SVNException{           if(!SVNWCUtil.isVersionedDirectory(wc)){               this.addEntry(wc);           }           if(wc.isDirectory()){               for(File sub:wc.listFiles()){                 if(sub.isDirectory() && sub.getName().equals(".svn")){                     continue;                   }                   checkVersiondDirectory(sub);               }           }       }          /**      * 删除目录（文件夹）以及目录下的文件      * @param   sPath 被删除目录的文件      * @return  目录删除成功返回true，否则返回false      */       private boolean deleteDirectory(String sPath) {         //如果sPath不以文件分隔符结尾，自动添加文件分隔符           if (!sPath.endsWith(File.separator)) {               sPath = sPath + File.separator;           }           File dirFile = new File(sPath);           //如果dir对应的文件不存在，或者不是一个目录，则退出           if (!dirFile.exists() || !dirFile.isDirectory()) {               return false;           }           boolean flag = true;           //删除文件夹下的所有文件(包括子目录)           File[] files = dirFile.listFiles();           for (int i = 0; i < files.length; i++) {               //删除子文件               if (files[i].isFile()) {                   flag = deleteFile(files[i].getAbsolutePath());                   if (!flag) break;               } //删除子目录               else {                   flag = deleteDirectory(files[i].getAbsolutePath());                   if (!flag) break;               }           }           if (!flag) return false;           //删除当前目录           if (dirFile.delete()) {               return true;           } else {               return false;           }       }          /**      * 删除单个文件      * @param   sPath    被删除文件的文件      * @return 单个文件删除成功返回true，否则返回false      */       private boolean deleteFile(String sPath) {           boolean flag = false;         File file = new File(sPath);           // 路径为文件且不为空则进行删除           if (file.isFile() && file.exists()) {               file.delete();               flag = true;           }         return flag;       }          /**      *  根据路径删除指定的目录或文件，无论存在与否      *@param sPath  要删除的目录或文件      *@return 删除成功返回 true，否则返回 false。      */       private boolean DeleteFolder(String sPath) {           boolean flag = false;         File file = new File(sPath);           // 判断目录或文件是否存在           if (!file.exists()) {  // 不存在返回 false               return flag;           } else {               // 判断是否为文件               if (file.isFile()) {  // 为文件时调用删除文件方法                   return deleteFile(file.getAbsolutePath());               } else {  // 为目录时调用删除目录方法                   return deleteDirectory(file.getAbsolutePath());               }           }       }            /**      * 更新SVN工作区      * @param wcPath 工作区路径      * @param updateToRevision 更新版本      * @param depth update的深度：目录、子目录、文件      * @return 返回信息      * @throws SVNException 异常信息      */       private long update(File wcPath,SVNRevision updateToRevision, SVNDepth depth) throws SVNException{           SVNUpdateClient updateClient = clientManager.getUpdateClient();           updateClient.setIgnoreExternals(false);           try {               return updateClient.doUpdate(wcPath, updateToRevision,depth, false, false);           } catch (SVNException e) {               throw new RuntimeException("更新SVN工作区失败：" + e.getMessage());         }       }              /**      * SVN仓库文件检出      * @param url 文件url      * @param revision 检出版本      * @param destPath 目标路径      * @param depth checkout的深度，目录、子目录、文件      * @return 返回信息      * @throws SVNException 异常信息      */       private long checkout(SVNURL url, SVNRevision revision, File destPath, SVNDepth depth) throws SVNException{           SVNUpdateClient updateClient = clientManager.getUpdateClient();            updateClient.setIgnoreExternals(false);           try {               return updateClient.doCheckout(url, destPath, revision, revision,depth, false);           } catch (SVNException e) {               throw new RuntimeException("检出SVN仓库失败：" + e.getMessage());         }       }     /**      * @param svnUrl svn地址      * @param workspace 工作区      * @param filepath 上传的文件地址      * @param filename 文件名称      * @throws SVNException 异常信息      */     private void checkWorkCopy(String svnUrl,String workspace,String filepath,String filename)throws SVNException{             SVNURL repositoryURL = null;         try {               repositoryURL = SVNURL.parseURIEncoded(svnUrl);           } catch (SVNException e) {               throw new RuntimeException("解析svnUrl失败：" + e.getMessage());         }         String fPath = "";         if(filepath.indexOf("/") != -1) {             fPath = filepath.substring(0,filepath.lastIndexOf("/"));         }         File wc = new File(workspace+"/"+fPath);         File wc_project = new File( workspace + "/" + fPath);                    SVNURL projectURL = null;         try {               projectURL = repositoryURL.appendPath(filename, false);           } catch (SVNException e) {               throw new RuntimeException("解析svnUrl文件失败：" + e.getMessage());         }                      if(!this.isWorkingCopy(wc)){               if(!this.isURLExist(projectURL)){                 this.checkout(repositoryURL, SVNRevision.HEAD, wc, SVNDepth.EMPTY);               }else{                   this.checkout(projectURL, SVNRevision.HEAD, wc_project, SVNDepth.INFINITY);               }           }else{             this.update(wc, SVNRevision.HEAD, SVNDepth.INFINITY);           }       }     /**      * 循环删除.svn目录      * @param spath      */     private void deletePointSVN(String spath){         File wc = new File(spath);         for(File sub:wc.listFiles()){             if(sub.isDirectory() && sub.getName().equals(".svn")){                 this.deleteDirectory(sub.getAbsolutePath());                 continue;             }             if(sub.isDirectory()){                 deletePointSVN(sub.getAbsolutePath());             }           }       }     /**      *      * @param svnUrl svn地址      * @param workspace 工作区      * @param filepath 上传的文件地址      * @param filename 上传的文件名称      * @param isOverwrite 是否覆盖      * @throws SVNException 异常信息      */     public void upload(String svnUrl,String workspace,String filepath,String filename,Boolean isOverwrite)throws SVNException{         String svnfilePath = svnUrl+"/"+filename;         //开始前删除以前的.svn文件目录         deletePointSVN(workspace);         boolean flag = this.isURLExist(SVNURL.parseURIDecoded(svnfilePath));         if(flag){             if(isOverwrite){                 this.uploadFile(svnUrl, workspace, filepath,filename);             }         }else{             this.uploadFile(svnUrl, workspace, filepath,filename);         }         //结束后删除当前的.svn文件目录         deletePointSVN(workspace);     }     /**      *      * @param svnUrl svn地址      * @param workspace 工作区      * @param filepath 上传的文件地址      * @param filename 文件名称      * @throws SVNException 异常信息      */     private void uploadFile(String svnUrl,String workspace,String filepath,String filename)throws SVNException{         this.checkWorkCopy(svnUrl, workspace, filepath,filename);         File file = new File(workspace+"/"+filepath);         this.checkVersiondDirectory(file);         this.commit(file, false, "commit file:"+file);     }

    public static void main(String[] args) throws SVNException {         try {             String svnUrl = "http://192.168.18.200/svn/repository/hello";             String username = "admin";             String passwd = "admin";             String workspace = "C:\\Users\\kube\\Desktop\\test";             String upfile = "/q/w/a.txt,b.txt";             Boolean isOverwrite = true;             SvnUpload svnUpload = new SvnUpload(svnUrl,username, passwd);             String [] fileArray = upfile.split(",");             for(int i=0;i