# Maintainer: Kyle De'Vir (QuartzDragon) <kyle[dot]devir[at]mykolab[dot]com>
# Contributor: Jan Alexander Steffens (heftig) <jan.steffens@gmail.com>
# Contributor: Tobias Powalowski <tpowa@archlinux.org>
# Contributor: Thomas Baechler <thomas@archlinux.org>

### BUILD OPTIONS
# Set these variables to ANYTHING that is not null to enable them

# CPU sched_yield_type - Choose what sort of yield sched_yield will perform
# 0: No yield
# 1: Yield only to better priority/deadline tasks
# 2: Expire timeslice and recalculate deadline
_sched_yield_type=

# Tweak kernel options prior to a build via nconfig
_makenconfig=

# Optionally select a sub architecture by number if building in a clean chroot
# Leaving this entry blank will require user interaction during the build
# which will cause a failure to build if using makechrootpkg. Note that the
# generic (default) option is 27.
#
# Note - the march=native option is unavailable by this method, use the nconfig
# and manually select it.
#
#  1. AMD Opteron/Athlon64/Hammer/K8 (MK8)
#  2. AMD Opteron/Athlon64/Hammer/K8 with SSE3 (MK8SSE3)
#  3. AMD 61xx/7x50/PhenomX3/X4/II/K10 (MK10)
#  4. AMD Barcelona (MBARCELONA)
#  5. AMD Bobcat (MBOBCAT)
#  6. AMD Jaguar (MJAGUAR)
#  7. AMD Bulldozer (MBULLDOZER)
#  8. AMD Piledriver (MPILEDRIVER)
#  9. AMD Steamroller (MSTEAMROLLER)
#  10. AMD Excavator (MEXCAVATOR)
#  11. AMD Zen (MZEN)
#  12. AMD Zen 2 (MZEN2)
#  13. Intel P4 / older Netburst based Xeon (MPSC)
#  14. Intel Atom (MATOM)
#  15. Intel Core 2 (MCORE2)
#  16. Intel Nehalem (MNEHALEM)
#  17. Intel Westmere (MWESTMERE)
#  18. Intel Silvermont (MSILVERMONT)
#  19. Intel Sandy Bridge (MSANDYBRIDGE)
#  20. Intel Ivy Bridge (MIVYBRIDGE)
#  21. Intel Haswell (MHASWELL)
#  22. Intel Broadwell (MBROADWELL)
#  23. Intel Skylake (MSKYLAKE)
#  24. Intel Skylake X (MSKYLAKEX)
#  25. Intel Cannon Lake (MCANNONLAKE)
#  26. Intel Ice Lake (MICELAKE)
#  27. Generic-x86-64 (GENERIC_CPU)
#  28. Native optimizations autodetected by GCC (MNATIVE)
_subarch=

# Compile ONLY probed modules
# Build in only the modules that you currently have probed in your system VASTLY
# reducing the number of modules built and the build time.
#
# WARNING - ALL modules must be probed BEFORE you begin making the pkg!
#
# To keep track of which modules are needed for your specific system/hardware,
# give module_db script a try: https://aur.archlinux.org/packages/modprobed-db
# This PKGBUILD will call it directly to probe all the modules you have logged!
#
# More at this wiki page ---> https://wiki.archlinux.org/index.php/Modprobed-db
_localmodcfg=

pkgbase=linux-pds
_srcver_tag=5.2.8-arch1
pkgver="${_srcver_tag//-/.}"
pkgrel=1
arch=(x86_64)
url="https://git.archlinux.org/linux.git/log/?h=v$_srcver_tag"
license=(GPL2)
makedepends=(
    xmlto
    kmod
    inetutils
    bc
    libelf
    git
    python-sphinx
    python-sphinx_rtd_theme
    graphviz
    imagemagick
)
options=("!strip")

_reponame="linux"
_repo_url="https://git.archlinux.org/${_reponame}"

_reponame_gcc_patch="kernel_gcc_patch"
_repo_url_gcc_patch="https://github.com/graysky2/${_reponame_gcc_patch}"
_gcc_patch_name="enable_additional_cpu_optimizations_for_gcc_v9.1+_kernel_v4.13+.patch"

_pkgdesc_extra="~ featuring Alfred Chen's PDS CPU scheduler, rebased by TkG"

