<?php

namespace FileBird\Classes;



defined('ABSPATH') || exit;



use FileBird\Model\Folder as FolderModel;



class Tree {

    public static function getCount($folder_id, $from_data = null) {

      global $wpdb;



      $select = "SELECT COUNT(*) FROM {$wpdb->posts} as posts WHERE ";

      $where = array("post_type = 'attachment'");



      // With $folder_id == -1. We get all

      $where[] = "(posts.post_status = 'inherit' OR posts.post_status = 'private')";



      // with specific folder

      if($folder_id > 0 && ! apply_filters('fbv_speedup_get_count_query', false)) {

        $post__in = $wpdb->get_col("SELECT `attachment_id` FROM {$wpdb->prefix}fbv_attachment_folder WHERE `folder_id` = " . (int)$folder_id);

        if(count($post__in) == 0) {

          $post__in = array(0);

        }

        $where[] = "(ID IN (".implode(', ', $post__in)."))";

      } elseif ($folder_id == 0) {

        return 0;//return 0 if this is uncategorized folder

      }

      $query = apply_filters('fbv_get_count_query', $select . implode(' AND ', $where), $folder_id);

      return (int)$wpdb->get_var($query);

    }

    public static function getAllFoldersAndCount() {

      global $wpdb;

      $query = "SELECT fbva.folder_id, count(fbva.attachment_id) as count FROM {$wpdb->prefix}fbv_attachment_folder as fbva 

      INNER JOIN {$wpdb->prefix}fbv as fbv ON fbv.id = fbva.folder_id 

      INNER JOIN {$wpdb->posts} as posts ON fbva.attachment_id = posts.ID  

      WHERE (posts.post_status = 'inherit' OR posts.post_status = 'private') 

      AND (posts.post_type = 'attachment') 

      AND fbv.created_by = ".apply_filters('fbv_in_not_in_created_by', '0')." 

      GROUP BY fbva.folder_id";

      $query = apply_filters('fbv_all_folders_and_count', $query);

    

      $results = $wpdb->get_results($query);

      $return = array();

      if(is_array($results)) {

        foreach ($results as $k => $v) {

          $return[$v->folder_id] = $v->count;

        }

      }

      return $return;

    }

    public static function getFolders($order_by = null, $flat = false, $level = 0, $show_level = false) {

        $folders_from_db = FolderModel::allFolders('*', null, $order_by);

        $default_folders = array();

        

        $tree = self::getTree($folders_from_db, 0, $default_folders, $flat, $level, $show_level);

        return $tree;

    }

    

    private static function getTree($data, $parent = 0, $default = null,  $flat = false, $level = 0, $show_level = false) {

      $tree = is_null($default) ? array() : $default;

      foreach($data as $k => $v) {

        if($v->parent == $parent) {

          $children = self::getTree($data, $v->id, null, $flat, $level + 1, $show_level);

          $f = array(

            'id' => (int)$v->id,

            'text' => $show_level ? str_repeat('-', $level) . $v->name : $v->name,

            'li_attr' => array("data-count" => 0, "data-parent" => (int)$parent),

            'count' => 0

          );

           

          if($flat === true) {

            $tree[] = $f;

            foreach ($children as $k2 => $v2) {

              $tree[] = $v2;

            }

          } else {

            $f['children'] = $children;

            $tree[] = $f;

          }

        }

      }

      return $tree;

  }

}