#=----------------------------------------------------------------------------=#
#                            GNU GPL OS/K                                      #
#                                                                              #
#   Desc:   Project Makefile                                                   #
#                                                                              #
#                                                                              #
#   Copyright © 2018-2019 The OS/K Team                                        #
#                                                                              #
#   This file is part of OS/K.                                                 #
#                                                                              #
#   OS/K 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          #
#   any later version.                                                         #
#                                                                              #
#   OS/K 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.                               #
#                                                                              #
#   You should have received a copy of the GNU General Public License          #
#   along with OS/K.  If not, see <https://www.gnu.org/licenses/>.             #
#=----------------------------------------------------------------------------=#

## VARIABLES ----------------------------------------------------------------- #

#Programs
ASM=nasm
LD=ld
CCNAME=x86_64-elf-gcc

ASMFLAGS=-f elf64
LDFLAGS=-melf_x86_64
COPTIM=-O2
CWARNS=-Wall -Wextra -Werror=implicit-function-declaration
CINCLUDES=-Iinclude
CFLAGS1=-nostdlib -ffreestanding -mcmodel=large -std=gnu11
CFLAGS2= -c -g -mno-red-zone -mno-mmx -mno-sse -mno-sse2 -fno-strict-aliasing
CFLAGS=$(CFLAGS1) $(CFLAGS2)
KCC=$(CCNAME) $(COPTIM) $(CWARNS) $(CFLAGS) $(CINCLUDES) -D_OSK_SOURCE -D_KALEID_KERNEL

#Folders
MBRDIR=boot/grub
LOADERDIR=boot/loader
KALEIDDIR=kaleid
INCLUDEDIR=include
OBJDIR=build/obj
KOBJDIR=build/obj/kaleid
LOBJDIR=build/obj/boot
BINDIR=build/bin
BUILDDIR=build

#Color codes
CL='\033[0;32m'
CL2='\033[1;36m'
CL3='\033[0m'
NC='\033[1;37m'

.PHONY: all
all : OS/K

## KALEID MAKEFILE ----------------------------------------------------------- #

# Common objects
kal_com_obj= $(KOBJDIR)/atoi.o $(KOBJDIR)/ctype.o                              \
	 $(KOBJDIR)/itoa.o $(KOBJDIR)/mem.o                                    \
	 $(KOBJDIR)/rand.o $(KOBJDIR)/sprintf.o                                \
	 $(KOBJDIR)/status.o $(KOBJDIR)/string.o                               \
	 $(KOBJDIR)/strtol.o $(KOBJDIR)/argv.o                                 \
	 $(KOBJDIR)/prog.o $(KOBJDIR)/atol.o                                   \
	 $(KOBJDIR)/atou.o $(KOBJDIR)/atoul.o                                  \
	 $(KOBJDIR)/utoa.o $(KOBJDIR)/ltoa.o                                   \
	 $(KOBJDIR)/ultoa.o

