/* ownCloud Android client application
 *   Copyright (C) 2011  Bartek Przybylski
 *
 *   This program is free software: you can redistribute it and/or modify
 *   it under the terms of the GNU General Public License as published by
 *   the Free Software Foundation, either version 3 of the License, or
 *   (at your option) any later version.
 *
 *   This program is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *   GNU General Public License for more details.
 *
 *   You should have received a copy of the GNU General Public License
 *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */
package eu.alefzero.webdav;

import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.net.URLDecoder;

import org.apache.commons.httpclient.Credentials;
import org.apache.commons.httpclient.HttpClient;
import org.apache.commons.httpclient.UsernamePasswordCredentials;
import org.apache.commons.httpclient.auth.AuthScope;
import org.apache.commons.httpclient.methods.GetMethod;
import org.apache.commons.httpclient.methods.HeadMethod;
import org.apache.commons.httpclient.methods.PutMethod;
import org.apache.commons.httpclient.params.HttpMethodParams;
import org.apache.commons.httpclient.protocol.Protocol;
import org.apache.http.HttpStatus;
import org.apache.jackrabbit.webdav.client.methods.MkColMethod;

import eu.alefzero.owncloud.authenticator.EasySSLSocketFactory;

import android.net.Uri;
import android.util.Log;

public class WebdavClient extends HttpClient {
  private Uri mUri;
  private Credentials mCredentials;
  final private static String TAG = "WebdavClient";
  private static final String USER_AGENT = "Android-ownCloud";
  
  public WebdavClient(Uri uri) {
    mUri = uri;
    getParams().setParameter(HttpMethodParams.USER_AGENT, USER_AGENT);
  }
  
  public void setCredentials(String username, String password) {
    getParams().setAuthenticationPreemptive(true);
    getState().setCredentials(AuthScope.ANY, getCredentials(username, password));
  }
  
  private Credentials getCredentials(String username, String password) {
    if (mCredentials == null)
      mCredentials = new UsernamePasswordCredentials(username, password); 
    return mCredentials;
  }

  public void allowUnsignedCertificates() {
    // https
    Protocol.registerProtocol("https", new Protocol("https", new EasySSLSocketFactory(), 443));
  }
  
  public boolean downloadFile(String filepath, File targetPath) {
    //HttpGet get = new HttpGet(mUri.toString() + filepath.replace(" ", "%20"));
   
    Log.e("ASD", mUri.toString() + URLDecoder.decode(filepath) + "");
    GetMethod get = new GetMethod(mUri.toString() + URLDecoder.decode(filepath));
    
//    get.setHeader("Host", mUri.getHost());
//    get.setHeader("User-Agent", "Android-ownCloud");
    
    try {
      int status = executeMethod(get);
      if (status != HttpStatus.SC_OK) {
        return false;
      }
      BufferedInputStream bis = new BufferedInputStream(get.getResponseBodyAsStream());
      FileOutputStream fos = new FileOutputStream(targetPath);
      
      byte[] bytes = new byte[512];
      int readResult;
      while ((readResult = bis.read(bytes)) != -1) fos.write(bytes, 0, readResult);
      
    } catch (IOException e) {
      e.printStackTrace();
      return false;
    }
    return true;
  }
  
  public boolean putFile(String localFile,
                  String remoteTarget,
                  String contentType) {
    boolean result = true;

    try {
      FileRequestEntity entity = new FileRequestEntity(new File(localFile), contentType);
      PutMethod put = new PutMethod(mUri.toString() + remoteTarget.substring(1));
      put.setRequestEntity(entity);
      int status = executeMethod(put);
      Log.d(TAG, "PUT method return with status "+status);

      Log.i(TAG, "Uploading, done");
    } catch (final Exception e) {
      Log.i(TAG, ""+e.getMessage());
      result = false;
    }
    
    return result;
  }
  
  public int tryToLogin() {
    int r = 0; 
    HeadMethod head = new HeadMethod(mUri.toString());
    try {
      r = executeMethod(head);
    } catch (Exception e) {
      Log.e(TAG, "Error: " + e.getMessage());
    }
    return r;
  }

  public boolean createDirectory(String path) {
    try {
      MkColMethod mkcol = new MkColMethod(mUri.toString() + "/" + path + "/");
      int status = executeMethod(mkcol);
      Log.d(TAG, "Status returned " + status);
      Log.d(TAG, "uri: " + mkcol.getURI().toString());
      Log.i(TAG, "Creating dir completed");
    } catch (final Exception e) {
      e.printStackTrace();
      return false;
    }
    return true;
  }
}