diff --git a/src/components/AppShell.tsx b/src/components/AppShell.tsx
index 0083504d..c9f97de0 100644
--- a/src/components/AppShell.tsx
+++ b/src/components/AppShell.tsx
@@ -469,6 +469,7 @@ function AppSidebar() {
)
}
+const SANDBOX_CHEMISTRY_REPORT = '/ocotillo/chemistry-report'
const SANDBOX_GEOTHERMAL_GRID = '/geothermal/wells/records-grid'
const SANDBOX_GEOTHERMAL_INVENTORY = '/geothermal/wells/inventory'
const SANDBOX_GEOTHERMAL_TEMP_DEPTH = '/geothermal/wells/temp-depth'
@@ -476,6 +477,7 @@ const SANDBOX_GEOTHERMAL_TEMP_DEPTH = '/geothermal/wells/temp-depth'
function isSandboxPath(pathname: string): boolean {
return (
pathname.startsWith('/example') ||
+ pathname.startsWith(SANDBOX_CHEMISTRY_REPORT) ||
pathname.startsWith(SANDBOX_GEOTHERMAL_GRID) ||
pathname.startsWith(SANDBOX_GEOTHERMAL_INVENTORY) ||
pathname.startsWith(SANDBOX_GEOTHERMAL_TEMP_DEPTH)
@@ -516,6 +518,19 @@ function ExampleNavItem() {
Typography
+ {/* Gated on the AMP.Staging group via ocotillo.chemistry-report. */}
+
+
+
+ Chemistry Reports
+
+
+
{
+ const { open: notify } = useNotification()
+ const [isGenerating, setIsGenerating] = useState(false)
+
+ const handleDownload = async () => {
+ if (!well) return
+
+ try {
+ setIsGenerating(true)
+ const filename = await downloadChemistryReport({
+ well,
+ contacts,
+ observations,
+ waterLevels,
+ year,
+ sections,
+ })
+
+ notify?.({
+ message: 'Chemistry report generated',
+ type: 'success',
+ description: filename,
+ })
+ } catch (error) {
+ console.error(error)
+ notify?.({
+ message: 'Chemistry report generation failed',
+ type: 'error',
+ })
+ } finally {
+ setIsGenerating(false)
+ }
+ }
+
+ return (
+
+ )
+}
diff --git a/src/components/Button/WellPDFActions.tsx b/src/components/Button/WellPDFActions.tsx
index ddd61c17..83aa22a0 100644
--- a/src/components/Button/WellPDFActions.tsx
+++ b/src/components/Button/WellPDFActions.tsx
@@ -1,20 +1,36 @@
-import { useState } from 'react'
+import { pdf } from '@react-pdf/renderer'
import { BaseRecord, useGo, useNotification } from '@refinedev/core'
-import { useParams } from 'react-router'
import { DownloadIcon, EyeIcon } from 'lucide-react'
+import { useState } from 'react'
+import { useParams } from 'react-router'
+import { WellPDF } from '@/components'
+import { downloadChemistryReport } from '@/components/pdf/chemistry'
import { Button } from '@/components/ui/button'
+import {
+ Select,
+ SelectContent,
+ SelectItem,
+ SelectTrigger,
+ SelectValue,
+} from '@/components/ui/select'
import {
Tooltip,
TooltipContent,
TooltipTrigger,
} from '@/components/ui/tooltip'
-import { WellPDF } from '@/components'
-import { buildPdfFilename, SensorDeploymentRow } from '@/utils'
-import { pdf } from '@react-pdf/renderer'
-import { IContact, IObservation, ISample, IWell } from '@/interfaces/ocotillo'
-import { IPdfOptions } from '@/interfaces'
import { PDF_SINGLE_PAGE_OPTION } from '@/config'
-import { useAccessCapabilities } from '@/hooks'
+import { useAccessCapabilities, useWellChemistryReport } from '@/hooks'
+import { IPdfOptions } from '@/interfaces'
+import { IContact, IObservation, ISample, IWell } from '@/interfaces/ocotillo'
+import { buildPdfFilename, SensorDeploymentRow } from '@/utils'
+
+/** The kinds of PDF the well details page can produce. */
+export type WellReportType = 'field-sheet' | 'chemistry-report'
+
+const REPORT_TYPE_LABELS: Record = {
+ 'field-sheet': 'Field sheet',
+ 'chemistry-report': 'Chemistry report',
+}
type WellPDFActionsButtonProps = {
isPreviewLoading: boolean
@@ -48,60 +64,136 @@ export const WellPDFActionsButton = ({
isLoading: isPermissionsLoading,
canManageAmp,
canViewConfidential,
+ canViewAmpStaging,
} = useAccessCapabilities()
+ const [reportType, setReportType] = useState('field-sheet')
const [isGenerating, setIsGenerating] = useState(false)
+ const isChemistry = reportType === 'chemistry-report'
+
+ const {
+ reportYear,
+ hasChemistry,
+ isLoading: isChemistryLoading,
+ fetchYearObservations,
+ fetchWaterLevels,
+ } = useWellChemistryReport({
+ thingId: id,
+ // Only worth asking once the report is on offer at all.
+ enabled: canViewAmpStaging,
+ })
+
+ // A well with no chemistry on file still gets a report, marked as having no
+ // results — the same thing the exporter produces, and the honest answer to
+ // "what does this well's water look like". The note only warns what is coming.
+ const chemistryNote =
+ isChemistry && !isChemistryLoading && !hasChemistry
+ ? 'No water chemistry on file — the report will show no results'
+ : undefined
+
+ // Waiting on the year is the one thing that has to hold the actions back,
+ // since acting early would report on the wrong one.
+ const isChemistryYearPending = isChemistry && isChemistryLoading
+
const previewDisabled =
- isPreviewLoading || isPermissionsLoading || !canManageAmp
+ isPreviewLoading ||
+ isPermissionsLoading ||
+ !canManageAmp ||
+ isChemistryYearPending
const downloadDisabled =
isDownloadLoading ||
isPermissionsLoading ||
!canManageAmp ||
- isGenerating
+ isGenerating ||
+ isChemistryYearPending
const handlePreview = () => {
+ if (isChemistry) {
+ // The chemistry exporter already renders a full preview; hand it the
+ // well and year so it opens on this report rather than an empty picker.
+ go({
+ to: '/ocotillo/chemistry-report',
+ query: { thing_id: id, year: reportYear },
+ type: 'push',
+ })
+ return
+ }
+
go({ to: `/ocotillo/well/pdf-preview/${id}`, type: 'push' })
}
+ const handleDownloadFieldSheet = async (opts: IPdfOptions) => {
+ const filename = buildPdfFilename(well)
+
+ const blob = await pdf(
+
+ ).toBlob()
+
+ const url = URL.createObjectURL(blob)
+ const a = document.createElement('a')
+ a.href = url
+ a.download = filename.endsWith('.pdf') ? filename : `${filename}.pdf`
+ a.click()
+ URL.revokeObjectURL(url)
+
+ return a.download
+ }
+
+ const handleDownloadChemistryReport = async (year: number) => {
+ const elevationFt = (
+ well?.current_location?.properties as
+ | { elevation?: number | null }
+ | undefined
+ )?.elevation
+
+ const [yearObservations, waterLevels] = await Promise.all([
+ fetchYearObservations(year),
+ fetchWaterLevels(year, { elevationFt }),
+ ])
+
+ return downloadChemistryReport({
+ well,
+ contacts,
+ observations: yearObservations,
+ waterLevels,
+ year,
+ })
+ }
+
const handleDownload = async (opts: IPdfOptions) => {
if (!well?.id) return
try {
setIsGenerating(true)
- const filename = buildPdfFilename(well)
-
- const blob = await pdf(
-
- ).toBlob()
-
- const url = URL.createObjectURL(blob)
- const a = document.createElement('a')
- a.href = url
- a.download = filename.endsWith('.pdf') ? filename : `${filename}.pdf`
- a.click()
- URL.revokeObjectURL(url)
+ const filename = isChemistry
+ ? await handleDownloadChemistryReport(reportYear)
+ : await handleDownloadFieldSheet(opts)
notify?.({
- message: 'PDF generated successfully',
+ message: isChemistry
+ ? `Chemistry report generated for ${reportYear}`
+ : 'PDF generated successfully',
type: 'success',
- description: a.download,
+ description: filename,
})
} catch (error) {
console.error(error)
notify?.({
- message: 'PDF Generation Failed',
+ message: isChemistry
+ ? 'Chemistry report generation failed'
+ : 'PDF Generation Failed',
type: 'error',
})
} finally {
@@ -109,14 +201,45 @@ export const WellPDFActionsButton = ({
}
}
+ const downloadTooltip = isGenerating
+ ? 'Generating…'
+ : isChemistry
+ ? (chemistryNote ?? `Download chemistry report for ${reportYear}`)
+ : 'Download field sheet'
+
return (
+
+