#!/usr/bin/env python

import enum
import itertools
import json
import os
import re
import sys
from dataclasses import KW_ONLY, dataclass, field
from functools import partial, partialmethod
from operator import itemgetter
from tabnanny import verbose
from time import sleep

import defl
from defl import CLIError, DotDict, Null, Path, Undefined, cl, log
from defl._assert_ import Assert
from defl._typing_ import *
from defl.testing_ import Tester
from defl.lsblk_ import *

tester = Tester(name=__file__)

def rec(d, ind=0):
    for i in d:
        # print('  ' * ind, i.name, [i.name for i in i.children])
        rec(i.children, ind + 1)

@tester.add()
def lsblk():
    r = Lsblk.Run()
    rec(r.toList())
    # log.info(r)
    # log.info('r', lambda jc: jc.r)
    assert isinstance(r, LsblkCollection)
    k = next(r.toList())
    assert isinstance(k, Lsblk)
    assert isinstance(k.children, list)
    for i in r.toList(r):
        i.path

    a = r.filter(lambda x: x.diskTree.has(Path('/dev/sda')))
    Assert(next(a).path) == '/dev/sda'

    a = r.filter(lambda x: x.diskTree.has(Path('/dev/disk/by-partlabel/efi')))
    Assert(next(a).path) == '/dev/sda1'

    a = r.filter(lambda x: x.diskTree.has(Path('/dev/disk/by-diskseq/1-part1')))
    Assert(next(a).path) == '/dev/sda1'

    # for i in list(itertools.chain(*([x for x in y] for y in r))):
    #     print(i.path)
    # da

    # for k, v in defl.getObjAsDict(r[0]).items():
    #     print(k, type(v))

# @tester.add()
# def df():
#     r = Lsblk.Run()
#     d = r.df()
#     li(d)

@tester.add()
def diskTree():
    dt = DiskTree(root=Path(1), byId=[Path(2)])
    Assert(dt.has(Path(1))) == 'root'
    Assert(dt.has(Path(2))) == 'byId'
    a = DiskTree.Run()
    log.info('a', lambda x: x.a)

log.info(tester.run())
tester.exitWithStatus()

# for i in ch:
#     print(defl.printTable([[y for y in x] for x in ch], squareFill=True))