$(KOBJDIR)/atoi.o: $(KALEIDDIR)/libc/atoi.c $(INCLUDEDIR)/*/*.h
	@$(KCC) -D_NEED_ATOI $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/atol.o: $(KALEIDDIR)/libc/atoi.c $(INCLUDEDIR)/*/*.h
	@$(KCC) -D_NEED_ATOL $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/atou.o: $(KALEIDDIR)/libc/atoi.c $(INCLUDEDIR)/*/*.h
	@$(KCC) -D_NEED_ATOU $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/atoul.o: $(KALEIDDIR)/libc/atoi.c $(INCLUDEDIR)/*/*.h
	@$(KCC) -D_NEED_ATOUL $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/ctype.o: $(KALEIDDIR)/libc/ctype.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/itoa.o: $(KALEIDDIR)/libc/itoa.c $(INCLUDEDIR)/*/*.h
	@$(KCC) -D_NEED_ITOA $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/ltoa.o: $(KALEIDDIR)/libc/itoa.c $(INCLUDEDIR)/*/*.h
	@$(KCC) -D_NEED_LTOA $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/utoa.o: $(KALEIDDIR)/libc/itoa.c $(INCLUDEDIR)/*/*.h
	@$(KCC) -D_NEED_UTOA $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/ultoa.o: $(KALEIDDIR)/libc/itoa.c $(INCLUDEDIR)/*/*.h
	@$(KCC) -D_NEED_ULTOA $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/mem.o: $(KALEIDDIR)/libc/mem.c $(INCLUDEDIR)/*/*.h
	@$(KCC) -fno-strict-aliasing $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/rand.o: $(KALEIDDIR)/libc/rand.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/sprintf.o: $(KALEIDDIR)/libc/sprintf.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/status.o: $(KALEIDDIR)/libc/status.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/string.o: $(KALEIDDIR)/libc/string.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/strtol.o: $(KALEIDDIR)/libc/strtol.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}

$(KOBJDIR)/argv.o: $(KALEIDDIR)/extras/argv.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/prog.o: $(KALEIDDIR)/extras/prog.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}

$(KOBJDIR)/kernel/buf.o: $(KALEIDDIR)/libbuf/buf.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/kernel/bput.o: $(KALEIDDIR)/libbuf/bput.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/kernel/bprint.o: $(KALEIDDIR)/libbuf/bprint.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}

# Kernel objects
kal_kern_obj= $(KOBJDIR)/kernel/cpuid.o $(KOBJDIR)/kernel/init.o           \
	  $(KOBJDIR)/kernel/table.o $(KOBJDIR)/kernel/cursor.o                 \
	  $(KOBJDIR)/kernel/log.o   $(KOBJDIR)/kernel/vga.o                    \
	  $(KOBJDIR)/kernel/panic.o $(KOBJDIR)/kernel/map.o                    \
	  $(KOBJDIR)/kernel/heap.o  $(KOBJDIR)/kernel/malloc.o                 \
	  $(KOBJDIR)/kernel/buf.o   $(KOBJDIR)/kernel/sched.o                  \
	  $(KOBJDIR)/kernel/bput.o  $(KOBJDIR)/kernel/bprint.o                 \
	  $(KOBJDIR)/kernel/gdt.o

$(KOBJDIR)/kernel/cpuid.o: $(KALEIDDIR)/kernel/cpu/cpuid.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/kernel/idt.o: $(KALEIDDIR)/kernel/cpu/idt.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/kernel/init.o: $(KALEIDDIR)/kernel/init/init.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/kernel/table.o: $(KALEIDDIR)/kernel/init/table.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/kernel/cursor.o: $(KALEIDDIR)/kernel/io/cursor.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/kernel/log.o: $(KALEIDDIR)/kernel/ke/log.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/kernel/vga.o: $(KALEIDDIR)/kernel/io/vga.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/kernel/panic.o: $(KALEIDDIR)/kernel/ke/panic.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/kernel/map.o: $(KALEIDDIR)/kernel/mm/map.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/kernel/heap.o: $(KALEIDDIR)/kernel/mm/heap.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/kernel/malloc.o: $(KALEIDDIR)/kernel/mm/malloc.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
$(KOBJDIR)/kernel/gdt.o: $(KALEIDDIR)/kernel/mm/gdt.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}
$(KOBJDIR)/kernel/sched.o: $(KALEIDDIR)/kernel/ps/sched.c $(INCLUDEDIR)/*/*.h
	@$(KCC) $< -o $@
	@echo ${CL2}[$@] ${CL}Compiled.${CL3}


## MISC MAKEFILE ------------------------------------------------------------- #
./ProjectTree: ./.stylehlp_sh
	@cat ./.stylehlp_sh > ./ProjectTree
	@echo "\n" >> ./ProjectTree
	@tree --dirsfirst >> ./ProjectTree
	@echo ${CL2}[$@] ${CL}Generated.${CL3}

## MAIN MAKEFILE ------------------------------------------------------------- #

.PHONY: test
test: all
	@qemu-system-x86_64 -m 4G -hda $(BUILDDIR)/bin/disk.img \
	-d cpu_reset,guest_errors,pcall,int 2> $(BUILDDIR)/qemu.log &
.PHONY: test32
test32: all
	@qemu-system-i386 -hda $(BUILDDIR)/bin/disk.img -d cpu_reset,guest_errors,pcall,int 2> $(BUILDDIR)/qemu.log &

.PHONY: debug
debug: all
	@qemu-system-x86_64 -m 64M -hda $(BUILDDIR)/bin/disk.img -no-reboot \
	-no-shutdown -d cpu_reset,guest_errors,pcall,int 2> $(BUILDDIR)/qemu.log &
	@ndisasm $(BINDIR)/kaleid -o 0x00100000 -b 64 > $(BUILDDIR)/kaleid64_disasm.asm
	@ndisasm $(BINDIR)/kaleid -o 0x00100000 -b 32 > $(BUILDDIR)/kaleid32_disasm.asm

