001// --------------------------------------------------------------------------------
002// Copyright 2002-2026 Echo Three, LLC
003//
004// Licensed under the Apache License, Version 2.0 (the "License");
005// you may not use this file except in compliance with the License.
006// You may obtain a copy of the License at
007//
008//     http://www.apache.org/licenses/LICENSE-2.0
009//
010// Unless required by applicable law or agreed to in writing, software
011// distributed under the License is distributed on an "AS IS" BASIS,
012// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013// See the License for the specific language governing permissions and
014// limitations under the License.
015// --------------------------------------------------------------------------------
016
017package com.echothree.model.control.subscription.server.graphql;
018
019import com.echothree.model.control.graphql.server.graphql.BaseEntityInstanceObject;
020import com.echothree.model.control.graphql.server.graphql.count.Connections;
021import com.echothree.model.control.graphql.server.graphql.count.CountedObjects;
022import com.echothree.model.control.graphql.server.graphql.count.CountingDataConnectionFetcher;
023import com.echothree.model.control.graphql.server.graphql.count.CountingPaginatedData;
024import com.echothree.model.control.graphql.server.util.BaseGraphQl;
025import com.echothree.model.control.graphql.server.util.count.ObjectLimiter;
026import com.echothree.model.control.subscription.server.control.SubscriptionControl;
027import com.echothree.model.control.user.server.control.UserControl;
028import com.echothree.model.data.subscription.common.SubscriptionTypeConstants;
029import com.echothree.model.data.subscription.server.entity.SubscriptionKind;
030import com.echothree.model.data.subscription.server.entity.SubscriptionKindDetail;
031import com.echothree.util.server.persistence.Session;
032import graphql.annotations.annotationTypes.GraphQLDescription;
033import graphql.annotations.annotationTypes.GraphQLField;
034import graphql.annotations.annotationTypes.GraphQLName;
035import graphql.annotations.annotationTypes.GraphQLNonNull;
036import graphql.annotations.connection.GraphQLConnection;
037import graphql.schema.DataFetchingEnvironment;
038import java.util.ArrayList;
039import java.util.stream.Collectors;
040
041@GraphQLDescription("subscription kind object")
042@GraphQLName("SubscriptionKind")
043public class SubscriptionKindObject
044        extends BaseEntityInstanceObject {
045    
046    private final SubscriptionKind subscriptionKind; // Always Present
047    
048    public SubscriptionKindObject(SubscriptionKind subscriptionKind) {
049        super(subscriptionKind.getPrimaryKey());
050        
051        this.subscriptionKind = subscriptionKind;
052    }
053
054    private SubscriptionKindDetail subscriptionKindDetail; // Optional, use getSubscriptionKindDetail()
055    
056    private SubscriptionKindDetail getSubscriptionKindDetail() {
057        if(subscriptionKindDetail == null) {
058            subscriptionKindDetail = subscriptionKind.getLastDetail();
059        }
060        
061        return subscriptionKindDetail;
062    }
063    
064    @GraphQLField
065    @GraphQLDescription("subscription kind name")
066    @GraphQLNonNull
067    public String getSubscriptionKindName() {
068        return getSubscriptionKindDetail().getSubscriptionKindName();
069    }
070    
071    @GraphQLField
072    @GraphQLDescription("is default")
073    @GraphQLNonNull
074    public boolean getIsDefault() {
075        return getSubscriptionKindDetail().getIsDefault();
076    }
077    
078    @GraphQLField
079    @GraphQLDescription("sort order")
080    @GraphQLNonNull
081    public int getSortOrder() {
082        return getSubscriptionKindDetail().getSortOrder();
083    }
084    
085    @GraphQLField
086    @GraphQLDescription("description")
087    @GraphQLNonNull
088    public String getDescription(final DataFetchingEnvironment env) {
089        var subscriptionControl = Session.getModelController(SubscriptionControl.class);
090        var userControl = Session.getModelController(UserControl.class);
091
092        return subscriptionControl.getBestSubscriptionKindDescription(subscriptionKind, userControl.getPreferredLanguageFromUserVisit(BaseGraphQl.getUserVisit(env)));
093    }
094
095    @GraphQLField
096    @GraphQLDescription("subscription types")
097    @GraphQLNonNull
098    @GraphQLConnection(connectionFetcher = CountingDataConnectionFetcher.class)
099    public CountingPaginatedData<SubscriptionTypeObject> getSubscriptionTypes(final DataFetchingEnvironment env) {
100        if(SubscriptionSecurityUtils.getHasSubscriptionTypesAccess(env)) {
101            var subscriptionControl = Session.getModelController(SubscriptionControl.class);
102            var totalCount = subscriptionControl.countSubscriptionTypesBySubscriptionKind(subscriptionKind);
103
104            try(var objectLimiter = new ObjectLimiter(env, SubscriptionTypeConstants.COMPONENT_VENDOR_NAME, SubscriptionTypeConstants.ENTITY_TYPE_NAME, totalCount)) {
105                var entities = subscriptionControl.getSubscriptionTypesBySubscriptionKind(subscriptionKind);
106                var subscriptionTypes = entities.stream().map(SubscriptionTypeObject::new).collect(Collectors.toCollection(() -> new ArrayList<>(entities.size())));
107
108                return new CountedObjects<>(objectLimiter, subscriptionTypes);
109            }
110        } else {
111            return Connections.emptyConnection();
112        }
113    }
114
115}