app/nodejs/app.js

// Copyright 2024 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

/*
app.js - application
*/

'use strict';

import express from 'express';
import logger from './logger.js';
import utils from './utils.js';
import { engine } from 'express-handlebars';

const app = express();

app.engine('handlebars', engine());
app.set('view engine', 'handlebars');
app.set('views', './views');

app.get('/', async (req, res) => {
  if (!process.env.PROCESSED_DATA_BUCKET) {
    res.status(500).send('Environment variable PROCESSED_DATA_BUCKET required');
  } else {
    const fur = req.query.fur || 'Gray';
    const age = req.query.age || 'Adult';
    const location = req.query.location || 'Above Ground';

    logger.info(
      `Request received for fur: ${fur}, age: ${age}, location: ${location}`,
    );

    const [squirrelCount, dataPoints] = await utils.retrieveData(fur, age, location);

    res.render('index', {
      layout: false,
      squirrel_count: squirrelCount, data_points: dataPoints,
    });
  }
});

export default app;