add search function, add support for asas-sn bands

Author: nabeelre

src/App.jsx

@@ -122,6 +122,8 @@ export default function App() {
   const [filterOpen, setFilterOpen] = useState(true);
   const [bottomH, setBottomH] = useState(504);
   const [activeBands, setActiveBands] = useState(null);
+  const [searchInput, setSearchInput] = useState('');
+  const [searchError, setSearchError] = useState(null);
 
   const containerRef = useRef(null);
   const dsRef = useRef(null);
@@ -191,6 +193,26 @@ export default function App() {
     }
   };
 
+  const searchByID = async () => {
+    const id = searchInput.trim();
+    if (!id || !dsRef.current) return;
+    setLoading(true);
+    setSearchError(null);
+    setError(null);
+    try {
+      const found = await dsRef.current.findBySourceId(id);
+      if (found) {
+        setCurrentRow(found);
+      } else {
+        setSearchError(`No star with ID ${id}`);
+      }
+    } catch (err) {
+      setSearchError(err.message);
+    } finally {
+      setLoading(false);
+    }
+  };
+
   const toggleClass = (cls) =>
     setEnabledClasses((prev) => {
       const next = new Set(prev);
@@ -499,6 +521,58 @@ export default function App() {
           </svg>
           Random star
         </button>
+
+        <form
+          onSubmit={(e) => { e.preventDefault(); searchByID(); }}
+          style={{ ...GLASS, padding: '14px 16px' }}
+        >
+          <div style={{ ...KICKER, marginBottom: 8 }}>
+            Search by Gaia DR3 ID
+          </div>
+          <div style={{ display: 'flex', gap: 6 }}>
+            <input
+              type="text"
+              inputMode="numeric"
+              value={searchInput}
+              onChange={(e) => { setSearchInput(e.target.value); setSearchError(null); }}
+              disabled={loading}
+              style={{
+                flex: 1, minWidth: 0,
+                padding: '8px 10px', borderRadius: 6,
+                border: '1px solid rgba(125,169,255,0.2)',
+                background: 'rgba(125,169,255,0.08)',
+                color: '#e8ecf6',
+                fontFamily: 'JetBrains Mono, monospace', fontSize: 15,
+                letterSpacing: 0.3, outline: 'none',
+              }}
+            />
+            <button
+              type="submit"
+              disabled={loading || !searchInput.trim()}
+              style={{
+                padding: '0 14px', borderRadius: 6,
+                border: '1px solid rgba(125,169,255,0.25)',
+                background: 'rgba(125,169,255,0.12)',
+                color: ACCENT,
+                fontFamily: 'JetBrains Mono, monospace', fontSize: 14,
+                fontWeight: 700, letterSpacing: 1.1, textTransform: 'uppercase',
+                cursor: (loading || !searchInput.trim()) ? 'default' : 'pointer',
+                opacity: (loading || !searchInput.trim()) ? 0.45 : 1,
+              }}
+            >
+              Go
+            </button>
+          </div>
+          {searchError && (
+            <div style={{
+              marginTop: 8,
+              fontFamily: 'JetBrains Mono, monospace', fontSize: 13,
+              color: '#ff8a72', wordBreak: 'break-all',
+            }}>
+              {searchError}
+            </div>
+          )}
+        </form>
       </div>
       )}
 

src/bands.js

@@ -27,6 +27,14 @@ export const SURVEY_LIBRARY = [
     survey: 'CSS',
     bands: [
       { key: 'clear_CSS', label: 'clear', color: '#c8d4e8' },
+      { key: 'clear',     label: 'clear', color: '#c8d4e8' },
+    ],
+  },
+  {
+    survey: 'ASAS-SN',
+    bands: [
+      { key: 'g_ASASSN', label: 'g', color: '#6ee7a8' },
+      { key: 'V_ASASSN', label: 'V', color: '#f4c542' },
     ],
   },
 ];

src/data/DataSource.js

@@ -41,4 +41,14 @@ export class DataSource {
   async getSummary() {
     return null;
   }
+
+  /**
+   * Find a row by its gaia_dr3_source_id. Returns the row object or null.
+   * @param {string|number|bigint} id
+   * @returns {Promise<object|null>}
+   */
+  // eslint-disable-next-line no-unused-vars
+  async findBySourceId(id) {
+    throw new Error("DataSource.findBySourceId not implemented");
+  }
 }

src/data/HFDataSource.js

@@ -87,6 +87,38 @@ export class HFDataSource extends DataSource {
     return collected;
   }
 
+  // Uses the datasets-server /filter endpoint so the lookup happens server-side
+  // at full int64 precision (gaia_dr3_source_id is 19 digits; JS Number rounds).
+  // Datasets store the id as either int64 or string; DuckDB's WHERE is type-
+  // strict, so we try the quoted-string form first then the bare-int form.
+  async findBySourceId(id) {
+    const numeric = String(id).trim();
+    if (!/^\d+$/.test(numeric)) {
+      throw new Error("Source ID must be numeric");
+    }
+    let lastErr = null;
+    for (const literal of [`'${numeric}'`, numeric]) {
+      const url = `${BASE}/filter?${this._qs({
+        dataset: this.dataset,
+        config: this.config,
+        split: this.split,
+        where: `gaia_dr3_source_id=${literal}`,
+        length: "1",
+      })}`;
+      const res = await fetch(url);
+      if (!res.ok) {
+        lastErr = `Search failed: ${res.status} ${res.statusText}`;
+        continue;
+      }
+      const data = await res.json();
+      const rows = (data?.rows || []).map((r) => r.row);
+      if (rows[0]) return rows[0];
+      lastErr = null;
+    }
+    if (lastErr) throw new Error(lastErr);
+    return null;
+  }
+
   // Fetches a pre-computed summary.json from the dataset repo. Build/upload via
   // scripts/build_summary.py. Returns null gracefully if absent so the app
   // degrades to global random sampling without sky map or class filter.

src/data/HFDiskDataSource.js

@@ -160,6 +160,24 @@ export class HFDiskDataSource extends DataSource {
     return this._summary;
   }
 
+  // Arrow ids are int64 (BigInt); compare as strings to preserve precision.
+  async findBySourceId(id) {
+    await this._scan();
+    const target = String(id).trim();
+    for (let shardIdx = 0; shardIdx < this._shards.length; shardIdx++) {
+      const table = await this._loadShard(shardIdx);
+      const idCol = table.getChild("gaia_dr3_source_id");
+      if (!idCol) return null;
+      for (let i = 0; i < table.numRows; i++) {
+        const v = idCol.get(i);
+        if (v != null && String(v) === target) {
+          return extractRow(table, table.schema.fields, i);
+        }
+      }
+    }
+    return null;
+  }
+
   async _loadShard(idx) {
     if (this._cachedShard?.idx === idx) return this._cachedShard.table;
     const file = this._shards[idx];

src/data/LocalDataSource.js

@@ -57,6 +57,12 @@ export class LocalDataSource extends DataSource {
     return rows.slice(offset, offset + length);
   }
 
+  async findBySourceId(id) {
+    const rows = await this._load();
+    const target = String(id).trim();
+    return rows.find((r) => String(r.gaia_dr3_source_id) === target) ?? null;
+  }
+
   async getSummary() {
     const rows = await this._load();
     const classIndices = {};