.PHONY: gdb
gdb: all
	@qemu-system-x86_64 -m 64M -hda $(BUILDDIR)/bin/disk.img -no-reboot \
	-no-shutdown -d cpu_reset,guest_errors,pcall,int -s 2> $(BUILDDIR)/qemu.log &
	@ndisasm $(BINDIR)/kaleid -o 0x00100000 -b 64 > kaleid64_disasm.asm
	@ndisasm $(BINDIR)/kaleid -o 0x00100000 -b 32 > kaleid32_disasm.asm
	@gdb \
	-ex "set arch i386:x86-64:intel"   \
	-ex "target remote localhost:1234" \
	-ex "symbol-file $(BINDIR)/kaleid" \
	-ex "break BtStartKern"            \

.PHONY: install_mbr
install_mbr: $(BINDIR)/disk.img $(MBRDIR)/grub.cfg
	@mkdir -p $(BINDIR)/disk
	@echo ${CL2}[$@] ${NC}Installing MBR on image...${CL3}
	-@$(MBRDIR)/umount.sh $(BINDIR)/disk || /bin/true
	@$(MBRDIR)/grub-install.sh $(BINDIR)/disk.img $(BINDIR)/disk $(MBRDIR)/grub.cfg
	@rmdir $(BINDIR)/disk
	@echo ${CL2}[$@] ${CL}Success.${CL3}

.PHONY: clean
clean:
	-@$(MBRDIR)/umount.sh $(BINDIR)/disk || /bin/true
	@rm -Rvf ./ProjectTree
	@rm -Rvf $(BINDIR)/* $(OBJDIR)/*.o \
		$(OBJDIR)/*/*.o $(OBJDIR)/*/*.x86_64 $(OBJDIR)/*/*/*.o
	@echo ${CL2}[[$@]] ${CL}Cleaned.${CL3}

$(BINDIR)/kaleid: $(LOBJDIR)/kaleid.x86_64
	@echo ${CL2}[$@] ${NC}Objcopy...${CL3}
	@x86_64-elf-objcopy -I elf64-x86-64 -O elf32-i386 $(LOBJDIR)/kaleid.x86_64 $(BINDIR)/kaleid
	@echo ${CL2}[$@] ${CL}Success.${CL3}

$(LOBJDIR)/kaleid.x86_64: $(kal_kern_obj) $(kal_com_obj) $(LOBJDIR)/loader.o
	@echo ${CL2}[$@] ${NC}Linking kernel objects...${CL3}
	@$(LD) $(LDFLAGS) -T $(BUILDDIR)/kernel.ld	\
		$(LOBJDIR)/loader.o		\
		$(KOBJDIR)/*.o 		\
		$(KOBJDIR)/kernel/*.o	\
		-o $(LOBJDIR)/kaleid.x86_64
	@echo ${CL2}[$@] ${CL}Success.${CL3}

$(LOBJDIR)/loader.o: $(LOADERDIR)/loader.asm $(LOADERDIR)/*/*.inc
	@echo ${CL2}[$@] ${NC}Making loader...${CL3}
	@$(ASM) $(ASMFLAGS) $(LOADERDIR)/loader.asm -o $(LOBJDIR)/loader.o > /dev/null
	@echo ${CL2}[$@] ${CL}Success.${CL3}

$(BINDIR)/disk.img: $(MBRDIR)/create_disk.sh
	@echo ${CL2}[$@]${NC} Constructing disk image...${CL3}
	-@$(MBRDIR)/umount.sh $(BINDIR)/disk || /bin/true
	@$(MBRDIR)/create_disk.sh $(BINDIR)/disk.img
	@make install_mbr
	@echo ${CL2}[$@]${NC} Constructing disk image...${CL3}

.PHONY:OS/K
OS/K: $(BINDIR)/kaleid $(BINDIR)/disk.img ./ProjectTree
	@mkdir -p $(BINDIR)/disk
	@echo ${CL2}[[$@]] ${NC}Integrating kernel...${CL3}
	-@$(MBRDIR)/umount.sh $(BINDIR)/disk || /bin/true
	@$(MBRDIR)/mount.sh $(BINDIR)/disk.img $(BINDIR)/disk
	@cp $(BINDIR)/kaleid $(BINDIR)/disk/boot/kaleid
	@$(MBRDIR)/umount.sh $(BINDIR)/disk
	@echo ${CL2}[[$@]] ${CL}Success.${CL3}
	@rmdir $(BINDIR)/disk