<?php
  
  namespace App\Http\Controllers\V1\Components;
  
  use App\Http\Controllers\Controller;
  use Carbon\Carbon;
  use Illuminate\Http\Request;
  use Illuminate\Support\Facades\DB;
  use Illuminate\Support\Facades\Http;
  use Illuminate\Support\Facades\Log;

  use Illuminate\Support\Facades\Schema;

  
  class MltdController extends Controller
  {
      public function start(Request $request)
      {
          $data = [
              'model_id'  => 1,
              'mqtt_host' => config('mqtt.host'),
              'mqtt_port' => config('mqtt.port'),
              'mqtt_topic'    => 'xlsiem/event',
              'prediction_threshold'  => 0.3,
              'report_timedb_host' => env('DB_HOST'),
              'report_timedb_port' => env('DB_PORT'),
              'report_timedb_username' => env('DB_USERNAME'),
              'report_timedb_password' => env('DB_PASSWORD'),
              'report_timedb_database' => env('DB_DATABASE'),
              'report_timedb_table' => 'mltd',
              'report_timedb_ssl' => 'False',
              'report_asset_id' => 'server',
          ];
  
          $url = 'http://' . env('MLTD_HOST') . ':' . env('MLTD_PORT') . '/api/v1.0/mltd/prediction';
  
          $response = Http::withHeaders([
              'Application' => 'application/json'
          ])->post($url, $data);
  
          if (!isset($response->json()['process_id'])) {
              return response()->json(['error' => 'error'], 504);
          }
  
          $ret = [
              'component'  => 'mltd',
              'process_id' => $response->json()['process_id'],
              'created_at' => Carbon::now()
          ];
  
          DB::table('running_processes')->insert($ret);
  
          return $ret;
      }
  
      public function status(Request $request)
      {
          return DB::table('running_processes')
              ->where('component', '=', 'mltd')
              ->get()->toArray();
      }
  
      public function stop(Request $request, $pid)
      {
          $url = 'http://' . env('MLTD_HOST') . ':' . env('MLTD_PORT') . '/api/v1.0/mltd/prediction/stop/' . $pid;
          $response = Http::get($url);
  
          DB::table('running_processes')
              ->where('component', '=', 'mltd')
              ->where('process_id', '=', $pid)->delete();
  
          return $response;
      }
  
      public function train(Request $request, $train_id, $top)
      {
          $url = 'http://' . env('MLTD_HOST') . ':' . env('MLTD_PORT') . '/api/v1.0/mltd/threat-identification/' . $train_id . '/' . $top;
          $response = Http::get($url);
          return [
              'important_events' => explode(",", substr($response->json()['important_events'], 1, -1)),
              'timeframe' => $response->json()['timeframe']
          ];
      }

      public function count(Request $request)

      {

          if (!Schema::hasTable('mltd')) {
              return response()->json([
                  'error' => 'Relation for MLTD component does not exist'
              ], 503);
          }

          if ($request->has('start')) {
              $start = Carbon::parse($request->get('start'));
          } else {
              $start = Carbon::now()->subHour();

          }

  
          if ($request->has('end'))
              $end = Carbon::parse($request->get('end'));
          else {
              $end = Carbon::now();
          }
  
          return [
              'count' =>
                  DB::table('mltd')
                      ->whereBetween('created_on', [$start, $end])
                      ->count()
          ];

      }

  
      public function group(Request $request)
      {
          $request->validate([
              'unit' => 'required|string|in:second,minute,hour,day',
              'k' => 'required|integer',
              'limit' => 'nullable|integer'
          ]);
  
          if (!Schema::hasTable('mltd')) {
              return response()->json([
                  'error' => 'Relation for MLTD component does not exist'
              ], 503);
          }
  
          $q = $request->get('k') . ' ' . $request->get('unit');
          $query = DB::table('mltd')
              ->select(DB::raw('count(*)'), DB::raw("time_bucket('" . $q . "', created_on) AS interval"))
              ->groupby('interval')
              ->orderBy('interval', 'desc');
  
          if ($request->has('limit')) {
              $query = $query->limit($request->get('limit'));
          }
          return $query->get();
      }

  }