styleChecker.fsx

#!/usr/bin/env -S dotnet fsi

#r "nuget: Fsdk, Version=0.6.0--date20230326-0544.git-5c4f55b"
#load "../src/FileConventions/Helpers.fs"

open System
open System.IO

open Fsdk
open Fsdk.Process

open Helpers

let fantomlessToolVersion = "4.7.997-prerelease"
let prettierVersion = "2.8.3"

let StyleFSharpFiles() =
    Process
        .Execute(
            {
                Command = "dotnet"
                Arguments = "new tool-manifest --force"
            },
            Echo.Off
        )
        .UnwrapDefault()
    |> ignore

    Process
        .Execute(
            {
                Command = "dotnet"
                Arguments =
                    $"tool install fantomless-tool --version {fantomlessToolVersion}"
            },
            Echo.Off
        )
        .UnwrapDefault()
    |> ignore

    Process
        .Execute(
            {
                Command = "dotnet"
                Arguments = "fantomless --recurse ."
            },
            Echo.Off
        )
        .UnwrapDefault()
    |> ignore

let RunPrettier(arguments: string) =

    // We need this step so we can change the files using `npx prettier --write` in the next step.
    // Otherwise we get permission denied error in the CI.
    Process.Execute(
        {
            Command = "chmod"
            Arguments = "777 -R ."
        },
        Echo.Off
    )
    |> ignore

    let processResult =
        Process.Execute(
            {
                Command = "npx"
                Arguments = $"prettier {arguments}"
            },
            Echo.Off
        )

    let errMsg =
        sprintf
            "Error when running '%s %s'"
            processResult.Details.Command
            processResult.Details.Args

    match processResult.Result with
    | Success output -> output
    | Error(_, output) ->
        if processResult.Details.Echo = Echo.Off then
            output.PrintToConsole()
            Console.WriteLine()
            Console.Out.Flush()

        Console.Error.WriteLine errMsg
        raise <| ProcessFailed errMsg
    | WarningsOrAmbiguous output ->
        if processResult.Details.Echo = Echo.Off then
            output.PrintToConsole()
            Console.WriteLine()
            Console.Out.Flush()

        let fullErrMsg = sprintf "%s (with warnings?)" errMsg
        fullErrMsg
    |> printfn "%A"


    // Since after installing commitlint dependencies package.json file changes, we need to
    // run the following command to ignore package.json file
    Process
        .Execute(
            {
                Command = "git"
                Arguments = "restore package.json"
            },
            Echo.Off
        )
        .UnwrapDefault()
    |> ignore

let StyleTypeScriptFiles() =
    RunPrettier "--quote-props=consistent --write ./**/*.ts"

let StyleYmlFiles() =
    RunPrettier "--quote-props=consistent --write ./**/*.yml"

let ContainsFiles (rootDir: DirectoryInfo) (searchPattern: string) =
    Helpers.GetFiles rootDir searchPattern |> Seq.length > 0

let GitDiff() : ProcessResult =

    // Since we changed file modes in the prettier step we need the following command to
    // make git ignore mode changes in files and doesn't include them in the git diff command.
    Process.Execute(
        {
            Command = "git"
            Arguments = "config core.fileMode false"
        },
        Echo.Off
    )
    |> ignore

    let processResult =
        Process.Execute(
            {
                Command = "git"
                Arguments = "diff --exit-code"
            },
            Echo.Off
        )

    processResult

let GitRestore() =
    Process.Execute(
        {
            Command = "git"
            Arguments = "restore ."
        },
        Echo.Off
    )
    |> ignore

let InstallPrettier(version: string) =
    Process.Execute(
        {
            Command = "npm"
            Arguments = $"install prettier@{version}"
        },
        Echo.Off
    )
    |> ignore

let PrintProcessResult (processResult: ProcessResult) (suggestion: string) =
    let errMsg =
        sprintf
            "Error when running '%s %s'"
            processResult.Details.Command
            processResult.Details.Args

    match processResult.Result with
    | Success output -> output
    | Error(_, output) ->
        if processResult.Details.Echo = Echo.Off then
            output.PrintToConsole()
            Console.WriteLine()
            Console.Out.Flush()

        let fullErrMsg = errMsg + Environment.NewLine + suggestion
        fullErrMsg

    | WarningsOrAmbiguous output ->
        if processResult.Details.Echo = Echo.Off then
            output.PrintToConsole()
            Console.WriteLine()
            Console.Out.Flush()

        let fullErrMsg = sprintf "%s (with warnings?)" errMsg
        fullErrMsg

    |> printfn "%A"

let GetProcessExitCode(processResult: ProcessResult) : int =
    match processResult.Result with
    | Success output -> 0
    | _ -> 1

let CheckStyleOfFSharpFiles(rootDir: DirectoryInfo) : int =
    let suggestion =
        "Please style your F# code using: `dotnet fantomless --recurse .`"

    GitRestore()

    let exitCode =
        if ContainsFiles rootDir "*.fs" || ContainsFiles rootDir ".fsx" then
            StyleFSharpFiles()
            let processResult = GitDiff()
            PrintProcessResult processResult suggestion
            GetProcessExitCode processResult

        else
            0

    exitCode

let CheckStyleOfTypeScriptFiles(rootDir: DirectoryInfo) : int =
    let suggestion =
        "Please style your TypeScript code using: `npx prettier --quote-props=consistent --write ./**/*.ts`"

    GitRestore()

    let exitCode =
        if ContainsFiles rootDir "*.ts" then
            InstallPrettier(prettierVersion)
            StyleTypeScriptFiles()
            let processResult = GitDiff()
            PrintProcessResult processResult suggestion
            GetProcessExitCode processResult

        else
            0

    exitCode

let CheckStyleOfYmlFiles(rootDir: DirectoryInfo) : int =
    let suggestion =
        "Please style your YML code using: `npx prettier --quote-props=consistent --write ./**/*.yml`"

    GitRestore()

    let exitCode =
        if ContainsFiles rootDir "*.yml" then
            InstallPrettier(prettierVersion)
            StyleYmlFiles()
            let processResult = GitDiff()
            PrintProcessResult processResult suggestion
            GetProcessExitCode processResult
        else
            0

    exitCode


let rootDir = Path.Combine(__SOURCE_DIRECTORY__, "..") |> DirectoryInfo

let exitCodes =
    [|
        CheckStyleOfFSharpFiles rootDir
        CheckStyleOfTypeScriptFiles rootDir
        CheckStyleOfYmlFiles rootDir
    |]

if exitCodes |> Seq.contains 1 then
    Environment.Exit 1