FriDroid/src/main/java/com/alterdekim/fridaapp/controller/MainActivityController.java

package com.alterdekim.fridaapp.controller;

import android.content.Intent;
import android.util.Log;
import android.view.LayoutInflater;
import android.view.View;
import android.widget.TextView;
import android.widget.Toast;

import androidx.appcompat.app.AppCompatActivity;
import androidx.room.Room;

import com.alterdekim.fridaapp.R;
import com.alterdekim.fridaapp.activity.MainActivity;
import com.alterdekim.fridaapp.activity.SingleConfigActivity;
import com.alterdekim.fridaapp.room.AppDatabase;
import com.alterdekim.fridaapp.room.Config;
import com.alterdekim.fridaapp.service.FridaService;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.introspect.JacksonAnnotationIntrospector;
import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
import com.google.android.material.switchmaterial.SwitchMaterial;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import io.reactivex.rxjava3.android.schedulers.AndroidSchedulers;
import io.reactivex.rxjava3.schedulers.Schedulers;

public class MainActivityController implements IController {

    private static final String TAG = MainActivityController.class.getSimpleName();

    private AppDatabase db;

    private MainActivity mainActivity;

    @Override
    public ControllerId getControllerId() {
        return ControllerId.MainActivityController;
    }

    @Override
    public void onCreateGUI(AppCompatActivity activity) {
        this.db = Room.databaseBuilder(activity.getApplicationContext(), AppDatabase.class, "def-db").build();
        this.mainActivity = (MainActivity) activity;
        Intent intent = new Intent(this.mainActivity, FridaService.class);
        intent.putExtra("vpn_data", new byte[0]);
        intent.putExtra("vpn_uid", -1);
        intent.putExtra("vpn_state", false);
        this.mainActivity.startService(intent);
        this.initConfigListGUI();
    }

    private void initConfigListGUI() {
        LayoutInflater inflater = this.mainActivity.getLayoutInflater();
        this.db.userDao().getAll()
                .subscribeOn(Schedulers.io())
                .observeOn(AndroidSchedulers.mainThread())
                .doOnNext(cl -> {
                    this.mainActivity.getCfg_list().removeAllViews();
                    if( cl.isEmpty() ) {
                        this.mainActivity.getCfg_list().addView(inflater.inflate(R.layout.content_nocfg, this.mainActivity.getCfg_list(), false));
                        return;
                    }
                    Iterator<Config> iter = cl.iterator();
                    List<SwitchMaterial> switches = new ArrayList<>();
                    while( iter.hasNext() ) {
                        Config config = iter.next();
                        View cfg_instance = inflater.inflate(R.layout.single_config, this.mainActivity.getCfg_list(), false);
                        this.mainActivity.getCfg_list().addView(cfg_instance);
                        TextView view_name = cfg_instance.findViewById(R.id.config_name);
                        SwitchMaterial view_switch = cfg_instance.findViewById(R.id.config_switch);
                        switches.add(view_switch);
                        view_switch.setUseMaterialThemeColors(true);
                        view_switch.setOnCheckedChangeListener((compoundButton, b) -> {
                            if( b ) {
                                for (SwitchMaterial ss : switches) {
                                    if (!ss.equals(view_switch)) {
                                        Log.i(TAG, "DEACTIVATED");
                                        ss.setChecked(false);
                                    }
                                }
                            }
                            toggleVpn(config, b);
                        });
                        view_name.setText(config.getTitle());
                        view_name.setOnClickListener(view -> {
                            Intent intent = new Intent(this.mainActivity, SingleConfigActivity.class);
                            intent.putExtra("config_data", config.getData_raw());
                            intent.putExtra("config_title", config.getTitle());
                            this.mainActivity.startActivity(intent);
                        });
                        if( iter.hasNext() ) this.mainActivity.getCfg_list().addView(inflater.inflate(R.layout.single_divider, this.mainActivity.getCfg_list(), false));
                    }
                })
                .doOnError(throwable -> this.mainActivity.getCfg_list().addView(inflater.inflate(R.layout.content_error, this.mainActivity.getCfg_list(), false)))
                .subscribe();
    }

    public void insertNewConfig(String name, byte[] config) {
        try {
            new ObjectMapper(new YAMLFactory()).setAnnotationIntrospector(new JacksonAnnotationIntrospector()).readValue(config, com.alterdekim.frida.config.Config.class);
        } catch (IOException e) {
            Toast.makeText(this.mainActivity, R.string.config_adding_error, Toast.LENGTH_LONG).show();
            return;
        }
        db.userDao().insertAll(new Config(name, config))
                .subscribeOn(Schedulers.io())
                .observeOn(AndroidSchedulers.mainThread())
                .doOnError(throwable -> Toast.makeText(MainActivityController.this.mainActivity, R.string.config_adding_error, Toast.LENGTH_LONG).show())
                .doOnComplete(this::initConfigListGUI)
                .subscribe();
    }

    private void toggleVpn(Config config, boolean val) {
        Intent intent = new Intent(this.mainActivity, FridaService.class);
        intent.putExtra("vpn_data", config.getData_raw());
        intent.putExtra("vpn_uid", config.getUid());
        intent.putExtra("vpn_state", val);
        this.mainActivity.startService(intent);
    }
}