source=(
    "git+${_repo_url}?signed#tag=v${_srcver_tag}"
    "git+${_repo_url_gcc_patch}"
    config         # the main kernel config file
    60-linux.hook  # pacman hook for depmod
    90-linux.hook  # pacman hook for initramfs regeneration
    linux.preset   # standard config files for mkinitcpio ramdisk
    01-Undead-PDS-0.99o-rebase-by-TkG.patch
    02-Glitched-PDS-by-TkG.patch
)
validpgpkeys=(
    "ABAF11C65A2970B130ABE3C479BE3E4300411886"  # Linus Torvalds
    "647F28654894E3BD457199BE38DBBDC86092693E"  # Greg Kroah-Hartman
    "8218F88849AAC522E94CF470A5E9288C4FA415FA"  # Jan Alexander Steffens (heftig)
)
sha512sums=('SKIP'
            'SKIP'
            '34570a46bb4cee800923a9b1a50d97927daf1db9eb3a4b080f1626787bd33c65584d160ee4dcedf55cdbd8c2a8ebcab178d8590619fc37ed20c60b1813d424ac'
            '7ad5be75ee422dda3b80edd2eb614d8a9181e2c8228cd68b3881e2fb95953bf2dea6cbe7900ce1013c9de89b2802574b7b24869fc5d7a95d3cc3112c4d27063a'
            '2718b58dbbb15063bacb2bde6489e5b3c59afac4c0e0435b97fe720d42c711b6bcba926f67a8687878bd51373c9cf3adb1915a11666d79ccb220bf36e0788ab7'
            '2dc6b0ba8f7dbf19d2446c5c5f1823587de89f4e28e9595937dd51a87755099656f2acec50e3e2546ea633ad1bfd1c722e0c2b91eef1d609103d8abdc0a7cbaf'
            '7345e16b9b8731cb9961823a5e35f6785eaa5759f502d4f7829f5353985e5a27de3cec36752fb1a449d9376af31a4277bce576f0ae4421a8bc28fe21d259af46'
            '3ff796cbc213ae5f43a55f1ba92406bba04703db3459040beacacd9baceb3138021e908f440bd101cc76cb725e418ebdc8ab776327801690da30a1477bc84753')

