main.py

import os
from dotenv import load_dotenv
import asyncio
import time
import socket

load_dotenv()

from viam.robot.client import RobotClient
from viam.rpc.dial import Credentials, DialOptions
from viam.components.board import Board
from viam.components.servo import Servo

async def connect():
    opts = RobotClient.Options.with_api_key( 
        api_key=os.getenv('api_key'),
        api_key_id=os.getenv('api_key_id')
    )
    return await RobotClient.at_address('lunch-gong-main.ldnf5a9i14.viam.cloud', opts)

async def moveHammer():
    machine = await connect()

    print('Resources:')
    print(machine.resource_names)
    
    # Note that the pin supplied is a placeholder. Please change this to a valid pin you are using.
    # board-1
    board_1 = Board.from_robot(machine, "board-1")
    board_1_return_value = await board_1.gpio_pin_by_name("16")
    print(f"board-1 gpio_pin_by_name return value: {board_1_return_value}")
  
    # servo-1
    servo_1 = Servo.from_robot(machine, "servo-1")
    servo_1_return_value = await servo_1.get_position()
    print(f"servo-1 get_position return value: {servo_1_return_value}")
    await servo_1.move(0)
    time.sleep(1)
    await servo_1.move(180)
    # Don't forget to close the machine when you're done!
    await machine.close()


async def handle_client(client_socket, address):
    print(f"Connection from {address}")
    while True:
        data = client_socket.recv(1024)
        if not data:
            break
        print(f"Received from {address}: {data.decode('utf-8')}")
        await moveHammer()
    client_socket.close()
    print(f"Connection closed from {address}")

async def start_server(host='0.0.0.0', port=7555):
    server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server.bind((host, port))
    server.listen(5)
    print(f"Server listening on {host}:{port}")

    while True:
        client_sock, address = server.accept()
        await handle_client(client_sock, address)

if __name__ == '__main__':
    asyncio.run(start_server())