com.gti.redirects.Redirects.RedirectStorage.java Source code

Introduction

Here is the source code for com.gti.redirects.Redirects.RedirectStorage.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package com.gti.redirects.Redirects;

import java.io.*;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import com.gti.redirects.StorageI;
import org.json.simple.JSONArray;
import org.json.simple.JSONObject;
import org.json.simple.parser.JSONParser;
import org.json.simple.parser.ParseException;

/**
 * @author xach
 */
public class RedirectStorage implements StorageI {
    private File jsonFile;

    public RedirectStorage(String filePath) {
        File redirectFile = new File(filePath);
        try {
            if (redirectFile.createNewFile()) {
                JSONArray jsonArray = new JSONArray();
                jsonFile = redirectFile;
                writeJsonArray(jsonArray);
            } else {
                jsonFile = redirectFile;
            }
        } catch (IOException e) {
            e.printStackTrace();
        }

    }

    public RedirectStorage() {
        this(System.getProperty("user.dir") + "/redirects.json");
    }

    @Override
    public List<Map<String, String>> getAll() {
        List<Map<String, String>> redirects = new ArrayList<Map<String, String>>();
        JSONArray jsonArray = null;
        try {
            jsonArray = parseFile();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (ParseException e) {
            e.printStackTrace();
        }
        if (jsonArray != null) {
            for (int i = 0; i < jsonArray.size(); i++) {
                JSONObject object = (JSONObject) jsonArray.get(i);
                Map<String, String> map = new HashMap<String, String>();
                map.put("domain", object.get("domain").toString());
                map.put("status", object.get("status").toString());
                map.put("redirect_to", object.get("redirect_to").toString());

                redirects.add(map);
                // now do something with the Object
            }
        }
        return redirects;
    }

    @Override
    public boolean add(Map redirect) {
        try {
            System.out.println("Successfully Copied JSON Object to File...");
            JSONArray jsonArray = parseFile();
            JSONObject redirectJson = convertRedirectToJson(redirect);
            jsonArray.add(redirectJson);
            writeJsonArray(jsonArray);
            return true;
        } catch (IOException e) {
            e.printStackTrace();
        } catch (ParseException e) {
            e.printStackTrace();
        }
        return false;
    }

    @Override
    public boolean delete(int index) {
        try {
            JSONArray jsonArray = parseFile();
            jsonArray.remove(index);
            writeJsonArray(jsonArray);
        } catch (IOException e) {
            e.printStackTrace();
        } catch (ParseException e) {
            e.printStackTrace();
        }
        return false;
    }

    private JSONArray parseFile() throws IOException, ParseException {
        JSONParser parser = new JSONParser();
        return (JSONArray) parser.parse(new FileReader(jsonFile.getPath()));
    }

    private JSONObject convertRedirectToJson(Map redirect) {
        JSONObject json = new JSONObject();

        json.put("domain", redirect.get("domain"));
        json.put("type", redirect.get("status"));
        json.put("redirect_to", redirect.get("redirect_to"));
        return json;
    }

    private boolean writeJsonArray(JSONArray jsonArray) {
        try (FileWriter file = new FileWriter(jsonFile.getPath())) {
            file.write(jsonArray.toJSONString());
            System.out.println("Successfully Copied JSON Object to File...");
            System.out.println("\nJSON Object: " + jsonArray);
            return true;
        } catch (IOException e) {
            e.printStackTrace();
        }
        return false;
    }

}