_kernelname=${pkgbase#linux}
: ${_kernelname:=-ARCH}

prepare() {
    cd ${_reponame}

    msg2 "Setting version..."
    scripts/setlocalversion --save-scmversion
    echo "-${pkgrel}" > localversion.10-pkgrel
    echo "${_kernelname}" > localversion.20-pkgname

    # https://github.com/graysky2/kernel_gcc_patch
    msg2 "Patching to enabled additional gcc CPU optimizatons..."
    patch -Np1 -i "${srcdir}/${_reponame_gcc_patch}/${_gcc_patch_name}"

    # From https://github.com/Tk-Glitch/PKGBUILDS/tree/master/linux52-tkg/linux52-tkg-patches
    msg2 "Patching Undead PDS 0.99o 5.2 rebase by TkG"
    patch -Np1 -i "${srcdir}/01-Undead-PDS-0.99o-rebase-by-TkG.patch"
    patch -Np1 -i "${srcdir}/02-Glitched-PDS-by-TkG.patch"
    
    msg2 "Setting config..."
    cp ../config .config

    if [ -n "${_subarch}" ]; then
        yes "${_subarch}" | make oldconfig
    else
        make prepare
    fi

    ### Optionally load needed modules for the make localmodconfig
    # See https://aur.archlinux.org/packages/modprobed-db
    if [ -n "${_localmodcfg}" ]; then
        msg "If you have modprobed-db installed, running it in recall mode now"
        if [ -e /usr/bin/modprobed-db ]; then
            [[ -x /usr/bin/sudo ]] || {
            echo "Cannot call modprobe with sudo. Install sudo and configure it to work with this user."
            exit 1; }
            sudo /usr/bin/modprobed-db recall
            make localmodconfig
        fi
    fi

    if [ -z "${_sched_yield_type}" ]; then
        plain ""
        plain "CPU sched_yield_type - Choose what sort of yield sched_yield will perform."
        plain ""
        plain "For PDS:"
        plain "0: No yield."
        plain "1: Yield only to better priority/deadline tasks."
        plain "2: Expire timeslice and recalculate deadline."
        read -rp "`echo $'\n> 0 (Recommended option for gaming on PDS - "tkg" default)\n  1 (Default, but can lead to stability issues on some platforms)\n  2 (Usually the slowest option for PDS, not recommended unless you have issues with 0 or 1)\n  [0-2?]: '`" _sched_yield_type;
    fi
    if [ "${_sched_yield_type}" == "1" ]; then
        msg2 "Using default CPU sched yield type (1)"
    elif [ "${_sched_yield_type}" == "2" ]; then
        sed -i -e "s/int sched_yield_type __read_mostly = 1;/int sched_yield_type __read_mostly = 2;/" ./kernel/sched/pds.c
    else
        sed -i -e "s/int sched_yield_type __read_mostly = 1;/int sched_yield_type __read_mostly = 0;/" ./kernel/sched/pds.c
    fi

    # do not run "make olddefconfig" as it sets default options
    yes "" | make config >/dev/null

    make -s kernelrelease > ../version
    msg2 "Prepared %s version %s" "${pkgbase}" "$(<../version)"

    [[ -z "${_makenconfig}" ]] || make nconfig

    # save configuration for later reuse
    cat .config > "${startdir}/config.last"
}

build() {
    cd ${_reponame}
    make bzImage modules htmldocs
}

_package() {
    pkgdesc="The ${pkgbase/linux/Linux} kernel and modules ${_pkgdesc_extra}"
    depends=(
        coreutils
        linux-firmware
        kmod
        mkinitcpio
        thrash-protect
    )
    optdepends=("crda: to set the correct wireless channels of your country")
    provides=("${pkgbase}=${pkgver}")
    backup=("etc/mkinitcpio.d/${pkgbase}.preset")
    install=linux.install

    local kernver="$(<version)"
    local modulesdir="${pkgdir}/usr/lib/modules/${kernver}"

    cd ${_reponame} 

    msg2 "Installing boot image..."
    # systemd expects to find the kernel here to allow hibernation
    # https://github.com/systemd/systemd/commit/edda44605f06a41fb86b7ab8128dcf99161d2344
    install -Dm644 "$(make -s image_name)" "${modulesdir}/vmlinuz"
    install -Dm644 "${modulesdir}/vmlinuz" "${pkgdir}/boot/vmlinuz-${pkgbase}"

    msg2 "Installing modules..."
    make INSTALL_MOD_PATH="${pkgdir}/usr" modules_install

    # a place for external modules,
    # with version file for building modules and running depmod from hook
    local extramodules="extramodules${_kernelname}"
    local extradir="${pkgdir}/usr/lib/modules/${extramodules}"
    install -Dt "${extradir}" -m644 ../version
    ln -sr "${extradir}" "${modulesdir}/extramodules"

    # remove build and source links
    rm "${modulesdir}"/{source,build}

    msg2 "Installing hooks..."
    # sed expression for following substitutions
    local subst="
        s|%PKGBASE%|${pkgbase}|g
        s|%KERNVER%|${kernver}|g
        s|%EXTRAMODULES%|${extramodules}|g
    "

    # hack to allow specifying an initially nonexisting install file
    sed "${subst}" "${startdir}/${install}" > "${startdir}/${install}.pkg"
    true && install=${install}.pkg

    # fill in mkinitcpio preset and pacman hooks
    sed "${subst}" ../linux.preset | install -Dm644 /dev/stdin \
        "${pkgdir}/etc/mkinitcpio.d/${pkgbase}.preset"
    sed "${subst}" ../60-linux.hook | install -Dm644 /dev/stdin \
        "${pkgdir}/usr/share/libalpm/hooks/60-${pkgbase}.hook"
    sed "${subst}" ../90-linux.hook | install -Dm644 /dev/stdin \
        "${pkgdir}/usr/share/libalpm/hooks/90-${pkgbase}.hook"

    msg2 "Fixing permissions..."
    chmod -Rc u=rwX,go=rX "${pkgdir}"
}

_package-headers() {
    pkgdesc="Header files and scripts for building modules for ${pkgbase/linux/Linux} kernel ${_pkgdesc_extra}"
    depends=("${pkgbase}=${pkgver}")
    provides=(
        "${pkgbase}-headers=${pkgver}"
        "linux-headers=${pkgver}"
    )

    local builddir="${pkgdir}/usr/lib/modules/$(<version)/build"

    cd ${_reponame}

    msg2 "Installing build files..."
    install -Dt "${builddir}" -m644 Makefile .config Module.symvers System.map vmlinux
    install -Dt "${builddir}/kernel" -m644 kernel/Makefile
    install -Dt "${builddir}/arch/x86" -m644 arch/x86/Makefile
    cp -t "${builddir}" -a scripts

    # add objtool for external module building and enabled VALIDATION_STACK option
    install -Dt "${builddir}/tools/objtool" tools/objtool/objtool

    # add xfs and shmem for aufs building
    mkdir -p "${builddir}"/{fs/xfs,mm}

    # ???
    mkdir "${builddir}/.tmp_versions"

    msg2 "Installing headers..."
    cp -t "${builddir}" -a include
    cp -t "${builddir}/arch/x86" -a arch/x86/include
    install -Dt "${builddir}/arch/x86/kernel" -m644 arch/x86/kernel/asm-offsets.s

    install -Dt "${builddir}/drivers/md" -m644 drivers/md/*.h
    install -Dt "${builddir}/net/mac80211" -m644 net/mac80211/*.h

    # http://bugs.archlinux.org/task/13146
    install -Dt "${builddir}/drivers/media/i2c" -m644 drivers/media/i2c/msp3400-driver.h

    # http://bugs.archlinux.org/task/20402
    install -Dt "${builddir}/drivers/media/usb/dvb-usb" -m644 drivers/media/usb/dvb-usb/*.h
    install -Dt "${builddir}/drivers/media/dvb-frontends" -m644 drivers/media/dvb-frontends/*.h
    install -Dt "${builddir}/drivers/media/tuners" -m644 drivers/media/tuners/*.h

    msg2 "Installing KConfig files..."
    find . -name "Kconfig*" -exec install -Dm644 {} "${builddir}/{}" \;

    msg2 "Removing unneeded architectures..."
    local arch
    for arch in "${builddir}"/arch/*/; do
        [[ ${arch} = */x86/ ]] && continue
        echo "Removing $(basename "${arch}")"
        rm -r "${arch}"
    done

    msg2 "Removing documentation..."
    rm -r "${builddir}/Documentation"

    msg2 "Removing broken symlinks..."
    find -L "${builddir}" -type l -printf "Removing %P\n" -delete

    msg2 "Removing loose objects..."
    find "${builddir}" -type f -name "*.o" -printf "Removing %P\n" -delete

    msg2 "Stripping build tools..."
    local file
    while read -rd "" file; do
        case "$(file -bi "${file}")" in
            application/x-sharedlib\;*)      # Libraries (.so)
                strip -v $STRIP_SHARED "${file}" ;;
            application/x-archive\;*)        # Libraries (.a)
                strip -v $STRIP_STATIC "${file}" ;;
            application/x-executable\;*)     # Binaries
                strip -v $STRIP_BINARIES "${file}" ;;
            application/x-pie-executable\;*) # Relocatable binaries
                strip -v $STRIP_SHARED "${file}" ;;
        esac
    done < <(find "${builddir}" -type f -perm -u+x ! -name vmlinux -print0)

    msg2 "Adding symlink..."
    mkdir -p "${pkgdir}/usr/src"
    ln -sr "${builddir}" "${pkgdir}/usr/src/${pkgbase}-${pkgver}"

    msg2 "Fixing permissions..."
    chmod -Rc u=rwX,go=rX "${pkgdir}"
}

