import { Injectable } from '@angular/core';
import { HttpClient, HttpParams } from '@angular/common/http';
import { Observable } from 'rxjs';
import { Artifact, ArtifactQuery, ApiInfo } from '../models/artifact.model';

@Injectable({
  providedIn: 'root'
})
export class ArtifactService {
  private apiUrl = '/api/v1/artifacts';
  private baseUrl = '/api';

  constructor(private http: HttpClient) {}

  getApiInfo(): Observable<ApiInfo> {
    return this.http.get<ApiInfo>(this.baseUrl);
  }

  listArtifacts(limit: number = 100, offset: number = 0): Observable<Artifact[]> {
    const params = new HttpParams()
      .set('limit', limit.toString())
      .set('offset', offset.toString());
    return this.http.get<Artifact[]>(this.apiUrl + '/', { params });
  }

  getArtifact(id: number): Observable<Artifact> {
    return this.http.get<Artifact>(`${this.apiUrl}/${id}`);
  }

  queryArtifacts(query: ArtifactQuery): Observable<Artifact[]> {
    return this.http.post<Artifact[]>(`${this.apiUrl}/query`, query);
  }

  uploadArtifact(formData: FormData): Observable<Artifact> {
    return this.http.post<Artifact>(`${this.apiUrl}/upload`, formData);
  }

  downloadArtifact(id: number): Observable<Blob> {
    return this.http.get(`${this.apiUrl}/${id}/download`, {
      responseType: 'blob'
    });
  }

  deleteArtifact(id: number): Observable<any> {
    return this.http.delete(`${this.apiUrl}/${id}`);
  }

  generateSeedData(count: number): Observable<any> {
    return this.http.post(`/api/v1/seed/generate/${count}`, {});
  }
}