<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
    "http://mybatis.org/dtd/mybatis-3-mapper.dtd" >

<!--
    Licensed to the Apache Software Foundation (ASF) under one
    or more contributor license agreements.  See the NOTICE file
    distributed with this work for additional information
    regarding copyright ownership.  The ASF licenses this file
    to you under the Apache License, Version 2.0 (the
    "License"); you may not use this file except in compliance
    with the License.  You may obtain a copy of the License at

      http://www.apache.org/licenses/LICENSE-2.0

    Unless required by applicable law or agreed to in writing,
    software distributed under the License is distributed on an
    "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
    KIND, either express or implied.  See the License for the
    specific language governing permissions and limitations
    under the License.
-->

<mapper namespace="org.apache.guacamole.auth.jdbc.connectiongroup.ConnectionGroupMapper" >

    <!-- Result mapper for connection objects -->
    <resultMap id="ConnectionGroupResultMap" type="org.apache.guacamole.auth.jdbc.connectiongroup.ConnectionGroupModel" >

        <!-- Connection group properties -->
        <id     column="connection_group_id"      property="objectID"               jdbcType="INTEGER"/>
        <result column="connection_group_name"    property="name"                   jdbcType="VARCHAR"/>
        <result column="parent_id"                property="parentIdentifier"       jdbcType="INTEGER"/>
        <result column="type"                     property="type"                   jdbcType="VARCHAR"
                javaType="org.apache.guacamole.net.auth.ConnectionGroup$Type"/>
        <result column="max_connections"          property="maxConnections"         jdbcType="INTEGER"/>
        <result column="max_connections_per_user" property="maxConnectionsPerUser"  jdbcType="INTEGER"/>
        <result column="enable_session_affinity"  property="sessionAffinityEnabled" jdbcType="BOOLEAN"/>

        <!-- Child connection groups -->
        <collection property="connectionGroupIdentifiers" resultSet="childConnectionGroups" ofType="java.lang.String"
                    column="connection_group_id" foreignColumn="parent_id">
            <result column="connection_group_id"/>
        </collection>

        <!-- Child connections -->
        <collection property="connectionIdentifiers" resultSet="childConnections" ofType="java.lang.String"
                    column="connection_group_id" foreignColumn="parent_id">
            <result column="connection_id"/>
        </collection>

        <!-- Arbitrary attributes -->
        <collection property="arbitraryAttributes" resultSet="arbitraryAttributes"
                    ofType="org.apache.guacamole.auth.jdbc.base.ArbitraryAttributeModel"
                    column="connection_group_id" foreignColumn="connection_group_id">
            <result property="name"     column="attribute_name"  jdbcType="VARCHAR"/>
            <result property="value"    column="attribute_value" jdbcType="VARCHAR"/>
        </collection>

    </resultMap>

    <!-- Select all connection group identifiers -->
    <select id="selectIdentifiers" resultType="string">
        SELECT connection_group_id 
        FROM guacamole_connection_group
    </select>

    <!--
      * SQL fragment which lists the IDs of all connection groups readable by
      * the entity having the given entity ID. If group identifiers are
      * provided, the IDs of the entities for all groups having those
      * identifiers are tested, as well. Disabled groups are ignored.
      *
      * @param entityID
      *     The ID of the specific entity to test against.
      *
      * @param groups
      *     A collection of group identifiers to additionally test against.
      *     Though this functionality is optional, a collection must always be
      *     given, even if that collection is empty.
      -->
    <sql id="getReadableIDs">
        SELECT DISTINCT connection_group_id
        FROM guacamole_connection_group_permission
        WHERE
            <include refid="org.apache.guacamole.auth.jdbc.base.EntityMapper.isRelatedEntity">
                <property name="column"          value="entity_id"/>
                <property name="entityID"        value="${entityID}"/>
                <property name="groups"          value="${groups}"/>
                <property name="caseSensitivity" value="${caseSensitivity}"/>
            </include>
            AND permission = 'READ'
    </sql>

    <!-- Select identifiers of all readable connection groups -->
    <select id="selectReadableIdentifiers" resultType="string">
        <include refid="org.apache.guacamole.auth.jdbc.connectiongroup.ConnectionGroupMapper.getReadableIDs">
            <property name="entityID"        value="#{user.entityID,jdbcType=INTEGER}"/>
            <property name="groups"          value="effectiveGroups"/>
            <property name="caseSensitivity" value="${caseSensitivity}"/>
        </include>
    </select>

    <!-- Select all connection identifiers within a particular connection group -->
    <select id="selectIdentifiersWithin" resultType="string">
        SELECT connection_group_id 
        FROM guacamole_connection_group
        WHERE
            <if test="parentIdentifier != null">parent_id = #{parentIdentifier,jdbcType=INTEGER}::integer</if>
            <if test="parentIdentifier == null">parent_id IS NULL</if>
    </select>

    <!-- Select identifiers of all readable connection groups within a particular connection group -->
    <select id="selectReadableIdentifiersWithin" resultType="string">
        SELECT guacamole_connection_group.connection_group_id
        FROM guacamole_connection_group
        WHERE
            <if test="parentIdentifier != null">parent_id = #{parentIdentifier,jdbcType=INTEGER}::integer</if>
            <if test="parentIdentifier == null">parent_id IS NULL</if>
            AND connection_group_id IN (
                <include refid="org.apache.guacamole.auth.jdbc.connectiongroup.ConnectionGroupMapper.getReadableIDs">
                    <property name="entityID"        value="#{user.entityID,jdbcType=INTEGER}"/>
                    <property name="groups"          value="effectiveGroups"/>
                    <property name="caseSensitivity" value="${caseSensitivity}"/>
                </include>
            )
    </select>

    <!-- Select multiple connection groups by identifier -->
    <select id="select" resultMap="ConnectionGroupResultMap"
            resultSets="connectionGroups,childConnectionGroups,childConnections,arbitraryAttributes">

        SELECT
            connection_group_id,
            connection_group_name,
            parent_id,
            type,
            max_connections,
            max_connections_per_user,
            enable_session_affinity
        FROM guacamole_connection_group
        WHERE connection_group_id IN
            <foreach collection="identifiers" item="identifier"
                     open="(" separator="," close=")">
                #{identifier,jdbcType=INTEGER}::integer
            </foreach>;

        SELECT parent_id, connection_group_id
        FROM guacamole_connection_group
        WHERE parent_id IN
            <foreach collection="identifiers" item="identifier"
                     open="(" separator="," close=")">
                #{identifier,jdbcType=INTEGER}::integer
            </foreach>;

        SELECT parent_id, connection_id
        FROM guacamole_connection
        WHERE parent_id IN
            <foreach collection="identifiers" item="identifier"
                     open="(" separator="," close=")">
                #{identifier,jdbcType=INTEGER}::integer
            </foreach>;

        SELECT
            connection_group_id,
            attribute_name,
            attribute_value
        FROM guacamole_connection_group_attribute
        WHERE connection_group_id IN
            <foreach collection="identifiers" item="identifier"
                     open="(" separator="," close=")">
                #{identifier,jdbcType=INTEGER}::integer
            </foreach>;

    </select>

    <!-- Select multiple connection groups by identifier only if readable -->
    <select id="selectReadable" resultMap="ConnectionGroupResultMap"
            resultSets="connectionGroups,childConnectionGroups,childConnections,arbitraryAttributes">

        SELECT
            guacamole_connection_group.connection_group_id,
            connection_group_name,
            parent_id,
            type,
            max_connections,
            max_connections_per_user,
            enable_session_affinity
        FROM guacamole_connection_group
        WHERE guacamole_connection_group.connection_group_id IN
            <foreach collection="identifiers" item="identifier"
                     open="(" separator="," close=")">
                #{identifier,jdbcType=INTEGER}::integer
            </foreach>
            AND guacamole_connection_group.connection_group_id IN (
                <include refid="org.apache.guacamole.auth.jdbc.connectiongroup.ConnectionGroupMapper.getReadableIDs">
                    <property name="entityID"        value="#{user.entityID,jdbcType=INTEGER}"/>
                    <property name="groups"          value="effectiveGroups"/>
                    <property name="caseSensitivity" value="${caseSensitivity}"/>
                </include>
            );

        SELECT parent_id, guacamole_connection_group.connection_group_id
        FROM guacamole_connection_group
        WHERE parent_id IN
            <foreach collection="identifiers" item="identifier"
                     open="(" separator="," close=")">
                #{identifier,jdbcType=INTEGER}::integer
            </foreach>
            AND guacamole_connection_group.connection_group_id IN (
                <include refid="org.apache.guacamole.auth.jdbc.connectiongroup.ConnectionGroupMapper.getReadableIDs">
                    <property name="entityID"        value="#{user.entityID,jdbcType=INTEGER}"/>
                    <property name="groups"          value="effectiveGroups"/>
                    <property name="caseSensitivity" value="${caseSensitivity}"/>
                </include>
            );

        SELECT parent_id, guacamole_connection.connection_id
        FROM guacamole_connection
        WHERE parent_id IN
            <foreach collection="identifiers" item="identifier"
                     open="(" separator="," close=")">
                #{identifier,jdbcType=INTEGER}::integer
            </foreach>
            AND guacamole_connection.connection_id IN (
                <include refid="org.apache.guacamole.auth.jdbc.connection.ConnectionMapper.getReadableIDs">
                    <property name="entityID"        value="#{user.entityID,jdbcType=INTEGER}"/>
                    <property name="groups"          value="effectiveGroups"/>
                    <property name="caseSensitivity" value="${caseSensitivity}"/>
                </include>
            );

        SELECT
            guacamole_connection_group_attribute.connection_group_id,
            attribute_name,
            attribute_value
        FROM guacamole_connection_group_attribute
        WHERE guacamole_connection_group_attribute.connection_group_id IN
            <foreach collection="identifiers" item="identifier"
                     open="(" separator="," close=")">
                #{identifier,jdbcType=INTEGER}::integer
            </foreach>
            AND guacamole_connection_group_attribute.connection_group_id IN (
                <include refid="org.apache.guacamole.auth.jdbc.connectiongroup.ConnectionGroupMapper.getReadableIDs">
                    <property name="entityID"        value="#{user.entityID,jdbcType=INTEGER}"/>
                    <property name="groups"          value="effectiveGroups"/>
                    <property name="caseSensitivity" value="${caseSensitivity}"/>
                </include>
            );

    </select>

    <!-- Select single connection group by name -->
    <select id="selectOneByName" resultMap="ConnectionGroupResultMap">

        SELECT
            connection_group_id,
            connection_group_name,
            parent_id,
            type,
            max_connections,
            max_connections_per_user,
            enable_session_affinity
        FROM guacamole_connection_group
        WHERE 
            <if test="parentIdentifier != null">parent_id = #{parentIdentifier,jdbcType=INTEGER}::integer</if>
            <if test="parentIdentifier == null">parent_id IS NULL</if>
            AND connection_group_name = #{name,jdbcType=VARCHAR}

    </select>

    <!-- Delete single connection group by identifier -->
    <delete id="delete">
        DELETE FROM guacamole_connection_group
        WHERE connection_group_id = #{identifier,jdbcType=INTEGER}::integer
    </delete>

    <!-- Insert single connection -->
    <insert id="insert" useGeneratedKeys="true" keyProperty="object.objectID"
            parameterType="org.apache.guacamole.auth.jdbc.connectiongroup.ConnectionGroupModel">

        INSERT INTO guacamole_connection_group (
            connection_group_name,
            parent_id,
            type,
            max_connections,
            max_connections_per_user,
            enable_session_affinity
        )
        VALUES (
            #{object.name,jdbcType=VARCHAR},
            #{object.parentIdentifier,jdbcType=INTEGER}::integer,
            #{object.type,jdbcType=VARCHAR}::guacamole_connection_group_type,
            #{object.maxConnections,jdbcType=INTEGER},
            #{object.maxConnectionsPerUser,jdbcType=INTEGER},
            #{object.sessionAffinityEnabled,jdbcType=BOOLEAN}
        )

    </insert>

    <!-- Update single connection group -->
    <update id="update" parameterType="org.apache.guacamole.auth.jdbc.connectiongroup.ConnectionGroupModel">
        UPDATE guacamole_connection_group
        SET connection_group_name    = #{object.name,jdbcType=VARCHAR},
            parent_id                = #{object.parentIdentifier,jdbcType=INTEGER}::integer,
            type                     = #{object.type,jdbcType=VARCHAR}::guacamole_connection_group_type,
            max_connections          = #{object.maxConnections,jdbcType=INTEGER},
            max_connections_per_user = #{object.maxConnectionsPerUser,jdbcType=INTEGER},
            enable_session_affinity  = #{object.sessionAffinityEnabled,jdbcType=BOOLEAN}
        WHERE connection_group_id = #{object.objectID,jdbcType=INTEGER}::integer
    </update>

    <!-- Delete attributes associated with connection group -->
    <delete id="deleteAttributes">
        DELETE FROM guacamole_connection_group_attribute
        WHERE connection_group_id = #{object.objectID,jdbcType=INTEGER}
    </delete>

    <!-- Insert attributes for connection group -->
    <insert id="insertAttributes" parameterType="org.apache.guacamole.auth.jdbc.base.ArbitraryAttributeModel">
        INSERT INTO guacamole_connection_group_attribute (
            connection_group_id,
            attribute_name,
            attribute_value
        )
        VALUES
            <foreach collection="object.arbitraryAttributes" item="attribute" separator=",">
                (#{object.objectID,jdbcType=INTEGER},
                 #{attribute.name,jdbcType=VARCHAR},
                 #{attribute.value,jdbcType=VARCHAR})
            </foreach>
    </insert>

</mapper>