_package-docs() {
    pkgdesc="Kernel hackers manual - HTML documentation that comes with the ${pkgbase/linux/Linux} kernel ${_pkgdesc_extra}"
    depends=("${pkgbase}=${pkgver}")
    provides=(
        "${pkgbase}-docs=${pkgver}"
        "linux-docs=${pkgver}"
    )

    local builddir="${pkgdir}/usr/lib/modules/$(<version)/build"

    cd ${_reponame}

    msg2 "Installing documentation..."
    mkdir -p "${builddir}"
    cp -t "${builddir}" -a Documentation

    # msg2 "Removing doctrees..."
    # rm -r "${builddir}/Documentation/output/.doctrees"

    msg2 "Moving HTML docs..."
    local src dst
    while read -rd "" src; do
        dst="${builddir}/Documentation/${src#${builddir}/Documentation/output/}"
        mkdir -p "${dst%/*}"
        mv "${src}" "${dst}"
        rmdir -p --ignore-fail-on-non-empty "${src%/*}"
    done < <(find "${builddir}/Documentation/output" -type f -print0)

    msg2 "Adding symlink..."
    mkdir -p "${pkgdir}/usr/share/doc"
    ln -sr "${builddir}/Documentation" "${pkgdir}/usr/share/doc/${pkgbase}"

    msg2 "Fixing permissions..."
    chmod -Rc u=rwX,go=rX "${pkgdir}"
}

pkgname=(
    "${pkgbase}"
    "${pkgbase}-headers"
    "${pkgbase}-docs"
)
for _p in "${pkgname[@]}"; do
    eval "package_${_p}() {
        $(declare -f "_package${_p#${pkgbase}}")
        _package${_p#${pkgbase}}
    }"
done

# vim:set ts=8 sts=2 sw=2 et: