package demo;

import com.alibaba.fastjson.JSONObject;
import com.alibaba.fastjson.serializer.SerializerFeature;
import okhttp3.*;
import org.junit.Test;

import java.io.*;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLConnection;
import java.text.SimpleDateFormat;
import java.util.Date;

public class OkHttpApi {

    public static final MediaType JSON = MediaType.get("application/json; charset=utf-8");

    OkHttpClient client = new OkHttpClient();

    //推送demo
    public String post(String url, String json) throws IOException {
        RequestBody body = RequestBody.create(json, JSON);
        Request request = new Request.Builder()
                .url(url)
                .post(body)
                .build();
        try (Response response = client.newCall(request).execute()) {
            return response.body().string();
        }
    }

    //拉取demo
    public String run(String url) throws IOException {
        Request request = new Request.Builder()
                .url(url)
                .build();

        try (Response response = client.newCall(request).execute()) {
            return response.body().string();
        }
    }

    //使用okhttp调用api
    public JSONObject example(String url) {
        OkHttpApi okHttpApi = new OkHttpApi();
        String text = new String();
        try {
            text = okHttpApi.run(url);
        } catch (IOException ioe) {
            ioe.printStackTrace();
        } finally {
            return JSONObject.parseObject(text);
        }
    }

    //下载图片-以时间戳命名
    public static void downloadPicture(String pictureName, String urlString) {
        SimpleDateFormat formatter = new SimpleDateFormat("yyyyMMdd_HHmmss");
        Date date = new Date(System.currentTimeMillis());
        String currentTime = formatter.format(date);
        URL url = null;
        InputStream is = null;
        FileOutputStream os = null;
        try {
            url = new URL(urlString);
            URLConnection con = url.openConnection();
            is = con.getInputStream();
            byte[] bs = new byte[1024];
            int len;
            String filename = "D:\\Download\\" + pictureName + "_" + currentTime + ".jpg";
            File file = new File(filename);
            os = new FileOutputStream(file, true);
            while ((len = is.read(bs)) != -1) {
                os.write(bs, 0, len);
            }
            System.out.println("Download image successfully");
            os.close();
            is.close();
        } catch (MalformedURLException mue) {
            mue.printStackTrace();
        } catch (IOException ioe) {
            ioe.printStackTrace();
        }
    }


    //Api
    private static String url = "https://route.showapi.com/1287-1?showapi_appid=8131&showapi_sign=a46c44f01c";

    //测试入口
    @Test
    public void testing() {
        JSONObject jsonObject = example(url);
        String pretty = com.alibaba.fastjson.JSON
                .toJSONString(jsonObject, SerializerFeature.PrettyFormat, SerializerFeature.WriteMapNullValue, SerializerFeature.WriteDateUseDateFormat);
        System.out.println("\n" + pretty + "\n");

        Object data = jsonObject.getJSONObject("showapi_res_body").get("data");
        String picturePath = JSONObject.parseObject(data.toString()).getString("img_1920").toString();
        String pictureName = picturePath.split("/")[picturePath.split("/").length - 1];

        downloadPicture(pictureName, picturePath);
    }


}