VvebOIDC/system/traits/crud.php

<?php

/**
 * Vvveb
 *
 * Copyright (C) 2022  Ziadin Givan
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 *
 */

namespace Vvveb\System\Traits;

use function Vvveb\__;
use function Vvveb\humanReadable;
use function Vvveb\model;
use function Vvveb\sanitizeHTML;
use Vvveb\System\Images;

trait Crud {
	protected $module;

	protected $type;

	protected $controller;

	protected $redirect = true;

	protected $data_id;

	protected $type_id;

	protected $model;

	protected $options = [];

	protected $data = [];

	//protected $fullPost;

	function delete() {
		if (APP == 'admin') {
			$this->checkCsrf();
		}

		$type    = $this->type;
		$type_id = $this->type_id ?? "{$type}_id";

		if (APP == 'admin') {
			$data_id = $this->request->post[$type_id];
		} else {
			$data_id = $this->request->post[$type_id] ?? $this->request->get[$type_id] ?? false;
		}

		if ($data_id) {
			if (is_numeric($data_id)) {
				$data_id = [$data_id];
			}

			if (! isset($this->modelName)) {
				$modelName = $type;
			}

			$model               = model($modelName);
			$options             = [$type_id => $data_id] + $this->global;
			$result              = $model->delete($options);
			$name                = ucfirst(__($type));

			if ($result && isset($result[$type])) {
				if ($result[$type]) {
					$this->view->success[] = sprintf(__('%s(s) deleted!'), humanReadable($name));
				} else {
					$this->view->info[] = sprintf(__('No rows affected!'), humanReadable($name));
				}
			} else {
				$this->view->errors[] = sprintf(__('Error deleting %s!'), humanReadable($name));
			}
		}

		return $this->index();
	}

	function save() {
		$type        = $this->type;
		$type_id     = "{$type}_id";
		$module      = $this->module;
		$controller  = $this->controller ?? $type;
		$result      = [];

		$this->data_id = $this->request->get[$type_id] ?? false;

		if (isset($this->fullPost) && $this->fullPost) {
			$this->data = $this->request->post ?? [];
		} else {
			$this->data = $this->request->post[$type] ?? [];
		}

		if ($this->data) {
			if (! isset($this->modelName)) {
				$this->modelName = $type;
			}

			$this->model = model($this->modelName);

			if (! $this->data_id) {
				$this->data['created_at'] = $this->data['created_at'] ?? date('Y-m-d H:i:s');
			}
			$this->data['updated_at'] = $this->data['updated_at'] ?? date('Y-m-d H:i:s');

			if (isset($this->fullPost) && $this->fullPost) {
				$options = $this->data + $this->global;
			} else {
				$options = [$type => $this->data] + $this->global;
			}

			foreach (['content', 'name', 'excerpt'] as $field) {
				if (isset($this->data[$field])) {
					$this->data[$field] = sanitizeHTML($this->data[$field]);
				}
			}

			if ($this->data_id) {
				$options[$type_id] = $this->data_id;
				$this->$type_id    = $this->data_id;
				$result            = $this->model->edit($options);
			} else {
				$result         = $this->model->add($options);
				$this->$type_id = $result[$type] ?? false;
			}

			if ($result && isset($result[$type])) {
				$successMessage             = humanReadable(__($type)) . __(' saved!');
				$this->view->success['get'] = $successMessage;
				$this->session->set('success', ['get' => $successMessage]);

				if (! $this->data_id && $this->redirect) {
					if ($this->redirect === true) {
						$this->redirect(['module' => "$module/$controller", $type_id => $result[$type], 'success' => $successMessage]);
					} else {
						$this->redirect($this->redirect . "?$type_id={$result[$type]}&success=$successMessage");
					}
				}
			} else {
				$this->view->errors[] = __('Error saving!');
			}
		}

		if ($this->redirect) {
			return $this->index();
		}

		return $result;
	}

	protected function index() {
		$type             = $this->type;
		$type_id          = $this->type_id ?? "{$type}_id";
		$this->data_id    = $this->request->get[$type_id] ?? false;
		$this->slug    	  = $this->request->get['slug'] ?? false;
		$this->data       = [];

		if ($this->data_id || $this->slug) {
			if (! isset($this->modelName)) {
				$this->modelName = $type;
			}

			$this->model = model($this->modelName);

			if ($this->data_id) {
				$this->options[$type_id] = $this->data_id;
			}

			if ($this->slug) {
				$this->options['slug'] = $this->slug;
			}

			$this->options += $this->global + $this->request->get;
			//unset($this->options['user_id']);

			$result = $this->model->get($this->options);

			if ($result) {
				$this->data = $result;

				if (isset($this->data['image'])) {
					$this->data['image_url'] = Images::image($this->data['image'], $type);
				}
			}
		}

		return $this->data;
	}
}