Twidere-App-Android-Twitter-Mastodon/twidere/src/main/java/edu/ucdavis/earlybird/UploadTask.java

package edu.ucdavis.earlybird;

import android.content.Context;
import android.content.SharedPreferences;
import android.os.AsyncTask;
import android.provider.Settings.Secure;

import org.mariotaku.restfu.annotation.method.POST;
import org.mariotaku.restfu.http.RestHttpClient;
import org.mariotaku.restfu.http.RestHttpRequest;
import org.mariotaku.restfu.http.RestHttpResponse;
import org.mariotaku.restfu.http.mime.FileTypedData;
import org.mariotaku.restfu.http.mime.MultipartTypedBody;
import org.mariotaku.twidere.util.TwitterAPIFactory;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;

import static org.mariotaku.twidere.util.Utils.copyStream;

public class UploadTask extends AsyncTask<Object, Object, Object> {

    private static final String LAST_UPLOAD_DATE = "last_upload_time";
    private static final double MILLSECS_HALF_DAY = 1000 * 60 * 60 * 12;

    private final String device_id;
    private final Context context;

    private final RestHttpClient client;

    private static final String PROFILE_SERVER_URL = "http://weik.metaisle.com/profiles";

    // private static final String PROFILE_SERVER_URL =
    // "http://192.168.0.105:3000/profiles";

    public UploadTask(final Context context) {
        this.context = context;
        this.client = TwitterAPIFactory.getDefaultHttpClient(context);
        device_id = Secure.getString(context.getContentResolver(), Secure.ANDROID_ID);
    }

    public void uploadMultipart(final String url, final File file) {
        final String app_root = file.getParent();
        final File tmp_dir = new File(app_root + "/tmp");
        if (!tmp_dir.exists()) {
            if (!tmp_dir.mkdirs()) {
                ProfilingUtil.log(context, "cannot create tmp, do nothing.");
                return;
            }
        }
        final File tmp = new File(tmp_dir, file.getName());
        file.renameTo(tmp);

        try {

            final RestHttpRequest.Builder builder = new RestHttpRequest.Builder();
            builder.url(PROFILE_SERVER_URL);
            builder.method(POST.METHOD);
            final MultipartTypedBody body = new MultipartTypedBody();
            body.add("upload", new FileTypedData(tmp));
            builder.body(body);
            final RestHttpResponse response = client.execute(builder.build());

            // Responses from the server (code and message)
            final int serverResponseCode = response.getStatus();

            ProfilingUtil.log(context, "server response code " + serverResponseCode);

            if (serverResponseCode / 100 == 2) {
                tmp.delete();
            } else {
                putBackProfile(context, tmp, file);
            }

        } catch (final IOException e) {
            e.printStackTrace();
            putBackProfile(context, tmp, file);
        }
    }

    @Override
    protected Object doInBackground(final Object... params) {

        final SharedPreferences prefs = context.getSharedPreferences("ucd_data_profiling", Context.MODE_PRIVATE);

        if (prefs.contains(LAST_UPLOAD_DATE)) {
            final long lastUpload = prefs.getLong(LAST_UPLOAD_DATE, System.currentTimeMillis());
            final double deltaDays = (System.currentTimeMillis() - lastUpload) / (MILLSECS_HALF_DAY * 2);
            if (deltaDays < 1) {
                ProfilingUtil.log(context, "Uploaded less than 1 day ago.");
                return null;
            }
        }

        final File root = context.getFilesDir();
        final File[] files = root.listFiles(new CSVFileFilter());

        uploadToNode(files);
        prefs.edit().putLong(LAST_UPLOAD_DATE, System.currentTimeMillis()).apply();
        return null;
    }

    private boolean uploadToNode(final File... files) {
        for (final File file : files) {
            if (file.isDirectory()) {
                continue;
            }
            final String url = PROFILE_SERVER_URL + "/" + device_id + "/"
                    + file.getName().replaceFirst("[.][^.]+$", "");
            ProfilingUtil.log(context, url);
            uploadMultipart(url, file);
        }
        return false;
    }

    public static void putBackProfile(final Context context, final File tmp, final File profile) {
        boolean success;
        if (profile.exists()) {
            try {
                final FileOutputStream os = new FileOutputStream(tmp, true);
                final FileInputStream is = new FileInputStream(profile);
                copyStream(is, os);
                is.close();
                os.close();
                success = true;
            } catch (final IOException e) {
                e.printStackTrace();
                success = false;
            }

            if (success && tmp.renameTo(profile) && tmp.delete()) {
                ProfilingUtil.log(context, "put profile back success");
            } else {
                ProfilingUtil.log(context, "put profile back failed");
            }
        } else {
            if (tmp.renameTo(profile)) {
                ProfilingUtil.log(context, "put profile back success");
            } else {
                ProfilingUtil.log(context, "put profile back failed");
            }
        }
    }
}