// Copyright 2025 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef IOS_CHROME_BROWSER_COMPOSEBOX_UI_COMPOSEBOX_INPUT_ITEM_H_
#define IOS_CHROME_BROWSER_COMPOSEBOX_UI_COMPOSEBOX_INPUT_ITEM_H_

#import <UIKit/UIKit.h>

#import "ios/chrome/browser/composebox/public/composebox_input_item_source.h"
#import "url/gurl.h"

namespace base {
class UnguessableToken;
}  // namespace base

// Enum for the loading state of an item.
enum class ComposeboxInputItemState {
  kLoading,
  kUploading,
  kLoaded,
  kError,
};

// Enum for the aim input item type.
enum class ComposeboxInputItemType {
  kComposeboxInputItemTypeImage,
  kComposeboxInputItemTypePDF,
  kComposeboxInputItemTypeRawFile,
  kComposeboxInputItemTypeTab,
  kComposeboxInputItemTypeDrive,
};

// Data object for an item in the AIM input.
@interface ComposeboxInputItem : NSObject <NSCopying>

- (instancetype)initWithComposeboxInputItemType:(ComposeboxInputItemType)type
                                        assetID:(NSString*)assetID
                                         source:
                                             (ComposeboxInputItemSource)source
    NS_DESIGNATED_INITIALIZER;
- (instancetype)initWithComposeboxInputItemType:(ComposeboxInputItemType)type
                                         source:
                                             (ComposeboxInputItemSource)source;
- (instancetype)init NS_UNAVAILABLE;

// The item's identifier.
@property(nonatomic, assign, readonly) const base::UnguessableToken& identifier;
// The server token for this item, used to identify the item on the server.
@property(nonatomic, assign) base::UnguessableToken serverToken;
// The preview image for this item.
@property(nonatomic, strong) UIImage* previewImage;
// The icon image for this item. Only set for file and tab types.
@property(nonatomic, strong) UIImage* leadingIconImage;
// The title for this item.
@property(nonatomic, copy) NSString* title;
// The local upload index for this item.
@property(nonatomic, assign) NSInteger uploadIndex;
// The current state of the item.
@property(nonatomic, assign) ComposeboxInputItemState state;
// The type of the input item.
@property(nonatomic, assign) ComposeboxInputItemType type;
// Optional, uniquely identifying the asset the item is associated with.
@property(nonatomic, copy, readonly) NSString* assetID;
// The source of the item.
@property(nonatomic, assign) ComposeboxInputItemSource source;
// Whether or not an animation was performed for this item. Only applied for tab
// items when composebox is presented in the assistant sheet.
@property(nonatomic, assign) BOOL performedAnimation;
// The image provider for this item. Only set for image types.
@property(nonatomic, strong) NSItemProvider* imageProvider;
// The file URL for this item. Only set for PDF and raw file types.
@property(nonatomic, strong) NSURL* fileURL;
// The tab URL for this item. Only set for tab types.
@property(nonatomic, assign) GURL tabURL;
// The Drive item identifier. Only set for Drive types.
@property(nonatomic, copy) NSString* driveIdentifier;
// The Drive resource key. Only set for Drive types.
@property(nonatomic, copy) NSString* driveResourceKey;
// The Drive mime type. Only set for Drive types.
@property(nonatomic, copy) NSString* driveMimeType;

@end

#endif  // IOS_CHROME_BROWSER_COMPOSEBOX_UI_COMPOSEBOX_INPUT_ITEM_H_