Imports System.Runtime.InteropServices

Imports StaxRip.UI

Public Class TaskDialogBaseForm
    Overridable Sub AdjustSize()
    End Sub

    Public Sub New()
        InitializeComponent()
        ApplyTheme()

        AddHandler InputTextEdit.TextBox.KeyDown, AddressOf InputTextEditTextBoxKeyDown
        AddHandler ThemeManager.CurrentThemeChanged, AddressOf OnThemeChanged
    End Sub

    Protected Overrides Sub Dispose(disposing As Boolean)
        RemoveHandler ThemeManager.CurrentThemeChanged, AddressOf OnThemeChanged
        components?.Dispose()
        MyBase.Dispose(disposing)
    End Sub

    Sub OnThemeChanged(theme As Theme)
        ApplyTheme(theme)
    End Sub

    Sub ApplyTheme()
        ApplyTheme(ThemeManager.CurrentTheme)
    End Sub

    Sub ApplyTheme(theme As Theme)
        If DesignHelp.IsDesignMode Then
            Exit Sub
        End If

        SuspendLayout()
        BackColor = theme.TaskDialog.BackColor
        ForeColor = theme.TaskDialog.ForeColor
        blCopyMessage.BackColor = BackColor
        blDetails.BackColor = BackColor
        ResumeLayout()
    End Sub

    <DllImport("user32.dll", EntryPoint:="SetWindowLong")>
    Shared Function SetWindowLong32(hWnd As IntPtr, nIndex As Integer, dwNewLong As Integer) As Integer
    End Function

    <DllImport("user32.dll", EntryPoint:="SetWindowLongPtr")>
    Shared Function SetWindowLongPtr64(hWnd As IntPtr, nIndex As Integer, dwNewLong As IntPtr) As IntPtr
    End Function

    Shared Function SetWindowLongPtr(hWnd As IntPtr, nIndex As Integer, dwNewLong As IntPtr) As IntPtr
        If IntPtr.Size = 8 Then
            Return SetWindowLongPtr64(hWnd, nIndex, dwNewLong)
        Else
            Return New IntPtr(SetWindowLong32(hWnd, nIndex, dwNewLong.ToInt32))
        End If
    End Function

    Class TaskDialogPanel
        Inherits PanelEx

        Property Form As TaskDialogBaseForm
        Property LineBreaks As Integer

        Protected Overrides Sub OnLayout(levent As LayoutEventArgs)
            MyBase.OnLayout(levent)

            If DesignMode OrElse Controls.Count = 0 Then
                MyBase.OnLayout(levent)
                Exit Sub
            End If

            If Form Is Nothing Then
                Form = DirectCast(FindForm(), TaskDialogBaseForm)
            End If

            Dim fh = FontHeight
            Dim previous As Control = Nothing

            Using g = CreateGraphics()
                For x = 0 To Controls.Count - 1
                    Dim c = Controls(x)

                    If x = 0 Then
                        c.Top = 0
                    Else
                        c.Top = previous.Top + previous.Height + CInt(fh * 0.2)
                    End If

                    c.Left = CInt(fh * 0.7)
                    c.Width = ClientSize.Width - CInt(fh * 0.7 * 2)

                    If TypeOf c Is LabelEx Then
                        Dim isExtended = Form.blDetails.Text = "Hide Details"

                        If c.Name = "Information" AndAlso isExtended OrElse c.Name = "ExpandedInformation" AndAlso Not isExtended Then
                            c.Visible = False
                            c.Height = 0
                        Else
                            c.Visible = True
                            Dim sz = g.MeasureString(c.Text, c.Font, c.Width)
                            c.Height = CInt(sz.Height + fh / 2)
                        End If
                    End If

                    If TryCast(c, CommandButton)?.AdjustSize() Then
                        LineBreaks += 1
                    End If

                    previous = c
                Next
            End Using
        End Sub
    End Class

    Class CommandButton
        Inherits Button

        Property Title As String
        Property Description As String

        Private _titleFont As Font

        ReadOnly Property TitleFont As Font
            Get
                If _titleFont Is Nothing Then _titleFont = FontManager.GetDefaultFont(1.5)
                Return _titleFont
            End Get
        End Property

        Sub New()
            If Not DesignHelp.IsDesignMode Then
                MinimumSize = New Size(20, 15)
                UseCompatibleTextRendering = False
                UseVisualStyleBackColor = False
                FlatStyle = FlatStyle.Flat
            End If

            FlatAppearance.BorderSize = 2

            ApplyTheme()

            AddHandler ThemeManager.CurrentThemeChanged, AddressOf OnThemeChanged
        End Sub

        Sub OnThemeChanged(theme As Theme)
            ApplyTheme(theme)
        End Sub

        Sub ApplyTheme()
            ApplyTheme(ThemeManager.CurrentTheme)
        End Sub

        Sub ApplyTheme(theme As Theme)
            If DesignHelp.IsDesignMode Then Exit Sub

            SuspendLayout()
            BackColor = theme.TaskDialog.CommandButton.BackColor
            ForeColor = theme.TaskDialog.CommandButton.ForeColor
            FlatAppearance.BorderColor = theme.TaskDialog.CommandButton.BorderColor
            ResumeLayout()
        End Sub


        Function AdjustSize() As Boolean
            Dim titleFontHeight = TitleFont.Height
            Dim h As Integer
            Dim hasLineBreak As Boolean

            If Title <> "" AndAlso Description <> "" Then
                Dim ts = GetTitleSize()
                Dim ds = GetDescriptionSize()
                h = CInt(titleFontHeight * 0.2 * 2) + ts.Height + ds.Height

                If titleFontHeight * 2 < ts.Height Then
                    hasLineBreak = True
                End If
            ElseIf Title <> "" Then
                Dim ts = GetTitleSize()
                h = CInt(titleFontHeight * 0.2 * 2) + ts.Height

                If titleFontHeight * 2 < ts.Height Then
                    hasLineBreak = True
                End If
            ElseIf Description <> "" Then
                Dim ds = GetDescriptionSize()
                h = CInt(titleFontHeight * 0.2 * 2) + ds.Height

                If FontHeight * 2 < ds.Height Then
                    hasLineBreak = True
                End If
            End If

            ClientSize = New Size(ClientSize.Width, h)
            Return hasLineBreak
        End Function

        Function GetTitleSize(Optional g1 As Graphics = Nothing) As Size
            If Title = "" Then Exit Function

            Dim g2 = If(g1, CreateGraphics())
            Dim tf = TitleFont.Height
            Dim w = ClientSize.Width - CInt(tf * 0.3 * 2)
            Dim sz = g2.MeasureString(Title, TitleFont, w)

            If g1 Is Nothing Then
                g2.Dispose()
            End If

            Return New Size(CInt(sz.Width), CInt(sz.Height))
        End Function

        Function GetDescriptionSize(Optional g1 As Graphics = Nothing) As Size
            If Description = "" Then Exit Function

            Dim g2 = If(g1, CreateGraphics())
            Dim tf = TitleFont.Height
            Dim w = ClientSize.Width - CInt(tf * 0.3 * 2)
            Dim sz = g2.MeasureString(Description, Font, w)

            If g1 Is Nothing Then
                g2.Dispose()
            End If

            Return New Size(CInt(sz.Width), CInt(sz.Height))
        End Function

        Protected Overrides Sub OnGotFocus(e As EventArgs)
            MyBase.OnGotFocus(e)
        End Sub

        Protected Overrides Sub OnLostFocus(e As EventArgs)
            MyBase.OnLostFocus(e)
        End Sub

        Protected Overrides Sub OnPaint(e As PaintEventArgs)
            MyBase.OnPaint(e)

            Dim g = e.Graphics
            g.TextRenderingHint = Drawing.Text.TextRenderingHint.AntiAlias
            Dim titleFontHeight = TitleFont.Height

            Dim x = CInt(titleFontHeight * 0.3)
            Dim y = CInt(titleFontHeight * 0.2)
            Dim w = ClientSize.Width - x * 2
            Dim h = ClientSize.Height - CInt(titleFontHeight * 0.2 * 2)
            Dim r = New Rectangle(x, y, w, h)

            If Title <> "" AndAlso Description <> "" Then
                TextRenderer.DrawText(g, Title, TitleFont, r, ForeColor, TextFormatFlags.Left Or TextFormatFlags.WordBreak Or TextFormatFlags.TextBoxControl)
                y = CInt(titleFontHeight * 0.2) + GetTitleSize().Height - 2
                r = New Rectangle(x, y, w, h)
                TextRenderer.DrawText(g, Description, Font, r, ForeColor, TextFormatFlags.Left Or TextFormatFlags.WordBreak Or TextFormatFlags.TextBoxControl)
            ElseIf Title <> "" Then
                TextRenderer.DrawText(g, Title, TitleFont, r, ForeColor, TextFormatFlags.Left Or TextFormatFlags.WordBreak Or TextFormatFlags.TextBoxControl)
            ElseIf Description <> "" Then
                TextRenderer.DrawText(g, Description, Font, r, ForeColor, TextFormatFlags.Left Or TextFormatFlags.WordBreak Or TextFormatFlags.TextBoxControl)
            End If
        End Sub

        Protected Overrides Sub Dispose(disposing As Boolean)
            RemoveHandler ThemeManager.CurrentThemeChanged, AddressOf OnThemeChanged
            TitleFont.Dispose()
            MyBase.Dispose(disposing)
        End Sub
    End Class

    Sub blDetails_Click(sender As Object, e As EventArgs) Handles blDetails.Click
        paMain.ScrollControlIntoView(paMain.Controls(0))

        If blDetails.Text = "Show Details" Then
            blDetails.Text = "Hide Details"
        Else
            Width = FontHeight * 22
            blDetails.Text = "Show Details"
        End If

        paMain.PerformLayout()
        AdjustSize()
        AdjustSize()
    End Sub

    Sub InputTextEditTextBoxKeyDown(sender As Object, e As KeyEventArgs)
        If e.KeyData = Keys.Enter AndAlso AcceptButton IsNot Nothing Then
            AcceptButton.PerformClick()
        End If
    End Sub
End Class