src/jalview/ws/utils/UrlDownloadClient.java

   1 /*
   2  * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
   3  * Copyright (C) $$Year-Rel$$ The Jalview Authors
   4  *
   5  * This file is part of Jalview.
   6  *
   7  * Jalview is free software: you can redistribute it and/or
   8  * modify it under the terms of the GNU General Public License
   9  * as published by the Free Software Foundation, either version 3
  10  * of the License, or (at your option) any later version.
  11  *
  12  * Jalview is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty
  14  * of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  15  * PURPOSE.  See the GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with Jalview.  If not, see <http://www.gnu.org/licenses/>.
  19  * The Jalview Authors are detailed in the 'AUTHORS' file.
  20  */
  21
  22 package jalview.ws.utils;
  23
  24 import jalview.util.Platform;
  25
  26 import java.io.File;
  27 import java.io.FileOutputStream;
  28 import java.io.IOException;
  29 import java.net.URL;
  30 import java.nio.channels.Channels;
  31 import java.nio.channels.ReadableByteChannel;
  32 import java.nio.file.Files;
  33 import java.nio.file.Path;
  34 import java.nio.file.Paths;
  35 import java.nio.file.StandardCopyOption;
  36
  37 public class UrlDownloadClient
  38 {
  39   /**
  40    * Download and save a file from a URL
  41    *
  42    * @param urlstring
  43    *          url to download from, as string
  44    * @param outfile
  45    *          the name of file to save the URLs to
  46    * @throws IOException
  47    */
  48   public static void download(String urlstring, String outfile)
  49           throws IOException
  50   {
  51
  52     FileOutputStream fos = null;
  53     ReadableByteChannel rbc = null;
  54     Path temp = null;
  55     try
  56     {
  57       temp = Files.createTempFile(".jalview_", ".tmp");
  58
  59       URL url = new URL(urlstring);
  60       rbc = Channels.newChannel(url.openStream());
  61       fos = new FileOutputStream(temp.toString());
  62       fos.getChannel().transferFrom(rbc, 0, Long.MAX_VALUE);
  63
  64       // copy tempfile to outfile once our download completes
  65       // incase something goes wrong
  66       Files.copy(temp, Paths.get(outfile),
  67               StandardCopyOption.REPLACE_EXISTING);
  68     } catch (IOException e)
  69     {
  70       throw e;
  71     } finally
  72     {
  73       try
  74       {
  75         if (fos != null)
  76         {
  77           fos.close();
  78         }
  79       } catch (IOException e)
  80       {
  81         System.out.println(
  82                 "Exception while closing download file output stream: "
  83                         + e.getMessage());
  84       }
  85       try
  86       {
  87         if (rbc != null)
  88         {
  89           rbc.close();
  90         }
  91       } catch (IOException e)
  92       {
  93         System.out.println("Exception while closing download channel: "
  94                 + e.getMessage());
  95       }
  96       try
  97       {
  98         if (temp != null)
  99         {
 100           Files.deleteIfExists(temp);
 101         }
 102       } catch (IOException e)
 103       {
 104         System.out.println("Exception while deleting download temp file: "
 105                 + e.getMessage());
 106       }
 107     }
 108
 109   }
 110
 111   public static void download(String urlstring, File tempFile)
 112           throws IOException
 113   {
 114     if (!Platform.setFileBytes(tempFile, urlstring))
 115     {
 116       download(urlstring, tempFile.toString());
 117     }
 118   }
 119 }