Added new tools for site and location discovery (#3)

* Added new tools for site and location discovery

* Updated tests for new tools

Author: dlrust

README.md

@@ -46,10 +46,10 @@ You can add the `devhub-cms-mcp` by updating the environment variables below
 
 ```
 claude mcp add devhub-cms-mcp \
-	-e DEVHUB_API_KEY=YOUR_KEY_HERE \
-	-e DEVHUB_API_SECRET=YOUR_SECRET_HERE \
-	-e DEVHUB_BASE_URL=https://yourbrand.cloudfrontend.net \
-	-- uvx devhub-cms-mcp
+    -e DEVHUB_API_KEY=YOUR_KEY_HERE \
+    -e DEVHUB_API_SECRET=YOUR_SECRET_HERE \
+    -e DEVHUB_BASE_URL=https://yourbrand.cloudfrontend.net \
+    -- uvx devhub-cms-mcp
 ```
 
 ### Installing via Smithery
@@ -115,11 +115,14 @@ uv run main.py
 
 This MCP provides the following tools for interacting with DevHub CMS:
 
-### Location Management
+### Business and Location Management
 
-- **get_hours_of_operation(location_id)**: Gets the hours of operation for a specific DevHub location. Returns a structured list of time ranges for each day of the week.
+- **get_businesses()**: Gets all businesses within the DevHub account. Returns a list of businesses with their IDs and names.
+- **get_locations(business_id)**: Gets all locations for a specific business. Returns detailed location information including address, coordinates, and URLs.
+- **get_hours_of_operation(location_id, hours_type='primary')**: Gets the hours of operation for a specific DevHub location. Returns a structured list of time ranges for each day of the week.
 - **update_hours(location_id, new_hours, hours_type='primary')**: Updates the hours of operation for a DevHub location.
 - **get_nearest_location(business_id, latitude, longitude)**: Finds the nearest DevHub location based on geographic coordinates.
+- **site_from_url(url)**: Gets the DevHub site ID and details from a URL. Returns site ID, URL, and associated location IDs.
 
 ### Content Management
 
@@ -150,13 +153,13 @@ uv pip install -e ".[test]"
 Run the tests with pytest:
 
 ```bash
-pytest
+uv run pytest
 ```
 
 For more detailed output and test coverage information:
 
 ```bash
-pytest -v --cov=devhub_cms_mcp
+uv run pytest -v --cov=devhub_cms_mcp
 ```
 
 ### Test Structure

src/devhub_cms_mcp/server.py

@@ -1,5 +1,6 @@
 import json
 import os
+from urllib.parse import urlparse
 
 from mcp.server.fastmcp import FastMCP
 from requests_oauthlib import OAuth1Session
@@ -45,6 +46,66 @@ def get_hours_of_operation(location_id: int, hours_type: str = 'primary') -> lis
     return content['hours_by_type'].get(hours_type, [])
 
 
+@mcp.tool()
+def get_businesses() -> list:
+    """Get all businesses within the DevHub account
+
+    Returns a list of businesses with the following fields:
+    - id: Business ID that can be used in the other tools
+    - business_name: Business name
+
+    If only one business exists in the account, you can assume that the user wants to use that business for any business_id related tools.
+    """
+    client, base_url = get_client()
+    params = {
+        'deleted': 0,
+        'limit': 20,
+        'order_by': 'business_name',
+        'project_type': 'default',
+    }
+    r = client.get('{}businesses/'.format(base_url), params=params)
+    content = json.loads(r.content)
+    return content['objects']
+
+
+@mcp.tool()
+def get_locations(business_id: int) -> list:
+    """Get all locations for a business
+
+    Returns a list of locations with the following fields:
+    - id: Location ID that can be used in the other tools
+    - location_name: Location name
+    - location_url: Location URL in DevHub
+    - street: Street address
+    - city: City
+    - state: State
+    - country: Country
+    - postal_code: Postal code
+    - lat: Latitude
+    - lon: Longitude
+    """
+    client, base_url = get_client()
+    params = {
+        'business_id': business_id,
+        'limit': 600,
+        'order_by': 'location_name',
+    }
+    r = client.get('{}locations/'.format(base_url), params=params)
+    content = json.loads(r.content)
+    return [{
+        'id': location['id'],
+        'location_name': location['location_name'],
+        'location_url': location['location_url'],
+        'street': location['street'],
+        'city': location['city'],
+        'state': location['state'],
+        'country': location['country'],
+        'postal_code': location['postal_code'],
+        'lat': location['lat'],
+        'lon': location['lon'],
+    } for location in content['objects']]
+
+
 @mcp.tool()
 def update_hours(location_id: int, new_hours: list, hours_type: str = 'primary') -> str:
     """Update the hours of operation for a DevHub location
@@ -81,6 +142,42 @@ def update_hours(location_id: int, new_hours: list, hours_type: str = 'primary')
     return 'Updated successfully'
 
 
+@mcp.tool()
+def site_from_url(url: str) -> str:
+    """Get the DevHub site ID from a URL.
+
+    Can prompt the user for the URL instead of passing a site_id.
+
+    Returns details about the Site matches the URL that can be used in the other tools.
+    - Site ID: ID of the DevHub site
+    - Site URL: URL of the DevHub site
+    - Site Location IDs: List of location IDs associated with the site
+
+    Args:
+        url: URL of the DevHub site, all lowercase and ends with a slash
+    """
+    parsed_url = urlparse(url)
+    subdomain = parsed_url.netloc.split('.', 1)[0] or 'www'
+    domain = parsed_url.netloc.split('.', 1)[1]
+    base_directory = parsed_url.path
+    client, base_url = get_client()
+    r = client.get('{}sites/'.format(base_url), params={
+        'base_directory': base_directory,
+        'deleted': 0,
+        'domain': domain,
+        'subdomain': subdomain,
+    })
+    content = json.loads(r.content)
+    if len(content['objects']) == 0:
+        return 'No site found'
+    site = content['objects'][0]
+    return f"""
+Site ID: {site['id']}
+Site URL: {site['formatted_url']}
+Site Location IDs: {", ".join([str(location_id) for location_id in site['location_ids']])}
+"""
+
+
 @mcp.tool()
 def upload_image(base64_image_content: str, filename: str) -> str:
     """Upload an image to the DevHub media gallery