#!/usr/bin/python
# coding=utf-8
#
# Copyright (C) 2018-2025 by dream-alpha
#
# In case of reuse of this source code please do not remove this copyright.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU 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 General Public License for more details.
#
# For more information on the GNU General Public License see:
# <http://www.gnu.org/licenses/>.


import os
from .Debug import logger
from .ServiceUtils import ALL_VIDEO


def countDiskFiles(bookmarks):
	files_count = 0
	dirs_count = 0
	dirs_list = []
	for bookmark in bookmarks:
		for root, dirs, files in os.walk(bookmark, followlinks=True):
			# logger.debug("root: %s, dirs: %s, files: %s", root, dirs, files)
			for afile in files:
				if os.path.splitext(afile)[1] in ALL_VIDEO:
					files_count += 1
			for adir in dirs:
				bdir = os.path.join(root, adir)
				dir_name = bdir[len(bookmark):]
				if "trashcan" not in dir_name:
					dirs_list.append(dir_name)
					dirs_count += 1
	logger.debug("files_count: %s, dirs_count: %s", files_count, dirs_count)
	return files_count, dirs_count, dirs_list