Chat Privado

Imports System.Data
Imports System.Data.OleDb
Public Class chat_privado
    Inherits System.Web.UI.Page

    Dim Conexion As Conexion = New Conexion()
    Dim DA As OleDbDataAdapter
    Dim DS As DataSet
    Dim CMD As OleDbCommand
    Dim Tabla As DataTable

    Dim Idprivado As Integer

    Dim Usuario As String
    Dim Privado As String

    Protected Sub Page_Load(ByVal sender As Object, ByVal e As System.EventArgs) Handles Me.Load
        Usuario = Request.QueryString("usuario")
        Privado = Request.QueryString("privado")

        lblconectado.Text = Usuario
        lblprivado.Text = Privado

    End Sub

    Public Function Extraer_Idprivado(ByVal Id_usuario As Integer, ByVal Id_privado As Integer) As Integer

        Dim id As Integer = 0

        Dim userID_R As DataTable = New DataTable
        Dim userID_D As DataTable = New DataTable

        userID_R = Verificar_R(Id_usuario)

        userID_D = Verificar_D(Id_usuario)

        If (userID_R.Rows.Count = 1) Then

            Tabla = New DataTable()

            Tabla = Buscar_PrivadoR(Id_usuario, Id_privado)


            If (Tabla.Rows.Count = 1) Then

                id = Tabla.Rows(0)("Idprivado")

            End If


        End If



        If (userID_D.Rows.Count = 1) Then

            Tabla = New DataTable()

            Tabla = Buscar_PrivadoD(Id_usuario, Id_privado)

            If (Tabla.Rows.Count = 1) Then

                id = Tabla.Rows(0)("Idprivado")

            End If


        End If






        Return id



    End Function
    Public Function Extraer_userID(ByVal usuario As String) As DataTable


        Dim consulta As String = "SELECT userID FROM usuarios WHERE Usuario='" + usuario.ToString() + "'"


        DA = New OleDbDataAdapter(consulta, Conexion.DB)

        DS = New DataSet()
        DS.Reset()

        DA.Fill(DS)

        Tabla = New DataTable()
        Tabla = DS.Tables(0)


        Return Tabla

    End Function


    Public Function Verificar_R(ByVal userID_R As Integer) As DataTable

        Dim consulta As String = "SELECT Idprivado FROM chat_privado WHERE userID_R=" & userID_R.ToString()

        DA = New OleDbDataAdapter(consulta, Conexion.DB)

        DS = New DataSet()
        DS.Reset()

        DA.Fill(DS)

        Tabla = New DataTable()
        Tabla = DS.Tables(0)


        Return Tabla



    End Function
    Public Function Verificar_D(ByVal userID_D As Integer) As DataTable

        Dim consulta As String = "select idprivado from chat_privado where userID_D=" & userID_D.ToString()


        DA = New OleDbDataAdapter(consulta, Conexion.DB)

        DS = New DataSet()
        DS.Reset()

        DA.Fill(DS)

        Tabla = New DataTable()
        Tabla = DS.Tables(0)


        Return Tabla



    End Function

    Public Function Buscar_PrivadoR(ByVal userID_R As Integer, userID_D As Integer) As DataTable

        Dim consulta As String = "SELECT Idprivado FROM chat_privado WHERE userID_R=" & userID_R.ToString() & "AND userID_D=" & userID_D.ToString()


        DA = New OleDbDataAdapter(consulta, Conexion.DB)

        DS = New DataSet()
        DS.Reset()

        DA.Fill(DS)

        Tabla = New DataTable()
        Tabla = DS.Tables(0)


        Return Tabla



    End Function

    Public Function Buscar_PrivadoD(ByVal userID_D As Integer, userID_R As Integer) As DataTable

        Dim consulta As String = "SELECT Idprivado FROM chat_privado WHERE userID_D=" & userID_D.ToString() & "AND userID_R=" & userID_R.ToString()


        DA = New OleDbDataAdapter(consulta, Conexion.DB)

        DS = New DataSet()
        DS.Reset()

        DA.Fill(DS)

        Tabla = New DataTable()
        Tabla = DS.Tables(0)


        Return Tabla



    End Function

    Public Function Iniciar_Privado(ByVal userID_R As Integer, ByVal userID_D As Integer) As Integer
        Dim Estado As Integer

        Dim Insertar As String = "insert into chat_privado(userID_R,userID_D)values(@R,@D)"
        CMD = New OleDbCommand(Insertar, Conexion.DB)

        CMD.Parameters.AddWithValue("@R", userID_R)
        CMD.Parameters.AddWithValue("@D", userID_D)

        Conexion.Conectar()
        Estado = CMD.ExecuteNonQuery()

        Conexion.Desconectar()

        Return Estado


    End Function

    Public Function Insertar_Msg(ByVal Msg As String, ByVal Fecha As String, ByVal enviado_por As Integer, Idpriv As Integer) As Integer
        Dim Estado As Integer

        Dim Insertar As String = "insert into mensajes_privado(Mensaje,Fecha,Enviado_por,Idprivado)values(@Msg,@Fecha,@env_por,@Idprivado)"
        CMD = New OleDbCommand(Insertar, Conexion.DB)

        CMD.Parameters.AddWithValue("@Msg", Msg)
        CMD.Parameters.AddWithValue("@Fecha", Fecha)
        CMD.Parameters.AddWithValue("@env_por", enviado_por)
        CMD.Parameters.AddWithValue("@Idprivado", Idpriv)

        Conexion.Conectar()
        Estado = CMD.ExecuteNonQuery()

        Conexion.Desconectar()

        Return Estado


    End Function

    Protected Sub btnEnviar_Click(sender As Object, e As EventArgs) Handles btnEnviar.Click


        If txtEnviar.Text.Trim().Length = 0 Then
            lblescriba.Visible = True
            txtEnviar.Focus()
            txtMsg.Text = ""
        Else

            Dim msg As String
            Dim Insertar As Integer
            Dim Iniciar As Integer
            msg = txtEnviar.Text

            Dim tabla_usuario As DataTable = New DataTable()

            tabla_usuario = Extraer_userID(Usuario)


            Dim tabla_privado As DataTable = New DataTable()


            tabla_privado = Extraer_userID(Privado)

            Idprivado = Extraer_Idprivado(tabla_usuario.Rows(0)("userID"), tabla_privado.Rows(0)("userID"))


            If (Idprivado = 0) Then

                Iniciar = Iniciar_Privado(tabla_usuario.Rows(0)("userID"), tabla_privado.Rows(0)("userID"))

                If (Iniciar = 1) Then


                    Idprivado = Extraer_Idprivado(tabla_usuario.Rows(0)("userID"), tabla_privado.Rows(0)("userID"))

                    Insertar = Insertar_Msg(msg, DateTime.Now.ToString("dd/MM/yyyy"), tabla_usuario.Rows(0)("userID"), Idprivado)

                    If (Insertar = 1) Then

                        txtEnviar.Text = ""
                        txtMsg.Text = ""


                    End If


                End If

            Else


                Insertar = Insertar_Msg(msg, DateTime.Now.ToString("dd/MM/yyyy"), tabla_usuario.Rows(0)("userID"), Idprivado)

                If (Insertar = 1) Then

                    txtEnviar.Text = ""
                    txtMsg.Text = ""


                End If

            End If


            End If

    End Sub
End Class