GitHawk/Classes/Issues/IssueViewModels.swift

//
//  IssueViewModels.swift
//  Freetime
//
//  Created by Ryan Nystrom on 5/19/17.
//  Copyright © 2017 Ryan Nystrom. All rights reserved.
//

import UIKit
import IGListKit

func titleStringSizing(title: String, width: CGFloat) -> NSAttributedStringSizing {
    let attributedString = NSAttributedString(
        string: title,
        attributes: [
            NSAttributedStringKey.font: Styles.Fonts.headline,
            NSAttributedStringKey.foregroundColor: Styles.Colors.Gray.dark.color
        ])
    return NSAttributedStringSizing(
        containerWidth: width,
        attributedText: attributedString,
        inset: IssueTitleCell.inset,
        backgroundColor: Styles.Colors.background
    )
}

func createIssueReactions(reactions: ReactionFields) -> IssueCommentReactionViewModel {
    var models = [ReactionViewModel]()

    for group in reactions.reactionGroups ?? [] {
        // do not display reactions for 0 count
        let count = group.users.totalCount
        guard count > 0 else { continue }

        let nodes: [String] = {
            guard let filtered = group.users.nodes?.filter({ $0?.login != nil }) as? [ReactionFields.ReactionGroup.User.Node] else {
                return []
            }

            return filtered.map({ $0.login })
        }()

        models.append(ReactionViewModel(content: group.content, count: count, viewerDidReact: group.viewerHasReacted, users: nodes))
    }

    return IssueCommentReactionViewModel(models: models)
}

func commentModelOptions(owner: String, repo: String) -> GitHubMarkdownOptions {
    return GitHubMarkdownOptions(owner: owner, repo: repo, flavors: [.issueShorthand, .usernames])
}

func createCommentModel(
    id: String,
    commentFields: CommentFields,
    reactionFields: ReactionFields,
    width: CGFloat,
    owner: String,
    repo: String,
    threadState: IssueCommentModel.ThreadState,
    viewerCanUpdate: Bool,
    viewerCanDelete: Bool,
    isRoot: Bool
    ) -> IssueCommentModel? {
    guard let author = commentFields.author,
        let date = commentFields.createdAt.githubDate,
        let avatarURL = URL(string: author.avatarUrl)
        else { return nil }

    let details = IssueCommentDetailsViewModel(
        date: date,
        login: author.login,
        avatarURL: avatarURL,
        didAuthor: commentFields.viewerDidAuthor,
        editedBy: commentFields.editor?.login,
        editedAt: commentFields.lastEditedAt?.githubDate
    )

    let options = commentModelOptions(owner: owner, repo: repo)
    let bodies = CreateCommentModels(
        markdown: commentFields.body,
        width: width,
        options: options,
        viewerCanUpdate: viewerCanUpdate
    )
    let reactions = createIssueReactions(reactions: reactionFields)
    let collapse = IssueCollapsedBodies(bodies: bodies, width: width)
    return IssueCommentModel(
        id: id,
        details: details,
        bodyModels: bodies,
        reactions: reactions,
        collapse: collapse,
        threadState: threadState,
        rawMarkdown: commentFields.body,
        viewerCanUpdate: viewerCanUpdate,
        viewerCanDelete: viewerCanDelete,
        isRoot: isRoot,
        number: GraphQLIDDecode(id: id, separator: "IssueComment")
    )
}

func createAssigneeModel(assigneeFields: AssigneeFields) -> IssueAssigneesModel {
    var models = [IssueAssigneeViewModel]()
    for node in assigneeFields.assignees.nodes ?? [] {
        guard let node = node,
            let url = URL(string: node.avatarUrl)
            else { continue }
        models.append(IssueAssigneeViewModel(login: node.login, avatarURL: url))
    }
    return IssueAssigneesModel(users: models, type: .assigned)
}