/*
 *                 Twidere - Twitter client for Android
 *
 *  Copyright (C) 2012-2015 Mariotaku Lee <mariotaku.lee@gmail.com>
 *
 *  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 org.mariotaku.twidere.util.net;

import android.content.ContentResolver;
import android.content.ContentValues;
import android.content.Context;

import com.squareup.okhttp.Interceptor;
import com.squareup.okhttp.OkHttpClient;
import com.squareup.okhttp.Request;
import com.squareup.okhttp.RequestBody;
import com.squareup.okhttp.Response;
import com.squareup.okhttp.ResponseBody;

import org.mariotaku.twidere.model.RequestType;
import org.mariotaku.twidere.provider.TwidereDataStore.NetworkUsages;
import org.mariotaku.twidere.util.Utils;

import java.io.IOException;

/**
 * Created by mariotaku on 15/6/24.
 */
public class NetworkUsageUtils {
    public static void initForHttpClient(Context context, OkHttpClient client) {
        client.networkInterceptors().add(new NetworkUsageInterceptor(context));
    }

    private static int sNetworkType;

    public static void setNetworkType(int networkType) {
        NetworkUsageUtils.sNetworkType = networkType;
    }

    private static class NetworkUsageInterceptor implements Interceptor {
        private final Context context;

        public NetworkUsageInterceptor(Context context) {
            this.context = context;
            setNetworkType(Utils.getActiveNetworkType(context));
        }

        @Override
        public Response intercept(Chain chain) throws IOException {
            final Request request = chain.request();
            final Object tag = request.tag();
            if (!(tag instanceof RequestType)) return chain.proceed(request);
            final ContentValues values = new ContentValues();
            values.put(NetworkUsages.TIME_IN_HOURS, System.currentTimeMillis() / 1000 / 60 / 60);
            values.put(NetworkUsages.KILOBYTES_SENT, getBodyLength(request.body()) / 1024.0);
            values.put(NetworkUsages.REQUEST_TYPE, ((RequestType) tag).getName());
            values.put(NetworkUsages.REQUEST_NETWORK, sNetworkType);
            final Response response = chain.proceed(request);
            values.put(NetworkUsages.KILOBYTES_RECEIVED, getBodyLength(response.body()) / 1024.0);
            final ContentResolver cr = context.getContentResolver();
            cr.insert(NetworkUsages.CONTENT_URI, values);
            return response;
        }

        private long getBodyLength(RequestBody body) throws IOException {
            if (body == null) return 0;
            final long length = body.contentLength();
            return length > 0 ? length : 0;
        }

        private long getBodyLength(ResponseBody body) throws IOException {
            if (body == null) return 0;
            final long length = body.contentLength();
            return length > 0 ? length : 0;
        }
    }
}